Update app.py

app.py

@@ -57,7 +57,18 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             page = driver.find_element(By.TAG_NAME, "html")
             print (dir(page))
             print (page.text)
-            driver.execute_script("arguments[0].style.background = 'blue';",page)
+
+            js="""
+            function toggleTheme() {
+               if (localStorage.getItem('theme') === 'theme-dark'){
+                   setTheme('theme-light');
+               } else {
+                   setTheme('theme-dark');
+               }
+            }
+            """
+            
+            driver.execute_script(js)
             main_head = driver.find_element(By.CLASS_NAME, "main-content-header")
             head_shot = main_head.screenshot(f'head-{uid}-tmp.png')