Update services/operations_maintenance/pothole_detection.py

services/operations_maintenance/pothole_detection.py

@@ -26,17 +26,6 @@ except Exception as e:
     model = None
 
 def process_potholes(frame: np.ndarray) -> Tuple[List[Dict[str, Any]], np.ndarray]:
-    """
-    Detect potholes in a video frame using YOLOv8m.
-
-    Args:
-        frame (np.ndarray): Input frame in BGR format.
-
-    Returns:
-        Tuple[List[Dict[str, Any]], np.ndarray]: A tuple containing:
-            - List of detected potholes.
-            - Annotated frame with bounding boxes and labels.
-    """
     # Validate input frame
     if not isinstance(frame, np.ndarray) or frame.size == 0:
         logging.error("Invalid input frame provided to pothole_detection.")
@@ -78,7 +67,7 @@ def process_potholes(frame: np.ndarray) -> Tuple[List[Dict[str, Any]], np.ndarra
                 "coordinates": [x_min, y_min, x_max, y_max]
             })
 
-            color = (255, 127, 80)  # Coral (red + orange mix) for potholes
+            color = (255, 0, 0)  # Red for potholes
             cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), color, 2)
             cv2.putText(
                 frame,