Update app.py

app.py

@@ -100,20 +100,31 @@ def inference(audio_file: str, model_name: str, vocals: bool, drums: bool, bass:
 
 # 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")
+    gr.Markdown("# Demucs Music Source Separation and Mixing")
+    gr.Markdown("Separate vocals, drums, bass, and other instruments from your music using Demucs and mix the selected stems.")
     
+    with gr.Row():
+        with gr.Column(scale=1):
+            audio_input = gr.Audio(type="filepath", label="Upload Audio File")
+            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"
+            )
+            with gr.Row():
+                vocals_checkbox = gr.Checkbox(label="Vocals", value=True)
+                drums_checkbox = gr.Checkbox(label="Drums", value=True)
+            with gr.Row():
+                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)
+            submit_btn = gr.Button("Process", variant="primary")
+
+        with gr.Column(scale=1):
+            output_audio = gr.Audio(type="filepath", label="Processed Audio")
+            separation_log = gr.Textbox(label="Separation Log", lines=10)
+
     submit_btn.click(
         fn=inference,
         inputs=[audio_input, model_dropdown, vocals_checkbox, drums_checkbox, bass_checkbox, other_checkbox, mp3_checkbox, mp3_bitrate],