Update app.py

app.py

@@ -7,18 +7,28 @@ import random
 from datetime import datetime
 from collections import Counter
 from services.video_service import get_next_video_frame, reset_video_index, preload_video, release_video
-from services.crack_detection_service import detect_cracks_and_objects
-from services.overlay_service import overlay_boxes
+from services.detection_service import process_frame as process_generic
 from services.metrics_service import update_metrics
+from services.overlay_service import overlay_boxes
 from services.salesforce_dispatcher import dispatch_to_salesforce
+from services.shadow_detection import detect_shadows
+from services.thermal_service import process_thermal
 # Under Construction services
 from services.under_construction.earthwork_detection import process_earthwork
 from services.under_construction.culvert_check import process_culverts
 from services.under_construction.bridge_pier_check import process_bridge_piers
-# Original services
-from services.detection_service import process_frame as process_generic
-from services.shadow_detection import detect_shadows
-from services.thermal_service import process_thermal
+# Operations Maintenance services
+from services.operations_maintenance.crack_detection import detect_cracks_and_objects
+from services.operations_maintenance.pothole_detection import process_potholes
+from services.operations_maintenance.signage_check import process_signages
+# Road Safety services
+from services.road_safety.barrier_check import process_barriers
+from services.road_safety.lighting_check import process_lighting
+from services.road_safety.accident_spot_check import process_accident_spots
+# Plantation services
+from services.plantation.plant_count import process_plants
+from services.plantation.plant_health import process_plant_health
+from services.plantation.missing_patch_check import process_missing_patches
 
 # Globals
 paused = False
@@ -33,7 +43,12 @@ last_timestamp = ""
 last_detected_images = []  # Store up to 100+ crack images
 gps_coordinates = []
 video_loaded = False
-enable_under_construction = False  # Toggle for under_construction services
+service_toggles = {
+    "under_construction": False,
+    "operations_maintenance": True,  # Default to crack detection focus
+    "road_safety": False,
+    "plantation": False
+}
 
 # Constants
 DEFAULT_VIDEO_PATH = "sample.mp4"
@@ -60,14 +75,14 @@ def initialize_video(video_file=None):
     log_entries.append(status)
     return status
 
-def toggle_under_construction(value):
+def toggle_service(service_name, value):
     """
-    Toggle the under_construction services.
+    Toggle a service category.
     """
