Update app.py

app.py

@@ -1,55 +1,34 @@
 import gradio as gr
 import cv2
-import os
-import sys
+from services.video_service import get_video_frame
+from services.detection_service import detect_objects
+from services.fault_service import detect_pole_faults
+from services.salesforce_dispatcher import send_to_salesforce
 
-# Add services path
-sys.path.append(os.path.join(os.path.dirname(__file__), "services"))
+# Initialize video feed
+frame_gen = get_video_frame("data/pole_feed.mp4")
 
-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
-
-# Initialize video reader from sample video
-frame_generator = get_video_frame("data/sample_pole_video.mp4")
-
-def detect_faults():
+def monitor_feed():
     try:
-        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)
-
-        # Fault detection
-        intrusion = detect_objects(temp_path)
-        thermal = detect_thermal_anomalies(temp_path)
-        shadow = detect_shadow_coverage(temp_path)
-
-        # 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
-        })
-
+        frame = next(frame_gen)
+        cv2.imwrite("temp.jpg", frame)
+        detections = detect_objects("temp.jpg")
+        faults = detect_pole_faults("temp.jpg")
+        alert_payload = {
+            "detections": detections,
+            "faults": bool(faults),
+            "fault_details": faults
+        }
+        send_to_salesforce(alert_payload)
         return frame
     except StopIteration:
         return None
 
-gr.Interface(
-    fn=detect_faults,
+iface = gr.Interface(
+    fn=monitor_feed,
     inputs=[],
     outputs="image",
-    title="VIEP Live Fault Detection (Simulated)",
-    live=True
-).launch()
+    live=True,
+    title="Pole Fault Detection Feed Simulation"
+)
+iface.launch()