Updating js_func

app.py

@@ -105,7 +105,19 @@ log files. I use gradio for `app` and map that to the open port of huggingface s
 The only communication between `app` and `main` is the log file.
 """
 
-with gr.Blocks() as ui:
+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;
+    }
+}
+"""
+
+
+with gr.Blocks(js=js_func) as ui:
     with gr.Tab("Application"):
         gr.Markdown(intro_md)
         gr.Image(str(proj_dir / 'media' / 'reddit_scraper.drawio.png'), type='filepath')
@@ -115,13 +127,6 @@ with gr.Blocks() as ui:
         gr.Markdown(how_to_md)
     with gr.Tab("How does it work?"):
         gr.Markdown(how_does_it_work_md)
-    ui.load(None,
-            js="""
-    () => {
-        document.body.classList.toggle('dark');
-        document.querySelector('gradio-app').style.backgroundColor = 'var(--color-background-primary)'
-    }
-    """, )
 
 app = WebhooksServer(ui=ui.queue(), webhook_secret=WEBHOOK_SECRET)