import gradio as gr
from utils.detector import detect_faults
from PIL import Image

def analyze_image(image):
    results = detect_faults(image)
    return results

demo = gr.Interface(
    fn=analyze_image,
    inputs=gr.Image(type="pil"),
    outputs="json",
    title="VIEP Smart Pole Fault Detection",
    description="Upload a surveillance image to detect faults like intrusion, overheating, or dust/shade issues."
)

if __name__ == "__main__":
    demo.launch()