parametrize sample rate of file conversion methods

moises-ai · Mar 5, 2024 · 8233648 · 8233648
1 parent c387675
commit 8233648
Show file tree

Hide file tree

Showing 3 changed files with 153 additions and 224 deletions.
diff --git a/maestro_worker_python/convert_files.py b/maestro_worker_python/convert_files.py
@@ -24,6 +24,7 @@ class FileToConvert:
     file_format: str
     output_file_path: str = None
     max_duration: int = 1200
+    sr: int = 44100
 
 
 def convert_files(convert_files: List[FileToConvert]):
@@ -89,7 +90,7 @@ def convert_files_manager(*convert_files: FileToConvert) -> None | str | list[st
             obj.close()
 
 
-def _convert_to_wav(input_file_path, output_file_path, max_duration):
+def _convert_to_wav(input_file_path, output_file_path, max_duration, sr=44100):
     _run_subprocess(
         [
             "ffmpeg",
@@ -102,13 +103,13 @@ def _convert_to_wav(input_file_path, output_file_path, max_duration):
             "-i",
             str(input_file_path),
             "-ar",
-            "44100",
+            str(sr),
             str(output_file_path),
         ]
     )
 
 
-def _convert_to_m4a(input_file_path, output_file_path, max_duration):
+def _convert_to_m4a(input_file_path, output_file_path, max_duration, sr=44100):
     _run_subprocess(
         [
             "ffmpeg",
@@ -125,7 +126,7 @@ def _convert_to_m4a(input_file_path, output_file_path, max_duration):
             "-b:a",
             "192k",
             "-ar",
-            "44100",
+            str(sr),
             "-movflags",
             "+faststart",
             str(output_file_path),