From 5e7f4ee88a3c8209ce98931462a730c5fd729480 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Mon, 22 Jul 2024 19:11:55 -0700
Subject: [PATCH] UI: simplify the interface load events

---
 server.py | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/server.py b/server.py
index 2794caa1..57e26be8 100644
--- a/server.py
+++ b/server.py
@@ -146,12 +146,21 @@ def create_interface():
         ui_model_menu.create_event_handlers()
 
         # Interface launch events
-        shared.gradio['interface'].load(None, None, None, js=f"() => {{if ({str(shared.settings['dark_theme']).lower()}) {{ document.getElementsByTagName('body')[0].classList.add('dark'); }} }}")
-        shared.gradio['interface'].load(None, None, None, js=f"() => {{{js}}}")
-        shared.gradio['interface'].load(None, gradio('show_controls'), None, js=f'(x) => {{{ui.show_controls_js}; toggle_controls(x)}}')
         shared.gradio['interface'].load(
-            partial(ui.apply_interface_values, {}, use_persistent=True), None, gradio(ui.list_interface_input_elements()), show_progress=False).then(
-            chat.redraw_html, gradio(ui_chat.reload_arr), gradio('display'), show_progress=False)
+            None,
+            gradio('show_controls'),
+            None,
+            js=f"""(x) => {{
+                if ({str(shared.settings['dark_theme']).lower()}) {{
+                    document.getElementsByTagName('body')[0].classList.add('dark');
+                }}
+                {js}
+                {ui.show_controls_js}
+                toggle_controls(x);
+            }}"""
+        )
+
+        shared.gradio['interface'].load(partial(ui.apply_interface_values, {}, use_persistent=True), None, gradio(ui.list_interface_input_elements()), show_progress=False)
 
         extensions_module.create_extensions_tabs()  # Extensions tabs
         extensions_module.create_extensions_block()  # Extensions block