import gradio as gr
from predict import predict

def diagnostico(image):
    result = predict(image)
    return f"Classe: {result['classe']}\nConfiança: {result['confidence']:.2%}"

demo = gr.Interface(
    fn=diagnostico,
    inputs=gr.Image(type="pil"),
    outputs="text",
    title="Pet AI Diagnosis",
    description="Classificação de raças e análise dermatológica"
)

demo.launch()