import gradio as gr
from transformers import pipeline

# Cargar modelo preentrenado para detección de desechos
detector = pipeline("object-detection", model="turhancan97/yolov5-detect-trash-classification")

def detect_waste(image):
    # Ejecutar detección de objetos
    results = detector(image)
    
    # Formatear resultados: {objeto: confianza}
    return {result["label"]: float(result["score"]) for result in results}

# Interfaz Gradio
iface = gr.Interface(
    fn=detect_waste,
    inputs=gr.Image(type="pil", label="Sube una imagen"),
    outputs=gr.Label(label="Detecciones"),
    title="Detección de Desechos Sólidos",
    description="Sube una foto para detectar desechos como plástico, vidrio, etc."
)

iface.launch()