Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import time
 import random
 import gradio as gr
 from selenium import webdriver
@@ -30,16 +31,13 @@ def main(url):
         driver.get(url)
     else:
         driver.get("http://" + url)
-
-    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")
     
     driver.set_window_size(page_width, page_height)
     
+    time.sleep(3)
     filename = str(random.randint(111111111, 999999999)) + ".png"
     file_path = os.path.join(os.path.dirname(__file__), filename)
     driver.save_screenshot(file_path)