import gradio as gr
from gradio.mix import Parallel, Series
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
"""""]
]

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")                   

iface = Parallel(io1, io2, 
                 theme='huggingface', 
                 inputs = gr.inputs.Textbox(lines = 10, label="Text"), title=title, description=description, examples=examples)

iface.launch(share=True)