import gradio as gr

# Función para resolver problemas
def resolver_problema(problema):
    problema = problema.lower()
    
    if "vomita" in problema or "vomitando" in problema:
        return "Tu mascota podría tener una infección o haber ingerido algo indebido. No le des comida y acude al veterinario si persiste."
    elif "pulgas" in problema or "rasca" in problema:
        return "Puede tener pulgas o alergias. Báñalo con un shampoo antipulgas y consulta a tu veterinario para tratamiento."
    elif "no quiere comer" in problema:
        return "Puede ser un problema digestivo o de ánimo. Observa si presenta otros síntomas y visita al veterinario si continúa."
    elif "herida" in problema:
        return "Limpia la herida con agua y antiséptico. Si es profunda o sangra mucho, acude al veterinario urgentemente."
    elif "maulla mucho" in problema or "ladra mucho" in problema:
        return "Puede ser ansiedad, aburrimiento o dolor. Evalúa cambios recientes en el ambiente o salud del animal."
    else:
        return "Por favor, lleva a tu mascota al veterinario para una revisión más detallada."

# Interfaz Gradio
interfaz = gr.Interface(
    fn=resolver_problema,
    inputs=gr.Textbox(lines=4, placeholder="Describe el problema de tu perro o gato..."),
    outputs="text",
    title="Asistente Veterinario",
    description="Escribe el problema de tu perro o gato y obtén una posible solución o consejo inicial."
)

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