Update app.py

try to remove background again

app.py

@@ -83,7 +83,7 @@ async def capture_screenshot(image_type: str):
         await browser.close()
         print("Screenshot saved!")
 
-def crop_based_on_bg(image_path: str, bg_color=(59, 59, 59)):
+def crop_based_on_bg(image_path: str, bg_color=(59, 59, 59), tolerance=10):
     img = Image.open(image_path).convert("RGB")
     img_array = np.array(img)
 
@@ -97,12 +97,15 @@ def crop_based_on_bg(image_path: str, bg_color=(59, 59, 59)):
     # 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
+    # Compute a mask for pixels that are NOT within the background tolerance
+    lower_bound = np.array(bg_color) - tolerance
+    upper_bound = np.array(bg_color) + tolerance
+    mask = np.any((img_no_header < lower_bound) | (img_no_header > upper_bound), axis=2)
 
-    # 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
+    # Find leftmost and rightmost non-background pixels
+    col_sums = mask.sum(axis=0)
+    left_crop = np.argmax(col_sums > 0)  # First column with content
+    right_crop = width - np.argmax(col_sums[::-1] > 0)  # Last column with content
 
     # Crop and save
     cropped_img = img.crop((left_crop, top_crop, right_crop, height))