Update app.py

app.py

@@ -15,9 +15,11 @@ def take_screenshot(url):
         wd.set_window_size(1080, 720)  # Adjust the window size here
         wd.get(url)
         wd.implicitly_wait(10)
-        screenshot = wd.get_screenshot_as_png()
+        page_content = wd.page_source
+
+        return page_content
     except WebDriverException as e:
-        return Image.new('RGB', (1, 1))
+        return "error handle website"
     finally:
         if wd:
             wd.quit()