Update frontend/webui/ui.py

frontend/webui/ui.py

@@ -15,11 +15,8 @@ app_settings = get_settings()
 model_list = ["model-a", "model-b", "model-c"]
 
 def update_model_array(text_input):
-    # Split lines, remove empty strings, strip whitespace
     updated = [line.strip() for line in text_input.split("\n") if line.strip()]
-    return gr.Dropdown.update(choices=updated, value=updated[0] if updated else None), "\n".join(updated)
-
-
+    return gr.update(choices=updated, value=updated[0] if updated else None), "\n".join(updated)
 
 
 def _get_footer_message() -> str:
@@ -46,22 +43,27 @@ def get_web_ui() -> gr.Blocks:
         title="FastSD CPU",
     ) as fastsd_web_ui:
         gr.HTML("<center><H1>FastSD CPU</H1></center>")
-        current_mode = "LCM"
-        if app_settings.settings.lcm_diffusion_setting.use_openvino:
-            current_mode = "LCM-OpenVINO"
-        elif app_settings.settings.lcm_diffusion_setting.use_lcm_lora:
-            current_mode = "LCM-LoRA"
-        model_dropdown = gr.Dropdown(choices=model_list, label="Select Model", value=model_list[0])
-        model_text_area = gr.Textbox(value="\n".join(model_list), lines=10, label="Edit Model List")
-        update_button = gr.Button("Update Dropdown")
-        update_button.click(            fn=update_model_array,            inputs=model_text_area,            outputs=[model_dropdown, model_text_area],        )
-        mode = gr.Radio(
-            ["LCM", "LCM-LoRA", "LCM-OpenVINO"],
-            label="Mode",
-            info="Current working mode",
-            value=current_mode,
-        )
-        mode.change(change_mode, inputs=mode)
+        gr.Column(scale=1):
+            current_mode = "LCM"
+            if app_settings.settings.lcm_diffusion_setting.use_openvino:
+                current_mode = "LCM-OpenVINO"
+            elif app_settings.settings.lcm_diffusion_setting.use_lcm_lora:
+                current_mode = "LCM-LoRA"
+        gr.Column(scale=1):
+            gr.Row():
+                model_text_area2 = gr.Textbox(value=" ", lines=10, label="Enter a new model")
+                update_button = gr.Button("Update Dropdown")
+                update_button.click(            fn=update_model_array,            inputs=model_text_area2,            outputs=[model_dropdown, model_text_area],        )
+            gr.Row():                    
+                model_dropdown = gr.Dropdown(choices=model_list, label="Select Model", value=model_list[0])
+                model_text_area = gr.Textbox(value="\n".join(model_list), lines=10, label="Edit Model List")
+                mode = gr.Radio(
+                ["LCM", "LCM-LoRA", "LCM-OpenVINO"],
+                label="Mode",
+                info="Current working mode",
+                value=current_mode,
+            )
+            mode.change(change_mode, inputs=mode)
 
         with gr.Tabs():
             with gr.TabItem("Text to Image"):