-    global enable_under_construction
-    enable_under_construction = value
-    log_entries.append(f"Under Construction Services {'Enabled' if value else 'Disabled'}")
-    return f"Under Construction Services: {'Enabled' if value else 'Disabled'}"
+    global service_toggles
+    service_toggles[service_name] = value
+    log_entries.append(f"{service_name.replace('_', ' ').title()} Services {'Enabled' if value else 'Disabled'}")
+    return f"{service_name.replace('_', ' ').title()} Services: {'Enabled' if value else 'Disabled'}"
 
 def monitor_feed():
     """
@@ -92,24 +107,38 @@ def monitor_feed():
         # Initialize detected items list
         all_detected_items = []
 
-        # Always run crack detection (primary focus)
-        crack_items = detect_cracks_and_objects(frame)
-        frame = overlay_boxes(frame, crack_items)
-        all_detected_items.extend(crack_items)
-
-        # Run Under Construction services if enabled
-        if enable_under_construction:
+        # Run services based on toggles
+        if service_toggles["under_construction"]:
             earthwork_dets, frame = process_earthwork(frame)
             culvert_dets, frame = process_culverts(frame)
             bridge_pier_dets, frame = process_bridge_piers(frame)
             all_detected_items.extend(earthwork_dets + culvert_dets + bridge_pier_dets)
 
+        if service_toggles["operations_maintenance"]:
+            crack_items = detect_cracks_and_objects(frame)
+            frame = overlay_boxes(frame, crack_items)
+            pothole_dets, frame = process_potholes(frame)
+            signage_dets, frame = process_signages(frame)
+            all_detected_items.extend(crack_items + pothole_dets + signage_dets)
+
+        if service_toggles["road_safety"]:
+            barrier_dets, frame = process_barriers(frame)
+            lighting_dets, frame = process_lighting(frame)
+            accident_dets, frame = process_accident_spots(frame)
+            all_detected_items.extend(barrier_dets + lighting_dets + accident_dets)
+
+        if service_toggles["plantation"]:
+            plant_dets, frame = process_plants(frame)
+            health_dets, frame = process_plant_health(frame)
+            missing_dets, frame = process_missing_patches(frame)
+            all_detected_items.extend(plant_dets + health_dets + missing_dets)
+
         # Fallback: Run generic detection if no items detected
         if not all_detected_items:
             generic_dets, frame = process_generic(frame)
             all_detected_items.extend(generic_dets)
 
-        # Optional: Run shadow detection (only log results for now)
+        # Optional: Run shadow detection
         shadow_results = detect_shadows(frame)
         shadow_dets = shadow_results["detections"]
         frame = shadow_results["frame"]
@@ -130,8 +159,8 @@ def monitor_feed():
         gps_coord = [17.385044 + random.uniform(-0.001, 0.001), 78.486671 + frame_count * 0.0001]
         gps_coordinates.append(gps_coord)
 
-        # Save frame if cracks are detected
-        if any(item['type'] == 'crack' for item in crack_items):
+        # Save frame if cracks are detected (only if operations_maintenance is enabled)
+        if service_toggles["operations_maintenance"] and any(item['type'] == 'crack' for item in all_detected_items if 'type' in item):
             captured_frame_path = os.path.join(CAPTURED_FRAMES_DIR, f"crack_{frame_count}.jpg")
             cv2.imwrite(captured_frame_path, frame)
             last_detected_images.append(captured_frame_path)
@@ -159,13 +188,15 @@ def monitor_feed():
         last_frame = frame.copy()
         last_detections = metrics
 
-        # Update logs and stats
-        crack_detected = len([item for item in crack_items if item['type'] == 'crack'])
-        crack_severity_all.extend([
-            item['severity']
-            for item in crack_items
-            if item['type'] == 'crack' and 'severity' in item
-        ])
+        # Update logs and stats (focus on cracks if operations_maintenance is enabled)
+        crack_detected = 0
+        if service_toggles["operations_maintenance"]:
+            crack_detected = len([item for item in all_detected_items if 'type' in item and item['type'] == 'crack'])
+            crack_severity_all.extend([
+                item['severity']
+                for item in all_detected_items
+                if 'type' in item and item['type'] == 'crack' and 'severity' in item
+            ])
 
         log_entries.append(f"{last_timestamp} - Frame {frame_count} - Cracks: {crack_detected} - Total Detections: {len(all_detected_items)} - GPS: {gps_coord} - Avg Conf: {metrics['avg_confidence']:.2f}")
         crack_counts.append(crack_detected)
@@ -182,9 +213,12 @@ def monitor_feed():
     cv2.putText(frame, f"Frame: {frame_count}", (10, 25), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
     cv2.putText(frame, f"{last_timestamp}", (10, 50), cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
 
-    # Generate charts
-    line_chart = generate_line_chart()
-    pie_chart = generate_pie_chart()
+    # Generate charts (only if operations_maintenance is enabled)
+    line_chart = None
+    pie_chart = None
+    if service_toggles["operations_maintenance"]:
+        line_chart = generate_line_chart()
+        pie_chart = generate_pie_chart()
 
     return frame[:, :, ::-1], json.dumps(last_detections, indent=2), "\n".join(log_entries[-10:]), line_chart, pie_chart, last_detected_images
 
@@ -285,10 +319,20 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         load_button = gr.Button("Load Video")
     video_status = gr.Textbox(label="Video Load Status", value="Please upload a video file or ensure 'sample.mp4' exists in the root directory.")
 
-    # Toggle for Under Construction services
+    # Toggles for service categories
     with gr.Row():
-        under_construction_toggle = gr.Checkbox(label="Enable Under Construction Services", value=False)
-        toggle_status = gr.Textbox(label="Toggle Status", value="Under Construction Services: Disabled")
+        with gr.Column():
+            uc_toggle = gr.Checkbox(label="Enable Under Construction Services", value=service_toggles["under_construction"])
+            uc_status = gr.Textbox(label="Under Construction Status", value="Under Construction Services: Disabled")
+        with gr.Column():
+            om_toggle = gr.Checkbox(label="Enable Operations Maintenance Services", value=service_toggles["operations_maintenance"])
+            om_status = gr.Textbox(label="Operations Maintenance Status", value="Operations Maintenance Services: Enabled")
+        with gr.Column():
+            rs_toggle = gr.Checkbox(label="Enable Road Safety Services", value=service_toggles["road_safety"])
+            rs_status = gr.Textbox(label="Road Safety Status", value="Road Safety Services: Disabled")
+        with gr.Column():
+            pl_toggle = gr.Checkbox(label="Enable Plantation Services", value=service_toggles["plantation"])
+            pl_status = gr.Textbox(label="Plantation Status", value="Plantation Services: Disabled")
 
     status_text = gr.Markdown("**Status:** 🟢 Ready (Upload a video to start)")
 
@@ -335,10 +379,25 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         outputs=[video_status]
     )
 
-    under_construction_toggle.change(
-        toggle_under_construction,
-        inputs=[under_construction_toggle],
-        outputs=[toggle_status]
+    uc_toggle.change(
+        toggle_service,
+        inputs=[gr.State("under_construction"), uc_toggle],
+        outputs=[uc_status]
+    )
+    om_toggle.change(
+        toggle_service,
+        inputs=[gr.State("operations_maintenance"), om_toggle],
+        outputs=[om_status]
+    )
+    rs_toggle.change(
+        toggle_service,
+        inputs=[gr.State("road_safety"), rs_toggle],
+        outputs=[rs_status]
+    )
+    pl_toggle.change(
+        toggle_service,
+        inputs=[gr.State("plantation"), pl_toggle],
+        outputs=[pl_status]
     )
 
     pause_btn.click(toggle_pause, outputs=status_text)