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();
+}
+
 """