Update app.py

app.py

@@ -14,7 +14,7 @@ def take_screenshot(url):
     wd = None
     try:
         wd = webdriver.Chrome(options=options)
-        wd.set_window_size(2181, 885)
+        wd.set_window_size(1920, 1080)  # Set window size to 1920x1080
         wd.get(url)
         time.sleep(5)
         screenshot = wd.get_screenshot_as_png()
@@ -23,7 +23,18 @@ def take_screenshot(url):
             print("⚠ Screenshot too small, likely error.")
             return None
 
-        return Image.open(BytesIO(screenshot))
+        image = Image.open(BytesIO(screenshot))
+
+        # Center crop to 900x900
+        width, height = image.size
+        crop_size = 900
+        left = (width - crop_size) // 2
+        top = (height - crop_size) // 2
+        right = left + crop_size
+        bottom = top + crop_size
+        cropped_image = image.crop((left, top, right, bottom))
+
+        return cropped_image
 
     except WebDriverException as e:
         print("WebDriverException:", e)
@@ -35,9 +46,9 @@ def take_screenshot(url):
 iface = gr.Interface(
     fn=take_screenshot,
     inputs=gr.Textbox(label="Website URL", value="https://github.com"),
-    outputs=gr.Image(type="pil", label="Screenshot"),
+    outputs=gr.Image(type="pil", label="Cropped Screenshot"),
     title="Website Screenshot",
-    description="Takes a screenshot of the website and displays it here."
+    description="Takes a 1920x1080 screenshot of the website, center-cropped to 900x900."
 )
 
 iface.launch()