import gradio as gr
from PIL import Image
from object_detection import run_od_pipeline

def detect_objects(image: Image.Image) -> Image.Image:
    return run_od_pipeline(image)

# Gradio Interface
title = "🧠 Object Detection using Hugging Face"
description = "Upload an image and run object detection powered by Hugging Face models."

demo = gr.Interface(
    fn=detect_objects,
    inputs=gr.Image(type="pil", label="Upload an image"),
    outputs=gr.Image(type="pil", label="Detected Objects"),
    title=title,
    description=description,
    theme="default",
    allow_flagging="never"
)

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