Update app.py

app.py

@@ -26,7 +26,7 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     # Construct the Demucs command
     cmd = [
         "python", "-m", "demucs", 
-        "--out", output_dir,
+        "--out", os.path.dirname(output_dir),  # Use the parent directory of output_dir
         "-n", model_name,
         audio_file
     ]
@@ -56,9 +56,13 @@ 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)
+    yield None, stream_log(f"Searching for stems in: {stem_search_dir}")
+
     stems: Dict[str, str] = {}
     for stem in ["vocals", "drums", "bass", "other"]:
-        stem_path = os.path.join(output_dir, model_name, f"{stem}.wav")
+        stem_path = os.path.join(stem_search_dir, f"{stem}.wav")
         yield None, stream_log(f"Checking for {stem} stem at: {stem_path}")
         if os.path.exists(stem_path):
             stems[stem] = stem_path
@@ -66,6 +70,18 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
         else:
             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
+        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}")
+            if os.path.exists(stem_path):
+                stems[stem] = stem_path
+                yield None, stream_log(f"Found {stem} stem")
+            else:
+                yield None, stream_log(f"Warning: {stem} stem not found")
+
     yield None, stream_log(f"All found stems: {list(stems.keys())}")
 
     selected_stems: List[str] = []