Updated with checkbox done

app.py

@@ -60,7 +60,6 @@
 
 
 
-
 import os
 import gradio as gr
 from scipy.io.wavfile import write
@@ -75,11 +74,13 @@ def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, bac
     audio_path = 'test.wav'
     write(audio_path, audio[0], audio[1])
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    print(f"Using device: {device}")
     if device=='cuda':
         use_cuda=True
         print(f"Using device: {device}")
     else:
         use_cuda=False
+        print(f"Using device: {device}")
 
     try:
         command = f"python3 -m demucs.separate -n htdemucs_6s -d {device} {audio_path} -o out"
@@ -87,14 +88,14 @@ def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, bac
         print("Demucs script output:", process.stdout.decode())
     except subprocess.CalledProcessError as e:
         print("Error in Demucs script:", e.stderr.decode())
-        return None, None, None, None, None, None, None, None
+        return [gr.Audio(visible=False)] * 8
 
     try:
-        separator = Separator("./out/htdemucs_6s/test/vocals.wav", model_name='UVR_MDXNET_KARA_2', use_cuda=device=='cuda', output_format='wav')
+        separator = Separator("./out/htdemucs_6s/test/vocals.wav", model_name='UVR_MDXNET_KARA_2', use_cuda=device==use_cuda, output_format='wav')
         primary_stem_path, secondary_stem_path = separator.separate()
     except Exception as e:
         print("Error in custom separation:", str(e))
-        return None, None, None, None, None, None, None, None
+        return [gr.Audio(visible=False)] * 8
 
     stem_paths = {
         "vocals": "./out/htdemucs_6s/test/vocals.wav" if vocals else None,
@@ -107,7 +108,7 @@ def inference(audio, vocals, bass, drums, other, piano, guitar, lead_vocals, bac
         "backing_vocals": secondary_stem_path if backing_vocals else None
     }
 
-    return [stem_paths[stem] for stem in stem_paths]
+    return [gr.Audio(stem_paths[stem], visible=bool(stem_paths[stem])) for stem in stem_paths]
 
 # Define checkboxes for each stem
 checkbox_labels = ["Full Vocals", "Bass", "Drums", "Other", "Piano", "Guitar", "Lead Vocals", "Backing Vocals"]
@@ -119,10 +120,11 @@ description = "Music Source Separation in the Waveform Domain. Upload your audio
 iface = gr.Interface(
     inference, 
     [gr.components.Audio(type="numpy", label="Input")] + checkboxes,
-    [gr.components.Audio(type="filepath", label=label) for label in checkbox_labels],
+    [gr.Audio(label=label, visible=False) for label in checkbox_labels],
     title=title,
     description=description,
-  
+    
 )
 
-iface.launch(debug=True)
+iface.launch()
+