Update app.py

app.py

@@ -30,6 +30,11 @@ except Exception:
 MAX_SIDE_PX = 80  # set >0 (e.g., 70) to filter detections with large side; -1 disables
 SEG_DEFAULT_ALPHA = 0.45
 
+# High-contrast colors for green backgrounds (BGR order)
+BERRIES_COLOR_BGR = (255, 0, 255)         # magenta/pink for detection boxes
+BUNCHES_FILL_COLOR_BGR = (255, 255, 0)    # cyan for mask fill
+BUNCHES_CONTOUR_COLOR_BGR = (255, 255, 255)  # white for mask contours
+
 # Fixed weights (no UI controls). If you want them editable, add Textbox components and wire them as inputs.
 WEIGHTS_DETECTION = "weights/berries.pt"
 WEIGHTS_SEGMENTATION = "weights/bunches.pt"
@@ -220,8 +225,7 @@ def _draw_boxes_overlay(image_rgb: np.ndarray, sahi_result, target_class: str, u
         cls = getattr(item.category, "name", "unknown")
         is_target = (cls == target_class) if use_target else False
 
-        # Colors in BGR for OpenCV, convert later when compositing
-        color_bgr = (0, 0, 255) if is_target and use_target else (0, 200, 0)
+        color_bgr = BERRIES_COLOR_BGR
 
         # Draw on overlay (BGR)
         cv2.rectangle(overlay, (x1, y1), (x2, y2), color_bgr, 2)
@@ -288,17 +292,17 @@ def _draw_seg_overlay(image_rgb: np.ndarray, yolo_result, target_class: str, use
                 if m.shape[:2] != (H, W):
                     m = cv2.resize(m, (W, H), interpolation=cv2.INTER_NEAREST)
 
-                overlay_bgr[m == 1] = color_bgr
+                overlay_bgr[m == 1] = BUNCHES_FILL_COLOR_BGR
                 alpha[m == 1] = np.maximum(alpha[m == 1], fa255)
 
                 cnts, _ = cv2.findContours(m, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
-                cv2.drawContours(overlay_bgr, cnts, -1, color_bgr, 2)
+                cv2.drawContours(overlay_bgr, cnts, -1, BUNCHES_CONTOUR_COLOR_BGR, 2)
                 cv2.drawContours(alpha, cnts, -1, 255, 2)
             except Exception:
                 try:
                     xyxy = boxes.xyxy[i].detach().cpu().numpy().astype(int)
                     x1, y1, x2, y2 = map(int, xyxy)
-                    cv2.rectangle(overlay_bgr, (x1, y1), (x2, y2), color_bgr, 2)
+                    cv2.rectangle(overlay_bgr, (x1, y1), (x2, y2), BUNCHES_CONTOUR_COLOR_BGR, 2)
                     cv2.rectangle(alpha, (x1, y1), (x2, y2), 255, 2)
                 except Exception:
                     pass
@@ -306,7 +310,7 @@ def _draw_seg_overlay(image_rgb: np.ndarray, yolo_result, target_class: str, use
             try:
                 xyxy = boxes.xyxy[i].detach().cpu().numpy().astype(int)
                 x1, y1, x2, y2 = map(int, xyxy)
-                cv2.rectangle(overlay_bgr, (x1, y1), (x2, y2), color_bgr, 2)
+                cv2.rectangle(overlay_bgr, (x1, y1), (x2, y2), BUNCHES_CONTOUR_COLOR_BGR, 2)
                 cv2.rectangle(alpha, (x1, y1), (x2, y2), 255, 2)
             except Exception:
                 pass