Update app.py

app.py

@@ -52,6 +52,8 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             driver.get(url)
             html=driver.page_source
             #print (driver.html)
+            driver.execute_script("arguments[0].theme='gab-black';")
+            
             driver.implicitly_wait(30)
             driver.set_window_size(int(width), int(height))
             page = driver.find_element(By.TAG_NAME, "html")
@@ -60,7 +62,7 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             dk=page.get_attribute('theme')
             print (dir(dk))
             print (page.get_attribute('theme'))
-            driver.execute_script("return arguments[0].theme='gab-black';",page)
+            #driver.execute_script("arguments[0].theme='gab-black';",page)
             print (page.get_attribute('theme'))
             print(dir(driver))
             #driver.get(driver.getCurrentUrl())