import gradio as gr
from transformers import pipeline, set_seed

set_seed(42)

# Modelo em português 100% funcional
modelo = pipeline("text-generation", model="pierreguillou/gpt2-small-portuguese")

# Função do chatbot
def responder(pergunta):
    resposta = modelo(pergunta, max_length=80, num_return_sequences=1)[0]['generated_text']
    return resposta

# Interface Gradio
gr.Interface(
    fn=responder,
    inputs=gr.Textbox(label="Digite sua pergunta em português"),
    outputs=gr.Textbox(label="Resposta do modelo"),
    title="Chatbot GPT-2 em Português 🇧🇷",
    description="Este chatbot responde perguntas usando o modelo GPT-2 treinado em português."
).launch()