import gradio as gr
import cv2
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

# Initialize video feed
frame_gen = get_video_frame("data/pole_feed.mp4")

def monitor_feed():
    try:
        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

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