import gradio as gr
from transformers import pipeline

# Creează pipeline-ul de rezumare
summarizer = pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")

# Funcția care face rezumatul
def rezuma_text(text):
    summary = summarizer(text, max_length=100, min_length=30, do_sample=False)
    return summary[0]['summary_text']

# Interfața Gradio
gr.Interface(
    fn=rezuma_text,
    inputs="text",
    outputs="text",
    title="SmartRezumat",
    description="AI care transformă texte lungi în rezumate scurte, în limba română"
).launch()