Update services/overlay_service.py

services/overlay_service.py

@@ -11,16 +11,6 @@ logging.basicConfig(
 )
 
 def overlay_boxes(frame: np.ndarray, detected_items: List[Dict[str, Any]]) -> np.ndarray:
-    """
-    Overlay bounding boxes and labels on the frame for detected items.
-
-    Args:
-        frame (np.ndarray): Input frame in BGR format.
-        detected_items (List[Dict[str, Any]]): List of detected items with coordinates and labels.
-
-    Returns:
-        np.ndarray: Frame with overlaid bounding boxes and labels.
-    """
     # Validate inputs
     if not isinstance(frame, np.ndarray) or frame.size == 0:
         logging.error("Invalid input frame provided to overlay_service.")
@@ -31,21 +21,19 @@ def overlay_boxes(frame: np.ndarray, detected_items: List[Dict[str, Any]]) -> np
 
     try:
         for item in detected_items:
-            # Check for required keys
             if "coordinates" not in item or "label" not in item:
                 logging.warning(f"Skipping item without coordinates or label: {item}")
                 continue
 
-            # Extract coordinates and label
             x_min, y_min, x_max, y_max = item["coordinates"]
             label = item["label"]
 
-            # Define colors based on detection type
+            # Define colors for Operations Maintenance
             type_colors = {
-                "crack": (147, 112, 219),    # Light Purple (green + pink mix) for cracks
-                "pothole": (255, 127, 80),   # Coral (red + orange mix) for potholes
-                "signage": (0, 191, 255),    # DeepSkyBlue for signage
-                "default": (0, 255, 255)     # Yellow as fallback
+                "crack": (0, 0, 255),    # Blue for cracks
+                "pothole": (255, 0, 0),  # Red for potholes
+                "signage": (0, 0, 255),  # Blue for signage
+                "default": (0, 255, 255) # Yellow as fallback
             }
             color = type_colors.get(item.get("type", "default"), type_colors["default"])