app.py

import gradio as gr
import cv2

# ✅ Import all services before calling anything
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

# ✅ Initialize frame generator only after imports
frame_gen = get_video_frame("data/drone_day.mp4")

def monitor_feed():
    global frame_gen
    try:
        frame = next(frame_gen)
    except StopIteration:
        # Auto-reload video when it ends
        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")

    alert_payload = {
        "detections": detections,
        "thermal": bool(thermal),
        "shadow_issue": shadow_flag,
    }
    send_to_salesforce(alert_payload)
    return frame

# ✅ Setup Gradio interface
iface = gr.Interface(
    fn=monitor_feed,
    inputs=[],
    outputs="image",
    live=True,
    title="Solar Surveillance Feed Simulation"
)

iface.launch()