Update services/overlay_service.py

services/overlay_service.py

@@ -1,31 +1,45 @@
 import cv2
+import logging
+
+# Setup logging
+logging.basicConfig(
+    filename="app.log",
+    level=logging.INFO,
+    format="%(asctime)s - %(levelname)s - %(message)s"
+)
 
 def overlay_boxes(frame, detected_items):
     """
-    Add overlays (bounding boxes and labels) to a frame.
+    Overlay bounding boxes and numbered labels on the frame for detected items.
     Args:
         frame: Input frame (numpy array)
-        detected_items: List of detection dictionaries
+        detected_items: List of detected items (dicts with type, label, coordinates, etc.)
     Returns:
         numpy array: Frame with overlays
     """
-    for item in detected_items:
-        if "coordinates" not in item or "label" not in item:
-            continue
-        x_min, y_min, x_max, y_max = item["coordinates"]
-        label = item["label"]
-        # Choose color based on type
-        if item["type"] == "crack":
-            color = (255, 0, 0)  # Red for cracks
-        elif item["type"] == "pothole":
-            color = (255, 165, 0)  # Orange for potholes
-        else:
-            color = (0, 255, 0)  # Green for other objects
+    try:
+        for item in detected_items:
+            if "coordinates" not in item or "label" not in item:
+                continue
+            x_min, y_min, x_max, y_max = item["coordinates"]
+            label = item["label"]
+
+            # Determine color based on type
+            type_colors = {
+                "crack": (255, 0, 0),        # Red for cracks
+                "pothole": (255, 165, 0),    # Orange for potholes
+                "signage": (0, 191, 255),    # DeepSkyBlue for signage
+                "default": (0, 255, 255)     # Yellow for others
+            }
+            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.5, color, 2)
 
-        cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
-        # Add severity if present
-        if "severity" in item:
-            label += f" - Severity: {item['severity'].capitalize()}"
-        cv2.putText(frame, label, (x_min, y_min - 10), 
-                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
-    return frame
+        logging.info(f"Overlaid {len(detected_items)} items on frame.")
+        return frame
+    except Exception as e:
+        logging.error(f"Error overlaying boxes: {str(e)}")
+        return frame