Update app.py

app.py

@@ -361,12 +361,16 @@ def generate_function_call(prompt):
 # theme = "gary109/HaleyCH_Theme"
 # theme = "derekzen/stardust"
 # theme = "Taithrah/Minimal"
-_js="""
-        () => {
-            document.body.classList.toggle('dark');
-            document.querySelector('gradio-app').style.backgroundColor = 'var(--color-background-primary)'
-        }
-        """
+js_func = """
+function refresh() {
+    const url = new URL(window.location);
+
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
+    }
+}
+"""
 theme = "rottenlittlecreature/Moon_Goblin"
 # theme = "xiaobaiyuan/theme_brief"
 # theme = "gary109/Emerald_Isle"
@@ -399,7 +403,7 @@ demo = gr.ChatInterface(
     chatbot=my_chatbot,
     title="Investor Assistant 🦹‍♂️",
     theme=theme,
-    css = _js
+    js= js_func
 )
 
 demo.queue().launch(show_api=False)