Update app.py

app.py

@@ -30,6 +30,12 @@ def main(url):
         driver.get(url)
     else:
         driver.get("http://" + url)
+
+    driver.execute_cdp_cmd("Page.setLifecycleEventsEnabled", {"enabled": True})
+    driver.execute_cdp_cmd("Page.setPageLoadStrategy", {"strategy": "eager"})
+    WebDriverWait(driver, 30).until(
+        lambda d: d.execute_script("return document.readyState") == "complete"
+    )
     
     page_width = driver.execute_script("return document.body.scrollWidth")
     page_height = driver.execute_script("return document.body.scrollHeight")