Update app.py

app.py

@@ -10,12 +10,23 @@ model = AutoModelForTokenClassification.from_pretrained(model_name)
 ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
 
 # UI ด้วย Streamlit
-st.title("NER Analysis App")
-text = st.text_area("Enter text for NER analysis:")
+col1, col2 = st.columns(2)
 
-if st.button("Analyze"):
-    ner_results = ner_pipeline(text)
-    text=""
-    for entity in ner_results:
-        text = text + f"Entity: {entity['word']}, Label: {entity['entity']}, Score: {entity['score']:.4f}"
-    st.text_area(text)
+with col1:
+    st.header("Input")
+    text = st.text_area("Enter text for NER analysis:")
+    analyze_button = st.button("Analyze")
+
+
+    
+with col2:
+    st.header("Output")
+    if analyze_button:
+        ner_results = ner_pipeline(text)
+        
+        # Display results in a structured output block
+        if ner_results:
+            output_data = [{"Entity": entity['word'], "Label": entity['entity'], "Score": f"{entity['score']:.4f}"} for entity in ner_results]
+            st.table(output_data)  # Display as a table
+        else:
+            st.write("No entities found.")