Update app.py

app.py

@@ -1,30 +1,23 @@
-import gradio as gr
-import cv2
-from services.video_service import get_video_frame
-from services.detection_service import detect_objects
-from services.thermal_service import detect_thermal_anomalies
-from services.shadow_detection import detect_shadow_coverage
-from services.salesforce_dispatcher import send_to_salesforce
-
 frame_gen = get_video_frame("data/drone_day.mp4")
 
 def monitor_feed():
+    global frame_gen
     try:
         frame = next(frame_gen)
-        cv2.imwrite("temp.jpg", frame)
-        detections = detect_objects("temp.jpg")
-        thermal = detect_thermal_anomalies("temp.jpg")
-        shadow_flag = detect_shadow_coverage("temp.jpg")
-
-        alert_payload = {
-            "detections": detections,
-            "thermal": bool(thermal),
-            "shadow_issue": shadow_flag,
-        }
-        send_to_salesforce(alert_payload)
-        return frame
     except StopIteration:
-        return None
+        # Auto-reload video when finished
+        frame_gen = get_video_frame("data/drone_day.mp4")
+        frame = next(frame_gen)
+
+    cv2.imwrite("temp.jpg", frame)
+    detections = detect_objects("temp.jpg")
+    thermal = detect_thermal_anomalies("temp.jpg")
+    shadow_flag = detect_shadow_coverage("temp.jpg")
 
-iface = gr.Interface(fn=monitor_feed, inputs=[], outputs="image", live=True, title="Solar Surveillance Feed Simulation")
-iface.launch()
+    alert_payload = {
+        "detections": detections,
+        "thermal": bool(thermal),
+        "shadow_issue": shadow_flag,
+    }
+    send_to_salesforce(alert_payload)
+    return frame