import gradio as gr
from transformers import pipeline

# Pipeline de geração de texto
pipe = pipeline("text-generation", model="gpt2")

# Função principal do CRM Doctor
def chat(user_input):
    output = pipe(user_input, max_length=100)[0]["generated_text"]
    return output

# Interface Gradio
iface = gr.Interface(
    fn=chat,
    inputs=gr.Textbox(lines=3, placeholder="Descreva seus sintomas aqui..."),
    outputs="text",
    title="CRM Doctor",
    description="Seu assistente de pré-diagnóstico de saúde.\n\n🚨 *Este assistente não substitui uma consulta médica presencial.*"
)

iface.launch()