Update services/overlay_service.py

services/overlay_service.py

@@ -1,16 +1,22 @@
+# services/overlay_service.py
 import cv2
 
-def overlay_boxes(frame, items):
-    for item in items:
-        box = item['box']
-        x_min, y_min, x_max, y_max = map(int, box)
-        severity = item['severity']
-        if item['type'] == 'crack':
-            color = (0, 0, 255) if severity == 'Severe' else (0, 255, 255) if severity == 'Moderate' else (0, 255, 0) if severity == 'Minor' else (255, 165, 0)  # Orange for Underlying
-            label = f"Crack: {severity}"
-        else:  # Hole
-            color = (255, 0, 0) if severity == 'Severe' else (255, 255, 0) if severity == 'Moderate' else (0, 128, 0)
-            label = f"Hole: {severity}"
+def add_overlay(frame, detections):
+    """
+    Add overlays (bounding boxes and labels) to a frame.
+    Args:
+        frame: Input frame (numpy array)
+        detections: List of detection dictionaries
+    Returns:
+        numpy array: Frame with overlays
+    """
+    for detection in detections:
+        if "coordinates" not in detection or "label" not in detection:
+            continue
+        x_min, y_min, x_max, y_max = detection["coordinates"]
+        label = detection["label"]
+        color = (0, 255, 0)  # Green for generic overlays
         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.putText(frame, label, (x_min, y_min - 10), 
+                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
     return frame