Update services/overlay_service.py

services/overlay_service.py

@@ -1,21 +1,10 @@
 import cv2
 
-def overlay_boxes(image_path, boxes):
-    image = cv2.imread(image_path)
-    if image is None:
-        return None
-    
+def overlay_boxes(image, boxes):
+    # Directly use the image (NumPy array)
     for box in boxes:
-        x1, y1, x2, y2, label = box
-        color = (0, 255, 0) if label == "normal" else (0, 0, 255)
-        cv2.rectangle(image, (x1, y1), (x2, y2), color, 2)
-        cv2.putText(
-            image, 
-            label, 
-            (x1, y1 - 10), 
-            cv2.FONT_HERSHEY_SIMPLEX, 
-            0.5, 
-            color, 
-            2
-        )
+        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