from transformers import pipeline
import gradio as gr

# Carrega o modelo de sumarização
#summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
summarizer = pipeline("summarization", model="t5-small")

# Função para sumarizar texto
def summarize_text(text, max_length=130, min_length=30):
    summary = summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)
    return summary[0]['summary_text']

# Interface Gradio
def gradio_interface(text):
    return summarize_text(text)

interface = gr.Interface(
    fn=gradio_interface,
    inputs=gr.Textbox(lines=10, placeholder="Cole seu texto aqui...", label="Texto de Entrada"),
    outputs=gr.Textbox(lines=5, label="Sumarização"),
    title="Sumarização de Texto com BART",
    description="Insira um texto longo e veja uma versão resumida gerada por IA."
)

# Inicia a interface
interface.launch()