Update app.py

app.py

@@ -66,3 +66,36 @@ with col2:
         
 with col3:
     st.header("Anotation")
+    # สร้าง NER Pipeline
+    ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
+    
+    def annotate_text(text):
+        ner_results = ner_pipeline(text)
+        
+        # สร้าง Annotation Output
+        annotations = []
+        highlighted_text = text
+        
+        for entity in ner_results:
+            word = entity['word']
+            entity_label = entity['entity']
+            score = round(entity['score'], 4)
+            
+            # สร้าง Dictionary ของ Entity
+            annotations.append({"Entity": word, "Label": entity_label, "Score": score})
+            
+            # Highlight คำที่เป็น Entity
+            highlighted_text = highlighted_text.replace(word, f'<mark style="background-color: yellow">{word} ({entity_label})</mark>')
+        
+        return highlighted_text, json.dumps(annotations, indent=2)
+    
+    # UI ด้วย Gradio
+    demo = gr.Interface(
+        fn=annotate_text,
+        inputs=gr.Textbox(lines=10, placeholder="Enter text for NER analysis..."),
+        outputs=[gr.HTML(label="Annotated Text"), gr.JSON(label="Entities List")],
+        title="NER Annotation Tool",
+        description="Highlight Named Entities and display them in a structured format."
+    )
+
+demo.launch()