radio button

app.py

@@ -59,6 +59,7 @@
 # ).launch()
 
 
+
 import os
 import gradio as gr
 from scipy.io.wavfile import write
@@ -108,24 +109,29 @@ def inference(audio):
 
     return existing_files
 
-# Function to return selected audios
+# Function to return selected audios along with their labels
 def get_selected_audios(stems, *selected_stems):
-    return [stems[stem] for stem in selected_stems if stem in stems]
+    return [(stem, stems[stem]) for stem in selected_stems if stem in stems and stems[stem]]
+
+# Custom output component
+def custom_output_component(label, filepath):
+    if filepath:
+        return gr.Audio(filepath, label=label, visible=True)
+    else:
+        return gr.Label("", visible=False)
 
 # Gradio Interface
 title = "Source Separation Demo"
 description = "Music Source Separation in the Waveform Domain. To use it, simply upload your audio."
 
-audio_input = gr.components.Audio(type="numpy", label="Input")
-checkboxes = [gr.components.Checkbox(label=stem) for stem in ["vocals", "bass", "drums", "other", "piano", "guitar", "lead_vocals", "backing_vocals"]]
-audio_outputs = [gr.components.Audio(type="filepath", label=stem) for stem in ["vocals", "bass", "drums", "other", "piano", "guitar", "lead_vocals", "backing_vocals"]]
+audio_input = gr.Audio(type="numpy", label="Input")
+checkboxes = [gr.Checkbox(label=stem) for stem in ["vocals", "bass", "drums", "other", "piano", "guitar", "lead_vocals", "backing_vocals"]]
+outputs = [gr.Dynamic(custom_output_component, label=stem) for stem in ["vocals", "bass", "drums", "other", "piano", "guitar", "lead_vocals", "backing_vocals"]]
 
 gr.Interface(
     fn=lambda audio, *args: get_selected_audios(inference(audio), *args),
-    inputs=[audio_input, *checkboxes], 
-    outputs=audio_outputs,
+    inputs=[audio_input, *checkboxes],
+    outputs=outputs,
     title=title,
     description=description
 ).launch()
-
-