import gradio as gr
from transformers import pipeline

# Imprimir a versão do Gradio para verificação
print("Gradio version:", gr.__version__)

# Carrega o modelo para conversação
generator = pipeline('text-generation', model='microsoft/DialoGPT-small')

# Função para gerar resposta a partir da entrada do usuário
def generate_response(prompt):
    response = generator(prompt, max_length=200, num_return_sequences=1)
    return response[0]['generated_text']

# Interface Gradio
iface = gr.Interface(
    fn=generate_response,
    inputs=gr.Textbox(lines=2, placeholder="Digite sua mensagem aqui..."),
    outputs=gr.Textbox(),
    title="Chatbot de Conversação",
    description="Este é um chatbot de conversação geral. Insira sua mensagem no campo abaixo e receba uma resposta."
)

if __name__ == "__main__":
    iface.launch()  # Remova enable_api=True, pois não é mais necessário