Update services/overlay_service.py

services/overlay_service.py

@@ -1,10 +1,16 @@
 import cv2
 
-def overlay_boxes(image, boxes):
-    # Directly use the image (NumPy array)
+def overlay_boxes(frame, boxes):
     for box in boxes:
-        x, y, w, h = box
-        x1, y1, x2, y2 = int(x), int(y), int(x + w), int(y + h)
-        cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
-        cv2.putText(image, 'Anomaly', (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
-    return image
+        x1, y1, x2, y2 = map(int, box['box'])
+        label = box.get('label', 'Anomaly')
+        confidence = box.get('score', 0)
+
+        # Draw box
+        cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
+
+        # Draw label
+        text = f"{label} ({confidence:.2f})"
+        cv2.putText(frame, text, (x1, max(y1 - 10, 0)), cv2.FONT_HERSHEY_SIMPLEX,
+                    0.5, (255, 0, 0), 2)
+    return frame