Update app.py

try to remove background again

app.py

@@ -8,6 +8,7 @@ import nest_asyncio
 #import matplotlib
 from playwright.async_api import async_playwright
 from PIL import Image
+import numpy as np
 import subprocess
 import json
 from rapidfuzz import process
@@ -84,25 +85,24 @@ async def capture_screenshot(image_type: str):
 
 def crop_based_on_bg(image_path: str, bg_color=(59, 59, 59)):
     img = Image.open(image_path).convert("RGB")
-    pixels = img.load()
+    img_array = np.array(img)
 
     # Define fixed crop for top header
     top_crop = 50  
-    width, height = img.size
-
-    # Find leftmost and rightmost non-background pixels
-    left_crop, right_crop = 0, width
-    for x in range(width):
-        column = [pixels[x, y] for y in range(top_crop, height)]
-        if any(pixel != bg_color for pixel in column):
-            left_crop = x
-            break
-
-    for x in range(width - 1, -1, -1):
-        column = [pixels[x, y] for y in range(top_crop, height)]
-        if any(pixel != bg_color for pixel in column):
-            right_crop = x
-            break
+    #width, height = img.size
+
+    # Get height & width
+    height, width, _ = img_array.shape
+
+    # Extract the portion of the image below the fixed header
+    img_no_header = img_array[top_crop:, :, :]
+
+    # Find columns that are NOT the background color
+    mask = np.any(img_no_header != bg_color, axis=(0, 2))  # True for non-bg pixels
+
+    # Find leftmost and rightmost non-bg columns
+    left_crop = np.argmax(mask)  # First non-bg column
+    right_crop = width - np.argmax(mask[::-1])  # Last non-bg column
 
     # Crop and save
     cropped_img = img.crop((left_crop, top_crop, right_crop, height))