import gradio as gr
from transformers import pipeline

ner_pipeline = pipeline("ner", model="dslim/bert-base-NER", grouped_entities=True)

def highlight_entities(text):
    entities = ner_pipeline(text)
    # Gradio มีระบบ Highlight ข้อความให้อัตโนมัติถ้าเราส่งค่าไปแบบนี้
    return {"text": text, "entities": entities}

demo = gr.Interface(
    fn=highlight_entities,
    inputs=gr.Textbox(placeholder="Enter sentence here..."),
    outputs=gr.HighlightedText(),
    title="Name Entity Recognition (NER) Finder"
)

demo.launch()