import gradio as gr
from ultralytics import YOLO

# Load YOLO model
model = YOLO("Suspicious_Activities_nano.pt")

# Define a function to predict and return the result
def predict_suspicious_activity(image):
    results = model.predict(source=image, show=True, conf=0.6)
    return results.pandas().xywh[0].to_dict()  # Adjust as per your model output structure

# Create Gradio interface
interface = gr.Interface(
    fn=predict_suspicious_activity,
    inputs=gr.inputs.Image(type="pil"),  # Input type for uploading images
    outputs="json"  # Output type (you can adjust this based on your needs)
)

# Launch the interface
interface.launch()