Update detect_crop_image.py

detect_crop_image.py

@@ -91,12 +91,62 @@ def detect_and_crop_image(image_path, output_image_path=None):
     if zoom_inset > 0:
         print(f"Smart Zoom applied: {zoom_inset}px inset to clear rounded corners.")
 
-    # Inset Logic (2px) - additional fixed safety margin
+    # --- Validate Crops ---
+    # Only crop if the excluded region is genuinely a white/black background
+    prop_x_min = x
+    prop_y_min = y
+    prop_x_max = x + w
+    prop_y_max = y + h
+
+    def validate_crop(region, border_region, edge_thresh=0.80, region_thresh=0.60):
+        if region.size == 0 or border_region.size == 0:
+            return False
+            
+        dark_edge = np.count_nonzero(border_region < 20) / border_region.size
+        light_edge = np.count_nonzero(border_region > 235) / border_region.size
+        
+        dark_region = np.count_nonzero(region < 20) / region.size
+        light_region = np.count_nonzero(region > 235) / region.size
+        
+        is_dark_bg = (dark_edge >= edge_thresh) and (dark_region >= region_thresh)
+        is_light_bg = (light_edge >= edge_thresh) and (light_region >= region_thresh)
+        
+        return is_dark_bg or is_light_bg
+
+    # Validate Top Crop
+    if prop_y_min > 0:
+        top_region = gray[0:prop_y_min, :]
+        top_border = gray[0:min(3, prop_y_min), :]
+        if not validate_crop(top_region, top_border):
+            prop_y_min = 0
+
+    # Validate Bottom Crop
+    if prop_y_max < img_h:
+        bottom_region = gray[prop_y_max:img_h, :]
+        bottom_border = gray[max(img_h-3, prop_y_max):img_h, :]
+        if not validate_crop(bottom_region, bottom_border):
+            prop_y_max = img_h
+
+    # Validate Left Crop
+    if prop_x_min > 0:
+        left_region = gray[:, 0:prop_x_min]
+        left_border = gray[:, 0:min(3, prop_x_min)]
+        if not validate_crop(left_region, left_border):
+            prop_x_min = 0
+
+    # Validate Right Crop
+    if prop_x_max < img_w:
+        right_region = gray[:, prop_x_max:img_w]
+        right_border = gray[:, max(img_w-3, prop_x_max):img_w]
+        if not validate_crop(right_region, right_border):
+            prop_x_max = img_w
+
+    # Inset Logic (2px) - additional fixed safety margin ONLY for valid crops
     inset = 2
-    x_min = max(x + inset, 0)
-    y_min = max(y + inset, 0)
-    x_max = min(x + w - inset, img_w)
-    y_max = min(y + h - inset, img_h)
+    x_min = prop_x_min + inset if prop_x_min > 0 else 0
+    y_min = prop_y_min + inset if prop_y_min > 0 else 0
+    x_max = prop_x_max - inset if prop_x_max < img_w else img_w
+    y_max = prop_y_max - inset if prop_y_max < img_h else img_h
     
     final_w = x_max - x_min
     final_h = y_max - y_min