Update demucs_split.py

demucs_split.py

@@ -1,32 +1,20 @@
-import demucs.separate
-import os
-import shutil
-import json
-import time
-
-def clear_folder(folder_path):
-        for item in os.listdir(folder_path):
-            item_path = os.path.join(folder_path, item)
-            if os.path.isfile(item_path) or os.path.islink(item_path):
-                os.unlink(item_path)
-            else:
-                shutil.rmtree(item_path)
-
-def demucs_split(
-        url_or_path: str,
-        demucs_model: str = "mdx_extra"
-        ):
-    
-    input_file = "input.mp3"
-    
-    def clean_temp():
-        clear_folder("splited_audio_output")
-        
-    input_file = url_or_path
-    # Разделение аудио
-    print("Separating audio...")
-    os.environ['XDG_CACHE_HOME'] = "./models_demucs/"
-    demucs_opts = ["--mp3", "--two-stems", "vocals", "-n", demucs_model, input_file, "-o", "splited_audio_output", "--filename", "{stem}.mp3"]
-    demucs.separate.main(demucs_opts)
-
+import demucs.separate
+import os
+def demucs_split(
+        url_or_path: str,
+        demucs_model: str = "mdx_extra"
+        ):
+    
+    input_file = "input.mp3"
+    
+    def clean_temp():
+        clear_folder("splited_audio_output")
+        
+    input_file = url_or_path
+    # Разделение аудио
+    print("Separating audio...")
+    os.environ['XDG_CACHE_HOME'] = "./models_demucs/"
+    demucs_opts = ["--mp3", "--two-stems", "vocals", "-n", demucs_model, input_file, "-o", "splited_audio_output", "--filename", "{stem}.mp3"]
+    demucs.separate.main(demucs_opts)
+
     return f"splited_audio_output/{demucs_model}/vocals.mp3", f"splited_audio_output/{demucs_model}/no_vocals.mp3"