import gradio as gr
from transformers import pipeline
import os

token = os.getenv("HF_TOKEN")
get_completion = pipeline(
    "ner", model="impresso-project/bert-newsagency-ner-fr", use_auth_token=token
)


def ner(input):
    output = get_completion(input)
    return {"text": input, "entities": output}


demo = gr.Interface(
    fn=ner,
    inputs=[
        gr.Textbox(
            label="Find the news agency mentions in the following text in French:",
            lines=2,
        )
    ],
    outputs=[gr.HighlightedText(label="Text with news agency mentions")],
    title="News Agency Recognition with impresso-project/bert-newsagency-ner-fr",
    description="Find entities using the `impresso-project/bert-newsagency-ner-fr` model under the hood!",
    allow_flagging="never",
    # Here we introduce a new tag, examples, easy to use examples for your application
    examples=[
        "Des chercheurs de l'Université de Cambridge ont développé une nouvelle technique de calcul quantique qui "
        "promet d'augmenter exponentiellement les vitesses de calcul. Cette percée, décrite comme un 'bond quantique' "
        "dans la technologie informatique, pourrait ouvrir la voie à des capacités de traitement de données "
        "ultra-rapides et sécurisées. Cette avancée est censée avoir le potentiel d'impacter significativement les "
        "domaines de la cryptographie, de la science des matériaux et de la modélisation des systèmes complexes. Les "
        "experts dans le domaine ont exprimé leur enthousiasme face aux progrès rapides et à leurs implications pour "
        "les innovations technologiques futures. Le rapport complet sur ces découvertes a été publié dans la "
        "prestigieuse revue 'Nature Physics'. (Source : Reuters)"
        "Les tensions continuent de monter alors que les récentes élections parlementaires en Suède ont conduit à un "
        "changement inattendu dans le paysage politique. Le parti d'extrême droite, connu pour ses politiques "
        "strictes en matière d'immigration et ses vues eurosceptiques, a gagné un nombre substantiel de sièges, "
        "remettant en question l'équilibre traditionnel des pouvoirs. Les analystes politiques débattent des effets à "
        "long terme de ce changement, beaucoup exprimant leur préoccupation quant au potentiel d'une polarisation "
        "accrue dans la société suédoise. Alors que les partis négocient pour former un nouveau gouvernement, "
        "la communauté internationale observe attentivement pour voir comment ces développements influenceront les "
        "politiques étrangères et domestiques de la Suède. (Reporté par l'AFP)",
        "UP - Une récente étude publiée par des chercheurs canadiens met en lumière les effets bénéfiques de la "
        "méditation sur la réduction du stress et de l'anxiété. Les résultats, issus d'une analyse approfondie de "
        "plusieurs groupes de volontaires, indiquent une nette amélioration de la santé mentale des participants qui "
        "ont suivi un programme de méditation guidée sur huit semaines. Ces découvertes pourraient influencer les "
        "futures recommandations en matière de pratiques de bien-être mental.",
        "(BUP.) Les tensions diplomatiques entre les États-Unis et la Chine ont atteint un nouveau sommet hier, "
        "suite à l'annonce de sanctions commerciales réciproques qui ont ébranlé les marchés internationaux. Selon "
        "des sources officielles, ces mesures pourraient avoir des répercussions significatives sur le commerce "
        "mondial, exacerbant les incertitudes économiques déjà présentes. Les discussions en cours visent à apaiser "
        "les tensions, mais les perspectives restent incertaines.",
        "British United Press - On the home front, the British populace remains steadfast in the face of ongoing air "
        "raids. In London, despite the destruction, the spirit of the people is unbroken, with volunteers and civil "
        "defense units working tirelessly to support the war effort. Reports from BUP correspondents highlight the "
        "nationwide push for increased production in factories, essential for supplying the front lines with the "
        "materials needed for victory.",
    ],
)
demo.launch()