box up

app.py

@@ -94,50 +94,33 @@ def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, bac
         print("Error in custom separation:", str(e))
         return None
 
-    # 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
-    }
+    selected_stems = [vocals, bass, drums, other, piano, guitar, lead_vocals, backing_vocals]
+    output_files = [stem_files[stem] if selected_stems[i] and os.path.isfile(stem_files[stem]) else None for i, stem in enumerate(stem_files)]
 
-    return {stem: stem_files[stem] if selected_stems[stem] and os.path.isfile(stem_files[stem]) else None for stem in selected_stems}
+    return output_files
 
 # 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.outputs.Dynamic(output_component),
+    [gr.components.Audio(type="filepath", label=stem) for stem in ["Vocals", "Bass", "Drums", "Other", "Piano", "Guitar", "Lead Vocals", "Backing Vocals"]],
     title=title,
     description=description,
     live=True  # Enable live updates