Update app.py

app.py

@@ -12,7 +12,7 @@ import os
 import re
 import glob
 import spaces
-from scipy.io.wavfile import write
+from pathlib import Path
 
 ### Initializations
 
@@ -234,7 +234,7 @@ def clear_audio():
 ###
 
 @spaces.GPU
-def voice_extract_demucs(audio):
+def voice_extract_demucs(audio_path):
     """
     Returns the path of the voice extracted file.
     """
@@ -245,12 +245,20 @@ def voice_extract_demucs(audio):
             "demucs",
             "--two-stems=vocals",
             "--out", "out_demucs",
-            audio
+            audio_path
         ]
         subprocess.run(cmd, check=True)
-        voice_path = os.path.join(".", "demucs", "htdemucs", "audio_file", "vocals.wav")
-        success_message = "✅ **Success!** Voice extracted. ("+voice_path+")"
-        return voice_path, voice_path, gr.Markdown(success_message)
+        base_name = Path(audio_path).stem
+        model_dirs = list(Path("out_demucs").glob("*"))
+        if not model_dirs:
+            vocals_path = audio_path
+            success_message = "❌ **Error:** No files found on out_demucs."
+        else:
+            result_dir = model_dirs[0] / base_name    
+            vocals_path = result_dir / "vocals.wav"
+            success_message = "✅ **Success!** Voice extracted. ("+vocals_path+")"
+
+        return vocals_path, vocals_path, gr.Markdown(success_message)
     except Exception as e:
         return None, None, gr.Markdown(f"❌ **Error:** An unexpected ERROR occurred: {e}")
 ###