Update app.py

app.py

@@ -14,41 +14,6 @@ import numpy as np
 import cv2
 import time
 
-def process_sequential(frame: np.ndarray):
-    # Detect bridge piers first
-    detections, annotated_frame = process_bridge_piers(frame)
-    cv2.imshow("Bridge Piers Detection", annotated_frame)
-    cv2.waitKey(1)  # Wait for a short time (e.g., 1 ms)
-    time.sleep(1)  # Pause for 1 second for the user to see the result
-
-    # Replay the video (or go to next frame if real-time processing)
-    cv2.imshow("Bridge Piers Detection", frame)
-    cv2.waitKey(1)
-
-    # Detect culverts next
-    detections, annotated_frame = process_culverts(frame)
-    cv2.imshow("Culverts Detection", annotated_frame)
-    cv2.waitKey(1)
-    time.sleep(1)
-
-    # Replay the video for culverts detection
-    cv2.imshow("Culverts Detection", frame)
-    cv2.waitKey(1)
-
-    # Detect earthwork activities
-    detections, annotated_frame = process_earthwork(frame)
-    cv2.imshow("Earthwork Detection", annotated_frame)
-    cv2.waitKey(1)
-    time.sleep(1)
-
-    # Replay the video for earthwork detection
-    cv2.imshow("Earthwork Detection", frame)
-    cv2.waitKey(1)
-
-    # Add more steps if needed...
-
-    cv2.waitKey(0)  # Wait for a key press to close the window after detection
-    cv2.destroyAllWindows()
 
 
 # Suppress Ultralytics warning by setting a writable config directory
@@ -76,6 +41,35 @@ logging.basicConfig(
     level=logging.INFO,
     format="%(asctime)s - %(levelname)s - %(message)s"
 )
+# controller.py
+from services.under_construction.bridge_pier_check import process_bridge_piers
+from services.under_construction.culvert_check import process_culverts
+from services.under_construction.earthwork_detection import process_earthwork
+
+def detect_under_construction_objects(frame, mode: str):
+    """
+    Dispatches frame to the appropriate detection function based on mode.
+    Args:
+        frame: Input frame as a numpy array.
+        mode: One of "bridge", "culvert", "earthwork".
+    Returns:
+        Tuple of (detections, annotated_frame)
+    """
+    if mode == "bridge":
+        return process_bridge_piers(frame)
+    elif mode == "culvert":
+        return process_culverts(frame)
+    elif mode == "earthwork":
+        return process_earthwork(frame)
+    else:
+        raise ValueError(f"Unknown mode: {mode}")
+# main.py (or your video loop)
+from controller import detect_under_construction_objects
+
+# Assume `mode` is selected externally: "bridge", "culvert", or "earthwork"
+mode = "bridge"  # change as needed
+detections, annotated_frame = detect_under_construction_objects(frame, mode)
+
 
 # Global variables
 paused: bool = False