Update services/overlay_service.py

services/overlay_service.py

@@ -2,15 +2,17 @@ import cv2
 
 def overlay_boxes(frame, boxes):
     for box in boxes:
-        x1, y1, x2, y2 = map(int, box['box'])
-        label = box.get('label', 'Anomaly')
-        confidence = box.get('score', 0)
+        score = box['score']
+        label = box['label']
+        bbox = box['box']  # x, y, width, height
+        
+        x1 = int(bbox['xmin'])
+        y1 = int(bbox['ymin'])
+        x2 = int(bbox['xmax'])
+        y2 = int(bbox['ymax'])
 
-        # Draw box
         cv2.rectangle(frame, (x1, y1), (x2, y2), (0, 255, 0), 2)
+        cv2.putText(frame, f"{label}: {score:.2f}", (x1, y1 - 10),
+                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, (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