Update services/overlay_service.py

services/overlay_service.py

@@ -1,31 +1,65 @@
 import cv2
+import numpy as np
 import logging
+from typing import List, Dict, Any
 
+# Configure logging
 logging.basicConfig(
     filename="app.log",
     level=logging.INFO,
     format="%(asctime)s - %(levelname)s - %(message)s"
 )
 
-def overlay_boxes(frame, detected_items):
+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.")
+        return frame
+    if not isinstance(detected_items, list):
+        logging.error("Detected items must be a list.")
+        return frame
+
     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
             type_colors = {
-                "crack": (147, 112, 219),    # Light purple (green + pink mix approximation)
+                "crack": (147, 112, 219),    # Light purple (green + pink mix)
                 "pothole": (255, 127, 80),   # Coral (red + orange mix)
-                "signage": (0, 191, 255),
-                "default": (0, 255, 255)
+                "signage": (0, 191, 255),    # DeepSkyBlue
+                "default": (0, 255, 255)     # Yellow
             }
-            color = type_colors.get(item.get("type", ""), type_colors["default"])
+            color = type_colors.get(item.get("type", "default"), type_colors["default"])
 
+            # Draw bounding box and label
             cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
-            cv2.putText(frame, label, (x_min, y_min - 10), 
-                        cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
+            cv2.putText(
+                frame,
+                label,
+                (x_min, y_min - 10),
+                cv2.FONT_HERSHEY_SIMPLEX,
+                0.6,
+                color,
+                2
+            )
 
         logging.info(f"Overlaid {len(detected_items)} items on frame.")
         return frame