Update services/overlay_service.py

services/overlay_service.py

@@ -11,6 +11,16 @@ 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.")
@@ -21,19 +31,21 @@ 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 for Operations Maintenance
+            # Define colors based on detection type
             type_colors = {
                 "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
+                "default": (0, 255, 255) # Yellow
             }
             color = type_colors.get(item.get("type", "default"), type_colors["default"])