Update app.py

app.py

@@ -28,9 +28,10 @@ def run_script(url: str, height: int, width: int):
             #html=driver.page_source
             driver.implicitly_wait(30)
             driver.set_window_size(int(width), int(height))
-            #driver.find_element(By.
+            page = driver.find_element(By.TAG_NAME, "html")
+            driver.execute_script("arguments[0].theme = 'gab-black';",page)
+            
             obj = driver.find_element(By.CLASS_NAME, "main")            
-            driver.execute_script("arguments[0].theme = 'gab-black';",obj)
             #obj = driver.find_element(By.ID, "app-container")            
             #obj = driver.find_element(By.ID, "conversation-feed")            
             #obj = driver.find_element_by_id("in_html")