Update app.py

app.py

@@ -33,13 +33,37 @@ def check_text(prompt, negative=""):
 
 
 js_func = """
-function refresh() {
-    const url = new URL(window.location);
-    if (url.searchParams.get('__theme') !== 'dark') {
+
+function toggleTheme() {
+    const url = new URL(window.location.href);
+    const currentTheme = url.searchParams.get('__theme');
+
+    if (currentTheme !== 'dark') {
         url.searchParams.set('__theme', 'dark');
-        window.location.href = url.href;
+    } else {
+        url.searchParams.delete('__theme');
     }
+
+    history.pushState({}, '', url.href);
+    
+    applyTheme();
 }
+
+function applyTheme() {
+    const url = new URL(window.location.href);
+    const currentTheme = url.searchParams.get('__theme');
+    
+    if (currentTheme === 'dark') {
+        document.body.classList.add('dark-theme');
+    } else {
+        document.body.classList.remove('dark-theme');
+    }
+}
+
+window.onload = function() {
+    applyTheme();
+}
+
 """