from transformers import pipeline
import gradio as gr

# Cargar un modelo de lenguaje médico de Hugging Face
modelo_medico = pipeline("text-generation", model="microsoft/BioGPT-Large")

# Función para hacer preguntas al bot
def pregunta_medica(pregunta):
    respuesta = modelo_medico(
        f"Eres un experto médico con vasto conocimiento en fisiología, bioquímica, farmacología y otras áreas médicas. Responde de manera sarcástica y directa, como el Dr. House. Pregunta: {pregunta}",
        max_length=150
    )
    return respuesta[0]['generated_text']

# Crear la interfaz web con Gradio
interfaz = gr.Interface(
    fn=pregunta_medica,
    inputs="text",
    outputs="text",
    title="Dr. House Bot",
    description="Pregúntale al Dr. House sobre medicina."
)

# Lanzar la interfaz
interfaz.launch()