import gradio as gr
from transformers import pipeline

instructions = "Enter a sentence in any language and click **Extract Entities**."

ner = pipeline(
    "ner",
    model="Keyurjotaniya007/xlm-roberta-base-xtreme-multilingual-ner-2.0",
    grouped_entities=True   
)

def extract_entities(text):
    results = ner(text)
    return [(ent['word'], ent['entity_group']) for ent in results]

with gr.Blocks() as demo:
    gr.Markdown(instructions)
    with gr.Row():
        inp = gr.Textbox(label="Enter Text", placeholder="Type a sentence in any language...", lines=3)
        out = gr.HighlightedText(label="Named Entities")   
    btn = gr.Button("Extract Entities")

    btn.click(fn=extract_entities, inputs=inp, outputs=out)

if __name__ == "__main__":
    demo.launch()