File size: 1,032 Bytes
3568f82
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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():
    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

iface = gr.Interface(fn=monitor_feed, inputs=[], outputs="image", live=True, title="Solar Surveillance Feed Simulation")
iface.launch()