app.py

import gradio as gr
import cv2
import os, sys

# ✅ Fix: Add services/ to path manually
sys.path.append(os.path.join(os.path.dirname(__file__), "services"))

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

frame_gen = get_video_frame("data/sample_pole_video.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="VIEP Smart Pole Video Fault Detector")
iface.launch()