import gradio as gr
from transformers import pipeline

# repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
repo_id = "luisvarona/clasificador-trek"
resumidor = pipeline('text2text-generation', model='luisvarona/modelo_resumen')

# Definimos una función que se encarga de llevar a cabo las predicciones
def predict(str):
    # dic_label_score = classifier(str)[0]
    # label = dic_label_score['label']
    # score = dic_label_score['score']
    # dicc = {'LABEL_0': 'ABBR', 'LABEL_1': 'ENTY', 'LABEL_2': 'DESC', 'LABEL_3': 'HUM', 'LABEL_4': 'LOC', 'LABEL_5': 'NUM'}
    # label = dicc[label]
    return resumidor(str)

texto_ejemplo_1 = 'Hermoso de Mendoza, actual portavoz del PSOE en el Ayuntamiento de Logroño, ha puesto a disposición del partido su cargo y próximamente dejará su acta de concejal. Ha explicado que es el «punto final a esta etapa política que ha durado 5 años».«En vista de la solicitud de la Secretaría General del Partido Socialista y en vista de lo que pudimos observar en la Comisión Ejecutiva Local, les quería anunciar que pongo a disposición del Partido Socialista el cargo de portavoz del Grupo Municipal Socialista y, al mismo tiempo quiero anunciarles que, en fechas próximas, dejaré el acta de concejal del Ayuntamiento de Logroño», ha comentado Hermoso. La reflexión que el Comité Ejecutivo local se autoimpuso hace unos días respecto al grupo municipal socialista ha causado un efecto inmediato en el hasta ahora portavoz, Pablo Hermoso de Mendoza, quien este lunes ha anunciado que pone la portavocía a disposición de su partido y que dejará su acta de concejal para iniciar una nueva etapa.'
texto_ejemplo_2 = 'La lluvia primero intermitente acabó arreciando fuerte sobre Logroño y, por tanto, afectando a las procesiones de Jueves Santo. La de Jesús Camino del Calvario llegó a la iglesia de Santiago poco después de las 21.00 horas cumpliendo prácticamente con el recorrido previsto. Sin embargo, la de La Piedad y la de las Siete Palabras tuvieron que acortar el recorrido. A mitad de recorrido la de La Piedad emprendió el regreso a la iglesia de Valvanera, donde los porteadores lograron introducir el paso, ya cubierto de plásticos, sobre las 21.20 horas, poco más de una hora después de haberlo sacado. Ya dentro del templo y a resguardo se cantó una saeta. «Nos hubiera gustado haber tenido un final como en otras ocasiones pero no ha sido posible», se expuso al final, agradeciendo la labor de la banda y de los maceros. La procesión logró finalizar, aunque antes de tiempo, con el rezo de la última estación.'

# Creamos la interfaz y la lanzamos. 
gr.Interface(fn=predict, inputs="textbox", outputs="textbox", examples=[texto_ejemplo_1, texto_ejemplo_2]).launch(share=True)