Update gui.py

gui.py

@@ -1,25 +1,13 @@
 import gradio as gr
 import os
-import json
 import logging
-from helpers import update_model_dropdown, handle_file_upload, clear_old_output
+from assets.i18n.i18n import I18nAuto
+from helpers import update_model_dropdown
 from model import MODEL_CONFIGS
 from processing import process_audio
-from assets.i18n.i18n import I18nAuto
-from config_manager import load_config
 
 logging.basicConfig(filename='sesa_gui.log', level=logging.DEBUG)
 
-# BASE_DIR and config setup
-BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-CONFIG_DIR = os.path.join(BASE_DIR, "assets")
-CONFIG_FILE = os.path.join(CONFIG_DIR, "config.json")
-
-# Load user config
-user_config = load_config()
-initial_settings = user_config["settings"]
-initial_favorites = user_config["favorites"]
-
 # Initialize I18nAuto
 i18n = I18nAuto()
 
@@ -32,69 +20,36 @@ def create_interface():
     """
 
     with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-        # Store state in hidden Textbox
-        favorites_state = gr.Textbox(value=json.dumps(initial_favorites), visible=False)
-
         gr.HTML(f"<h1>{i18n('SESA Audio Separation')}</h1>")
 
-        with gr.Tab(i18n("audio_separation_tab")):
-            with gr.Row():
-                with gr.Column():
-                    input_audio_file = gr.File(
-                        file_types=[".wav", ".mp3", ".flac"],
-                        label=i18n("upload_file")
-                    )
-                    model_category = gr.Dropdown(
-                        label=i18n("category"),
-                        choices=[i18n(cat) for cat in MODEL_CONFIGS.keys()],
-                        value=i18n(initial_settings["model_category"])
-                    )
-                    model_dropdown = gr.Dropdown(
-                        label=i18n("model"),
-                        choices=update_model_dropdown(i18n(initial_settings["model_category"]), favorites=initial_favorites)["choices"],
-                        value=initial_settings["selected_model"]
-                    )
-                    export_format = gr.Dropdown(
-                        label=i18n("format"),
-                        choices=['wav FLOAT', 'flac PCM_16'],
-                        value=initial_settings["export_format"]
-                    )
-                    process_btn = gr.Button(i18n("process"))
-                    clear_old_output_btn = gr.Button(i18n("reset"))
-
-                with gr.Column():
-                    original_audio = gr.Audio(label=i18n("original"))
-                    vocals_audio = gr.Audio(label=i18n("vocals"))
-                    instrumental_audio = gr.Audio(label=i18n("instrumental_output"))
-                    status = gr.Textbox(label=i18n("status"), interactive=False)
+        with gr.Row():
+            with gr.Column():
+                input_audio_file = gr.File(
+                    file_types=[".wav", ".mp3", ".flac"],
+                    label=i18n("upload_file")
+                )
+                model_dropdown = gr.Dropdown(
+                    label=i18n("model"),
+                    choices=update_model_dropdown("Vocal Models")["choices"],
+                    value=None
+                )
+                process_btn = gr.Button(i18n("process"))
+
+            with gr.Column():
+                output_audio = gr.Audio(label=i18n("output"))
+                status = gr.Textbox(label=i18n("status"), interactive=False)
 
         # Event handlers
-        def update_models(category):
-            logging.debug(f"Updating models for category: {category}")
-            choices = update_model_dropdown(category, favorites=json.loads(favorites_state.value))["choices"]
-            return gr.update(choices=choices)
-
-        model_category.change(
-            fn=update_models,
-            inputs=model_category,
-            outputs=model_dropdown
-        )
-
-        input_audio_file.upload(
-            fn=lambda x: handle_file_upload(x, None, is_auto_ensemble=False),
-            inputs=[input_audio_file],
-            outputs=[input_audio_file, original_audio]
-        )
-
         process_btn.click(
             fn=process_audio,
-            inputs=[input_audio_file, model_dropdown, export_format],
-            outputs=[vocals_audio, instrumental_audio, status]
+            inputs=[input_audio_file, model_dropdown],
+            outputs=[output_audio, status]
         )
 
-        clear_old_output_btn.click(
-            fn=clear_old_output,
-            outputs=status
+        input_audio_file.upload(
+            fn=lambda x: (x, x),
+            inputs=input_audio_file,
+            outputs=[input_audio_file, output_audio]
         )
 
         return demo