Update app.py

app.py

@@ -98,30 +98,34 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
 
     return output_file, separation_log
 
-# define the gradio interface
-iface: gr.Interface = gr.Interface(
-    fn=inference,
-    inputs=[
-        gr.Audio(type="filepath"),
-        gr.Dropdown(["htdemucs", "htdemucs_ft", "htdemucs_6s", "hdemucs_mmi", "mdx", "mdx_extra", "mdx_q", "mdx_extra_q"], label="model name", value="htdemucs_ft"),  # set default value
-        gr.Checkbox(label="vocals", value=True),
-        gr.Checkbox(label="drums", value=True),
-        gr.Checkbox(label="bass", value=True),
-        gr.Checkbox(label="other", value=True),
-        gr.Checkbox(label="save as mp3", value=False),  # set default value to false
-        gr.Slider(128, 320, step=32, label="mp3 bitrate", visible=False),  # set visible to false initially
-    ],
-    outputs=[
-        gr.Audio(type="filepath"),
-        gr.Textbox(label="separation log", lines=10),
-    ],
-    title="demucs music source separation and mixing",
-    description="separate vocals, drums, bass, and other instruments from your music using demucs and mix the selected stems.",
-)
+# Define the Gradio interface
+with gr.Blocks() as iface:
+    audio_input = gr.Audio(type="filepath")
+    model_dropdown = gr.Dropdown(["htdemucs", "htdemucs_ft", "htdemucs_6s", "hdemucs_mmi", "mdx", "mdx_extra", "mdx_q", "mdx_extra_q"], label="Model Name", value="htdemucs_ft")
+    vocals_checkbox = gr.Checkbox(label="Vocals", value=True)
+    drums_checkbox = gr.Checkbox(label="Drums", value=True)
+    bass_checkbox = gr.Checkbox(label="Bass", value=True)
+    other_checkbox = gr.Checkbox(label="Other", value=True)
+    mp3_checkbox = gr.Checkbox(label="Save as MP3", value=False)
+    mp3_bitrate = gr.Slider(128, 320, step=32, label="MP3 Bitrate", visible=False)
+    
+    output_audio = gr.Audio(type="filepath")
+    separation_log = gr.Textbox(label="Separation Log", lines=10)
+    
+    submit_btn = gr.Button("Process")
+    
+    submit_btn.click(
+        fn=inference,
+        inputs=[audio_input, model_dropdown, vocals_checkbox, drums_checkbox, bass_checkbox, other_checkbox, mp3_checkbox, mp3_bitrate],
+        outputs=[output_audio, separation_log]
+    )
+    
+    # Make MP3 bitrate slider visible only when "Save as MP3" is checked
+    mp3_checkbox.change(
+        fn=lambda mp3: gr.update(visible=mp3),
+        inputs=mp3_checkbox,
+        outputs=mp3_bitrate
+    )
 
-# make mp3 bitrate slider visible only when "save as mp3" is checked
-# iface.inputs[-2].change(fn=lambda mp3: gr.update(visible=mp3), inputs=iface.inputs[-2], outputs=iface.inputs[-1])
-iface.components[-2].change(fn=lambda mp3: gr.update(visible=mp3), inputs=iface.components[-2], outputs=iface.components[-1])
-
-# launch the gradio interface
+# Launch the Gradio interface
 iface.launch()