Update app.py

app.py

@@ -20,13 +20,15 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
         yield None, stream_log("Error: No audio file provided")
         raise gr.Error("Please upload an audio file")
 
-    output_dir = os.path.join("separated", model_name, os.path.splitext(os.path.basename(audio_file))[0])
+    # Use absolute paths
+    base_output_dir = os.path.abspath("separated")
+    output_dir = os.path.join(base_output_dir, model_name, os.path.splitext(os.path.basename(audio_file))[0])
     os.makedirs(output_dir, exist_ok=True)
 
-    # Construct the Demucs command
+    # Construct the Demucs command with full paths
     cmd = [
         "python", "-m", "demucs", 
-        "--out", os.path.dirname(output_dir),  # Use the parent directory of output_dir
+        "--out", base_output_dir,
         "-n", model_name,
         audio_file
     ]
@@ -56,8 +58,8 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
 
     yield None, stream_log("Separation completed. Processing stems...")
 
-    # Change the stem search directory
-    stem_search_dir = os.path.join(output_dir, model_name)
+    # Change the stem search directory using full path
+    stem_search_dir = os.path.join(base_output_dir, model_name, os.path.splitext(os.path.basename(audio_file))[0])
     yield None, stream_log(f"Searching for stems in: {stem_search_dir}")
 
     stems: Dict[str, str] = {}
@@ -71,8 +73,8 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
             yield None, stream_log(f"Warning: {stem} stem not found")
 
     if not stems:
-        yield None, stream_log("Error: No stems found. Checking parent directory...")
-        stem_search_dir = output_dir
+        yield None, stream_log("Error: No stems found. Checking alternative directory...")
+        stem_search_dir = os.path.join(base_output_dir, model_name)
         for stem in ["vocals", "drums", "bass", "other"]:
             stem_path = os.path.join(stem_search_dir, f"{stem}.wav")
             yield None, stream_log(f"Checking for {stem} stem at: {stem_path}")