Update app.py

app.py

@@ -3,6 +3,7 @@ from selenium import webdriver
 from selenium.common.exceptions import WebDriverException
 from PIL import Image
 from io import BytesIO
+import time  # <-- import time
 
 def take_screenshot(url):
     options = webdriver.ChromeOptions()
@@ -15,6 +16,7 @@ def take_screenshot(url):
         wd.set_window_size(1080, 720)  # Adjust the window size here
         wd.get(url)
         wd.implicitly_wait(10)
+        time.sleep(2)  # <-- Add delay before screenshot
         screenshot = wd.get_screenshot_as_png()
     except WebDriverException as e:
         return Image.new('RGB', (1, 1))
@@ -27,9 +29,9 @@ def take_screenshot(url):
 iface = gr.Interface(
     fn=take_screenshot,
     inputs=gr.Textbox(label="Website URL", value="https://github.com/aiclubiitropar"),
-    outputs=gr.Image(type="pil", label="Screenshot", height=360, width=540),  # Adjust the image size here
+    outputs=gr.Image(type="pil", label="Screenshot", height=360, width=540),
     title="Website Screenshot",
     description="Take a screenshot of a website."
 )
 
-iface.launch()
+iface.launch()