import gradio as gr
from transformers import pipeline, set_seed

set_seed(42)  # Para reprodutibilidade

# Configuração segura dos modelos
modelo_rapido = pipeline(
    "text-generation",
    model="facebook/bart-large-mnli",
    do_sample=False,  # Desativa aleatoriedade
    max_length=30
)

modelo_detalhado = pipeline(
    "text-generation",
    model="microsoft/DialoGPT-medium",
    do_sample=False,
    temperature=0.7,  # Reduz criatividade
    max_length=50
)

arbitro = pipeline(
    "text2text-generation",
    model="google/flan-t5-base",  # Versão mais leve
    max_length=100
)

def chatbot(prompt):
    try:
        # Gera respostas com fallback
        resposta_rapida = modelo_rapido(prompt)[0]['generated_text']
        resposta_detalhada = modelo_detalhado(prompt)[0]['generated_text']
        
        # Limpeza básica das respostas
        resposta_rapida = resposta_rapida.split(".")[0] + "."
        resposta_detalhada = resposta_detalhada.split("\n")[0]
        
        return f"🔵 Resposta Rápida: {resposta_rapida}\n\n🟢 Resposta Detalhada: {resposta_detalhada}"
    
    except Exception as e:
        return f"Erro: {str(e)}"