Update services/overlay_service.py

services/overlay_service.py

@@ -4,13 +4,13 @@ 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':
-            severity = item['severity']
-            color = (255, 0, 0) if severity == 'Large' else (255, 165, 0)  # Red for Large, Orange for Small
+            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:
-            color = (255, 105, 180)  # Pink for objects
-            label = f"Object: {item['label']}"
-        cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 3)
-        cv2.putText(frame, label, (x_min, y_min - 15), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 0), 2)  # Black text
+        else:  # Hole
+            color = (255, 0, 0) if severity == 'Severe' else (255, 255, 0) if severity == 'Moderate' else (0, 128, 0)
+            label = f"Hole: {severity}"
+        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)
     return frame