update checkbox

app.py

@@ -76,8 +76,8 @@ def inference(audio):
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
     print(f"Using device: {device}")
 
-    # Run Demucs script
     try:
+        # Using subprocess.run for better control
         command = f"python3 -m demucs.separate -n htdemucs_6s -d {device} {audio_path} -o out"
         process = subprocess.run(command, shell=True, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         print("Demucs script output:", process.stdout.decode())
@@ -85,9 +85,9 @@ def inference(audio):
         print("Error in Demucs script:", e.stderr.decode())
         return None
 
-    # Run custom separator
     use_cuda = device == 'cuda'
     try:
+        # Separating the stems using your custom separator
         separator = Separator("./out/htdemucs_6s/test/vocals.wav", model_name='UVR_MDXNET_KARA_2', use_cuda=use_cuda, output_format='mp3')
         primary_stem_path, secondary_stem_path = separator.separate()
     except Exception as e:
@@ -108,26 +108,24 @@ def inference(audio):
 
     return existing_files
 
-# Function to return selected audio
-def get_selected_audio(stems, selected_stem):
-    return stems.get(selected_stem, None)
+# Function to return selected audios
+def get_selected_audios(stems, *selected_stems):
+    return [stems[stem] for stem in selected_stems if stem in stems]
 
 # 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")
-dropdown = gr.components.Dropdown(label="Select Stem", choices=["vocals", "bass", "drums", "other", "piano", "guitar", "lead_vocals", "backing_vocals"])
-audio_output = gr.components.Audio(type="filepath", label="Output")
+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"]]
 
 gr.Interface(
-    fn=lambda audio, stem: get_selected_audio(inference(audio), stem),
-    inputs=[audio_input, dropdown], 
-    outputs=audio_output,
+    fn=lambda audio, *args: get_selected_audios(inference(audio), *args),
+    inputs=[audio_input, *checkboxes], 
+    outputs=audio_outputs,
     title=title,
-    description=description,
-    live=True  # Enable live update for dynamic output
+    description=description
 ).launch()
 
 
-