Update app.py

app.py

@@ -11,26 +11,16 @@ def take_screenshot(url):
     options.add_argument('--headless')
     options.add_argument('--no-sandbox')
     options.add_argument('--disable-dev-shm-usage')
+    wd = webdriver.Chrome(options=options)
+    wd.get(url)
+    wd.implicitly_wait(10)
+    resp=driver.find_element(By.TAG_NAME, "a")
+    return resp
 
-    try:
-        wd = webdriver.Chrome(options=options)
-        wd.set_window_size(1080, 720)  # Adjust the window size here
-        wd.get(url)
-        wd.implicitly_wait(5)
-        resp=driver.find_element(By.TAG_NAME, "a")
-        return resp
-        #screenshot = wd.get_screenshot_as_png()
-    except WebDriverException as e:
-        return resp
-    finally:
-        if wd:
-            wd.quit()
-
-    return Image.open(BytesIO(screenshot))
 
 iface = gr.Interface(
     fn=take_screenshot,
-    inputs=gr.Textbox(label="Website URL", value="https://kargaranamir.github.io"),
+    inputs=gr.Textbox(label="Website URL", value="https://spiegel.de"),
     outputs=gr.Textbox(),
     title="Website Screenshot",
     description="Take a screenshot of a website."