Update app.py

app.py

@@ -54,11 +54,10 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             #print (driver.html)
             driver.implicitly_wait(30)
             driver.set_window_size(int(width), int(height))
-            page = driver.find_element(By.TAG_NAME, "html")
-            print (dir(page))
+            page = driver.find_element(By.TAG_NAME, "html").setAttribute('theme','gab-black')
+            #print (dir(page))
             print (page.get_attribute('theme'))
-            page.get_attribute('theme')='gab-black'
-            driver.execute_script("innerHTML='gab-dark';",page)
+            #driver.execute_script("innerHTML='gab-dark';",page)
             main_head = driver.find_element(By.CLASS_NAME, "main-content-header")
             head_shot = main_head.screenshot(f'head-{uid}-tmp.png')