Update services/overlay_service.py

services/overlay_service.py

@@ -2,17 +2,9 @@ import cv2
 
 def overlay_boxes(frame, boxes):
     for box in boxes:
-        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'])
-
-        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)
-
+        x_min, y_min, x_max, y_max, score, label = box
+        color = (0, 255, 0) if score > 0.5 else (0, 0, 255)
+        cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
+        text = f"{label}: {score:.2f}"
+        cv2.putText(frame, text, (x_min, y_min - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 1)
     return frame