import gradio as gr
import re

from transformers import pipeline

ckpt = "Narrativaai/fake-news-detection-spanish"

classifier = pipeline("text-classification", model=ckpt)


def prediction(header, text):
        results = classifier(header + " [SEP] " + text)
        return results[0]["label"], results[0]["score"]


gradio_ui = gr.Interface(
    fn=prediction,
    title="Fake News Detector (Spanish)",
    description="Demo of our model for Fake News Detection Task (FakeDes) in IberLeF 2021",
    article="<p style = 'text-align: center' > <a href = 'https://www.narrativa.com/' target = '_blank' > Created by: Narrativa:</a> Natural Language Generation(NLG) | Gabriele, our machine learning-based platform, builds and deploys natural language solutions. # NLG #AI</p>",
    inputs=[
        gr.inputs.Textbox(lines=1, label="Type/Paste your headline here"),
        gr.inputs.Textbox(lines=6, label="Type/Paste the article body here"),
    ],
    outputs=[
        gr.outputs.Textbox(label="Label"),
        gr.outputs.Textbox(label="Score"),
    ],
    examples=[
    ["La candidata llamada a sustituir a Alberto Rodríguez renuncia a ocupar su escaño en el Congreso", '''Fátima González Bello, que concurrió como número dos en la lista de Unidas Podemos por la circunscripción de Tenerife en los pasados comicios generales, ha anunciado este miércoles que declina reemplazar al exdiputado Alberto Rodríguez tras la retirada de su escaño.

Una decisión que ha comunicado en redes sociales tras conocer que la Junta Electoral Central (JEC) ha expedido este miércoles la credencial para ocupar la vacante que deja Rodríguez, una vez que la presidenta del Congreso, Meritxell Batet, le comunicó la retirada del escaño en ejecución por la sentencia que el Tribunal Supremo dictó en su contra.

"Me resulta muy difícil poder ocupar el puesto del compañero Alberto en las circunstancias actuales e, igualmente, en lo personal tras cuatro años de dedicación como concejala en el Ayuntamiento de La Laguna y una vez abandonada la política institucional, no me veo en el momento de volver a la misma", ha detallado.'''],
        ['''La palabra "haiga", aceptada por la RAE''', '''La palabra "haiga", aceptada por la RAE La Real Academia de la Lengua (RAE), ha aceptado el uso de "HAIGA", para su utilización en las tres personas del singular del presente del subjuntivo del verbo hacer, aunque asegura que la forma más recomendable en la lengua culta para este tiempo, sigue siendo "haya".
Así lo han confirmado fuentes de la RAE, que explican que este cambio ha sido propuesto y aprobado por el pleno de la Academia de la Lengua, tras la extendida utilización por todo el territorio nacional, sobre todo, empleado por personas carentes de estudios o con estudios básicos de graduado escolar. Ya no será objeto de burla ese compañero que a diario repite aquello de "Mientras que haiga faena, no podemos quejarnos" o esa abuela que repite aquello de "El que haiga sacao los juguetes, que los recoja".
Entre otras palabras novedosas que ha aceptado la RAE, contamos también con "Descambiar", significa deshacer un cambio, por ejemplo "devolver la compra". Visto lo visto, nadie apostaría que la palabra "follamigos" sea la siguiente de la lista.'''],
    ]
)

gradio_ui.launch()