Update app.py

app.py

@@ -1,47 +1,55 @@
 import gradio as gr
-import os, sys
 import cv2
+import os
+import sys
 
-# Add 'services' folder to Python path
+# Add services path
 sys.path.append(os.path.join(os.path.dirname(__file__), "services"))
 
-from video_service import get_live_frame
+from video_service import get_video_frame
 from detection_service import detect_objects
 from thermal_service import detect_thermal_anomalies
 from shadow_detection import detect_shadow_coverage
 from salesforce_dispatcher import send_to_salesforce
 
-# 🔁 Live stream: Use webcam (0) or RTSP stream
-cap = cv2.VideoCapture(0)  # Replace 0 with RTSP URL if needed
-frame_gen = get_live_frame(cap)
+# Initialize video reader from sample video
+frame_generator = get_video_frame("data/sample_pole_video.mp4")
 
-def detect_live_faults():
+def detect_faults():
     try:
-        frame = next(frame_gen)
+        frame = next(frame_generator)
+        if frame is None:
+            return None  # Gradio will skip frame
+
+        # Save temporary frame
         temp_path = "temp.jpg"
         cv2.imwrite(temp_path, frame)
 
-        objects = detect_objects(temp_path)
+        # Fault detection
+        intrusion = detect_objects(temp_path)
         thermal = detect_thermal_anomalies(temp_path)
         shadow = detect_shadow_coverage(temp_path)
 
-        alert = {
-            "detections": objects,
-            "thermal": bool(thermal),
-            "shadow_issue": shadow,
-        }
+        # Print for debugging
+        print("Intrusion:", intrusion)
+        print("Thermal:", thermal)
+        print("Shadow:", shadow)
+
+        # Send to Salesforce mock
+        send_to_salesforce({
+            "detections": intrusion,
+            "thermal": thermal,
+            "shadow_issue": shadow
+        })
 
-        send_to_salesforce(alert)
         return frame
     except StopIteration:
         return None
 
 gr.Interface(
-    fn=detect_live_faults,
+    fn=detect_faults,
     inputs=[],
     outputs="image",
-    live=True,
-    title="📡 VIEP Live Smart Pole Fault Detection",
-    description="Live AI detection of faults like overheating, intrusion, and shadow coverage."
+    title="VIEP Live Fault Detection (Simulated)",
+    live=True
 ).launch()
-