Update app.py

app.py

@@ -25,11 +25,13 @@ def run_script(url: str, height: int, width: int):
         try:
             driver = webdriver.Chrome(options=options)
             driver.get(url)
-            #html=driver.page_source
+            html=driver.page_source
+            print (driver.html)
             driver.implicitly_wait(30)
             driver.set_window_size(int(width), int(height))
             page = driver.find_element(By.TAG_NAME, "html")
-            driver.execute_script("arguments[0].theme = 'gab-black';",page)
+            driver.execute_script("html.theme = 'gab-black';",html)
+            #driver.execute_script("arguments[0].theme = 'gab-black';",page)
             
             obj = driver.find_element(By.CLASS_NAME, "main")            
             #obj = driver.find_element(By.ID, "app-container")