import gradio as gr
import spacy
from spacy import displacy

nlp = spacy.load("sr_ner_tesla_j125")

def calculate(text):
    try:
        doc = nlp(text)
        return displacy.render(doc, style="ent")
    except Exception as e:
        return f"Error: {e}"

iface = gr.Interface(
    fn=calculate, 
    inputs=gr.Textbox(lines=5, placeholder="Unesite rečenicu ovde..."),
    outputs=gr.HTML(label="Imenovani entiteti"),
    title="NER for Serbian",
    description="Demonstracija spaCy NER modela sa jerteh125 BERT-om.",
    examples=[
        ["Ana voli Milovana."],
        ["Brograd je glavni grad Srbije."],
        ["Nikoli Tesla je bio izumitelj i naučnik."],
        ["Monaliza je slika Leonarda da Vinčija."],
        ["Sveti Sava je bio prvi srpski arhiepiskop."],
    ]
)

iface.launch()