Update app.py

app.py

@@ -44,6 +44,7 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
     if is_url:
         options = webdriver.ChromeOptions()
         options.add_argument('--headless')
+        options.add_argument('--force-dark-mode')
         options.add_argument('--no-sandbox')
         options.add_argument('--disable-dev-shm-usage')
         mes='<center>operation success'
@@ -56,9 +57,9 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             driver.set_window_size(int(width), int(height))
             page = driver.find_element(By.TAG_NAME, "html")
             print (dir(page))
-            page.send_keys('theme','gab-black')
+            #page.send_keys('theme','gab-black')
             print (page.get_attribute('theme'))
-            driver.execute_script("arguments[0].theme='gab-black';",page)
+            driver.execute_script("arguments[0].theme=gab-black;",page)
             main_head = driver.find_element(By.CLASS_NAME, "main-content-header")
             head_shot = main_head.screenshot(f'head-{uid}-tmp.png')