boxes modify

app.py

@@ -96,41 +96,49 @@ def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, bac
 
     # Collecting all file paths
     stem_files = {
-        "vocals": "./out/htdemucs_6s/test/vocals.wav",
-        "bass": "./out/htdemucs_6s/test/bass.wav",
-        "drums": "./out/htdemucs_6s/test/drums.wav",
-        "other": "./out/htdemucs_6s/test/other.wav",
-        "piano": "./out/htdemucs_6s/test/piano.wav",
-        "guitar": "./out/htdemucs_6s/test/guitar.wav",
-        "lead_vocals": primary_stem_path,
-        "backing_vocals": secondary_stem_path
+        "Vocals": "./out/htdemucs_6s/test/vocals.wav",
+        "Bass": "./out/htdemucs_6s/test/bass.wav",
+        "Drums": "./out/htdemucs_6s/test/drums.wav",
+        "Other": "./out/htdemucs_6s/test/other.wav",
+        "Piano": "./out/htdemucs_6s/test/piano.wav",
+        "Guitar": "./out/htdemucs_6s/test/guitar.wav",
+        "Lead Vocals": primary_stem_path,
+        "Backing Vocals": secondary_stem_path
     }
 
     # Filter out unchecked stems
     selected_stems = {
-        "vocals": vocals,
-        "bass": bass,
-        "drums": drums,
-        "other": other,
-        "piano": piano,
-        "guitar": guitar,
-        "lead_vocals": lead_vocals,
-        "backing_vocals": backing_vocals
+        "Vocals": vocals,
+        "Bass": bass,
+        "Drums": drums,
+        "Other": other,
+        "Piano": piano,
+        "Guitar": guitar,
+        "Lead Vocals": lead_vocals,
+        "Backing Vocals": backing_vocals
     }
 
-    return [stem_files[stem] if selected_stems[stem] and os.path.isfile(stem_files[stem]) else None for stem in selected_stems]
+    return {stem: stem_files[stem] if selected_stems[stem] and os.path.isfile(stem_files[stem]) else None for stem in selected_stems}
 
 # Checkbox for each stem
 checkboxes = [gr.components.Checkbox(label=stem) for stem in ["Vocals", "Bass", "Drums", "Other", "Piano", "Guitar", "Lead Vocals", "Backing Vocals"]]
 
+# Custom output component function
+def output_component(stem_results):
+    components = []
+    for stem, file_path in stem_results.items():
+        if file_path:
+            components.append(gr.Audio(file_path, label=stem))
+    return components
+
 # Gradio Interface
 title = "Source Separation Demo"
 description = "Music Source Separation in the Waveform Domain. To use it, simply upload your audio and select the stems you want to display."
 gr.Interface(
     inference, 
     [gr.components.Audio(type="numpy", label="Input")] + checkboxes,
-    [gr.components.Audio(type="filepath", label=stem, optional=True) for stem in ["Vocals", "Bass", "Drums", "Other", "Piano", "Guitar", "Lead Vocals", "Backing Vocals"]],
+    gr.outputs.Dynamic(output_component),
     title=title,
     description=description,
     live=True  # Enable live updates
-).launch()
+).launch()