Update app.py

app.py

@@ -82,9 +82,9 @@ async def capture_screenshot(image_type: str):
         await browser.close()
         print("Screenshot saved!")
 
-def crop_based_on_bg(img: Image, bg_color=(59, 59, 59)):
-    # img = Image.open(image_path).convert("RGB")
-    # pixels = img.load()
+def crop_based_on_bg(image_path: str, bg_color=(59, 59, 59)):
+    img = Image.open(image_path).convert("RGB")
+    pixels = img.load()
 
     # Define fixed crop for top header
     top_crop = 50  
@@ -127,13 +127,13 @@ def grab_image(image_type: str) -> Image:
     loop.run_until_complete(capture_screenshot(image_type))
 
     print("Loading image for Gradio...")
-    img = Image.open("img.png").RGB
-    header_height = 50  
-    left_crop = 0  # Adjust to remove background
-    right_crop = img.width - 0 
+
+    cropped_img = crop_based_on_bg("img.png", bg_color=(59, 59, 59))
+    #img = Image.open("img.png")
+    #header_height = 50  
 
     #cropped_img = img.crop((left_crop, header_height, right_crop, img.height))
-    cropped_img = crop_based_on_bg(img, left_crop, header_height, right_crop, img.height)
+    #cropped_img = crop_based_on_bg(img, left_crop, header_height, right_crop, img.height)
     return cropped_img
 
 @tool