Update app.py

app.py

@@ -44,7 +44,6 @@ 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'
@@ -65,6 +64,8 @@ def run_script(url: str, height: int, width: int, check_b,check_h):
             print (page.get_attribute('theme'))
             
             main_head = driver.find_element(By.CLASS_NAME, "main-content-header")
+            driver.execute_script("arguments[0].theme='gab-black';",main_head)
+            
             head_shot = main_head.screenshot(f'head-{uid}-tmp.png')
             
             obj = driver.find_element(By.CLASS_NAME, "main")