Update gui.py

gui.py

@@ -0,0 +1,73 @@
+# gui.py
+
+import gradio as gr
+from downloader import downloader_conf, url_media_conf, url_button_conf, show_components_downloader
+from pipeline import sound_separate
+
+# Constants
+TITLE = "<center><strong><font size='7'>Audio🔹separator</font></strong></center>"
+DESCRIPTION = "This demo uses the MDX-Net models for vocal and background sound separation."
+
+
+def get_gui(theme):
+    with gr.Blocks(theme=theme) as app:
+        gr.Markdown(TITLE)
+        gr.Markdown(DESCRIPTION)
+
+        # Downloader UI
+        downloader_gui = downloader_conf()
+        with gr.Row():
+            with gr.Column(scale=2):
+                url_media_gui = url_media_conf()
+            with gr.Column(scale=1):
+                url_button_gui = url_button_conf()
+
+        downloader_gui.change(
+            show_components_downloader,
+            [downloader_gui],
+            [url_media_gui, url_button_gui]
+        )
+
+        # Audio file upload input
+        aud = gr.File(label="Audio file", type="filepath")
+
+        url_button_gui.click(
+            fn=audio_downloader,
+            inputs=[url_media_gui],
+            outputs=[aud]
+        )
+
+        # Stem choice
+        stem_gui = gr.Radio(choices=["vocal", "background"], value="vocal", label="Stem")
+
+        # Options for effects
+        main_gui = gr.Checkbox(False, label="Main")
+        dereverb_gui = gr.Checkbox(False, label="Dereverb", visible=True)
+        vocal_effects_gui = gr.Checkbox(False, label="Vocal Effects", visible=True)
+        background_effects_gui = gr.Checkbox(False, label="Background Effects", visible=False)
+
+        stem_gui.change(
+            lambda stem: (
+                gr.update(visible=stem == "vocal"),
+                gr.update(visible=stem == "vocal"),
+                gr.update(visible=stem == "vocal"),
+                gr.update(visible=stem == "background"),
+            ),
+            [stem_gui],
+            [main_gui, dereverb_gui, vocal_effects_gui, background_effects_gui]
+        )
+
+        # Basic trigger
+        button_base = gr.Button("Inference", variant="primary")
+        output_base = gr.File(label="Result", file_count="multiple", interactive=False)
+
+        button_base.click(
+            fn=sound_separate,
+            inputs=[
+                aud, stem_gui, main_gui, dereverb_gui,
+                vocal_effects_gui, background_effects_gui
+            ],
+            outputs=[output_base]
+        )
+
+    return app