Update app.py

app.py

@@ -34,14 +34,11 @@ def check_text(prompt, negative=""):
 
 js_func = """
 
-function setThemeOnLoad() {
-    const url = new URL(window.location.href);
-    const currentTheme = url.searchParams.get('__theme');
-
-    if (currentTheme === 'dark') {
-        applyDarkTheme(); 
-    } else {
-        applyLightTheme(); 
+function refresh() {
+    const url = new URL(window.location);
+    if (url.searchParams.get('__theme') !== 'dark') {
+        url.searchParams.set('__theme', 'dark');
+        window.location.href = url.href;
     }
 }