Update app.py

app.py

@@ -15,12 +15,12 @@ def take_screenshot(url):
         wd = webdriver.Chrome(options=options)
         wd.set_window_size(1080, 720)  # Adjust the window size here
         wd.get(url)
-        wd.implicitly_wait(10)
+        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 Image.new('RGB', (1, 1))
+        return resp
     finally:
         if wd:
             wd.quit()