Update app.py

app.py

@@ -2,20 +2,22 @@ import gradio as gr
 import os, sys
 import cv2
 
-# Ensure services is importable
+# Add 'services' folder to Python path
 sys.path.append(os.path.join(os.path.dirname(__file__), "services"))
 
-from video_service import get_video_frame
+from video_service import get_live_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
 
-frame_generator = get_video_frame("data/sample_pole_video.mp4")
+# 🔁 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)
 
-def detect_frame():
+def detect_live_faults():
     try:
-        frame = next(frame_generator)
+        frame = next(frame_gen)
         temp_path = "temp.jpg"
         cv2.imwrite(temp_path, frame)
 
@@ -35,9 +37,11 @@ def detect_frame():
         return None
 
 gr.Interface(
-    fn=detect_frame,
+    fn=detect_live_faults,
     inputs=[],
     outputs="image",
     live=True,
-    title="VIEP Smart Pole Live Fault Detection"
+    title="📡 VIEP Live Smart Pole Fault Detection",
+    description="Live AI detection of faults like overheating, intrusion, and shadow coverage."
 ).launch()
+