Merge pull request #7 from joelmathewthomas/refactor/code-cleanup

Code refactoring

src/separation/demucs_wrapper.py

@@ -4,7 +4,7 @@ import soundfile as sf
 import tempfile
 from pathlib import Path
 
-def separate_audio_with_demucs(input_file, sample_rate, output_dir: str):
+def separate_audio_with_demucs(input_file, output_dir: str):
     """
     Use subprocess to run Demucs separation on an input audio file.
     Parameters:

tests/test_separation.py

@@ -36,7 +36,7 @@ def test_demucs_separation_with_preprocessing():
     new_audio_path = os.path.join(dir_path, f"{file_name}.wav")
     os.rename(temp_audio_path, new_audio_path)
 
-    separate_audio_with_demucs(new_audio_path, samplerate, output_path)
+    separate_audio_with_demucs(new_audio_path, output_path)
 
     # Verify the htdemucs folder exists
     demucs_dir = Path(output_path) / 'htdemucs'