Update app.py

app.py

@@ -59,19 +59,30 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
     stems: Dict[str, str] = {}
     for stem in ["vocals", "drums", "bass", "other"]:
         stem_path = os.path.join(output_dir, model_name, 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] = []
     for stem, selected in zip(["vocals", "drums", "bass", "other"], [vocals, drums, bass, other]):
-        if selected and stem in stems:
-            selected_stems.append(stems[stem])
-            yield None, stream_log(f"Selected {stem} stem for mixing")
+        if selected:
+            yield None, stream_log(f"{stem} is selected by user")
+            if stem in stems:
+                selected_stems.append(stems[stem])
+                yield None, stream_log(f"Selected {stem} stem for mixing")
+            else:
+                yield None, stream_log(f"Warning: {stem} was selected but not found")
+
+    yield None, stream_log(f"Final selected stems: {selected_stems}")
 
     if not selected_stems:
         yield None, stream_log("Error: No stems selected for mixing")
-        raise gr.Error("Please select at least one stem to mix.")
+        raise gr.Error("Please select at least one stem to mix and ensure it was successfully separated.")
 
     output_file: str = os.path.join(output_dir, "mixed.wav")
     yield None, stream_log("Mixing selected stems...")