Update services/under_construction/earthwork_detection.py

services/under_construction/earthwork_detection.py

@@ -15,11 +15,12 @@ def process_earthwork(frame):
         frame: Input frame (numpy array)
     Returns:
         list: List of detected earthwork activities
-        numpy array: Original frame (overlays handled elsewhere)
+        numpy array: Annotated frame with numbered labels
     """
     results = model(frame)
 
     detections = []
+    line_counter = 1  # Initialize counter for numbered labels
 
     for r in results:
         for box in r.boxes:
@@ -33,10 +34,21 @@ def process_earthwork(frame):
             xyxy = box.xyxy[0].cpu().numpy()
             x_min, y_min, x_max, y_max = map(int, xyxy)
 
+            # Add numbered label
+            detection_label = f"Line {line_counter} - {label.capitalize()} (Conf: {conf:.2f})"
             detections.append({
                 "type": label,
+                "label": detection_label,
                 "confidence": conf,
                 "coordinates": [x_min, y_min, x_max, y_max]
             })
 
+            # Draw bounding box and label
+            color = (255, 228, 181)  # Moccasin for earthwork
+            cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
+            cv2.putText(frame, detection_label, (x_min, y_min - 10), 
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, color, 2)
+            
+            line_counter += 1
+
     return detections, frame