import gradio as gr
from transformers import AutoTokenizer, AutoModelWithLMHead, AutoModelForSeq2SeqLM

title = "Spanish Text Simplification"
description = "Simplificación de noticias en español"
examples = [
    ["""
  Maider Etxebarria destaca que este chupinazo será para las más de 78.000 personas que nacieron en otros puntos de España y hoy viven aquí. 
  Muchas de ellas llegaron a nuestra ciudad en el Desarrollismo, pero desde entonces han ido llegando año a año a nuestra ciudad: 
  “Han contribuido a la construcción social y económica de nuestra ciudad”, ha asegurado Maider Etxebarria. 
  La alcaldesa ha destacado que estos centros regionales son un ejemplo de diversidad: 
  ”Hacen una gran labor para construir humanidad. Sin ellas, Vitoria no sería la ciudad abierta y diversa que es hoy” 
"""],
    ["""
La semana comienza con un descenso de las temperaturas en la mitad norte el lunes y el martes, 
según la predicción del portavoz de la Agencia Estatal de Meteorología (AEMET), Rubén del Campo. 
En concreto, este lunes la bajada será de hasta 10ºC en el Cantábrico con respecto a los valores del domingo. 
Además, los vientos del norte serán intensos en la costa gallega, Valle del Ebro y Ampurdán catalán durante el martes, 
jornada en la que el norte de Oviedo, Santander o Pamplona no llegarán a superar los 15ºC. 
Ya el miércoles, habrá una bajada de temperaturas más generalizada
"""]
]

# Cargar los modelos desde Hugging Face
io1 = gr.Interface.load("huggingface/crrodrvi/mbart-simplificacion")
io2 = gr.Interface.load("huggingface/facebook/bart-large-cnn")
# io3 = gr.Interface.load("huggingface/csebuetnlp/mT5_multilingual_XLSum")  
# io4 = gr.Interface.load("huggingface/google/pegasus-xsum")                   

# Crear la interfaz combinada usando Blocks
with gr.Blocks() as demo:
    gr.Markdown(f"### {title}")
    gr.Markdown(description)
    with gr.Row():
        io1.render()
        io2.render()

    gr.Examples(examples, inputs=gr.inputs.Textbox(lines=10, label="Text"))

# Lanzar la aplicación
demo.launch(share=True)