Update app.py

app.py

@@ -110,6 +110,57 @@ def highlight_html(text: str, entity_map: dict) -> str:
     )
     
 # NER Teks 
+def run_ner(text: str):
+    if not text or not text.strip():
+        return "<p style='color:#94a3b8;font-style:italic;padding:16px;'>Masukkan teks terlebih dahulu.</p>"
+    results = ner(text.strip())
+    if not results:
+        return "<p style='color:#94a3b8;font-style:italic;padding:16px;'>Tidak ada entitas yang ditemukan.</p>"
+
+    rows_html = ""
+    row_num = 1
+    seen_words = set()
+    for ent in results:
+        raw_label = ent["entity_group"]
+        label_id  = get_label_id(raw_label)
+        label_idn = LABEL_MAP.get(label_id, raw_label)
+        word      = clean_word(ent["word"])
+        if not word: continue
+        word_key = word.lower()
+        if word_key in seen_words: continue
+        seen_words.add(word_key)
+        hex_color = LABEL_HEX.get(label_id, "#e2e8f0")
+        score     = f"{ent['score']:.2%}"
+        row_bg    = "#f8faff" if row_num % 2 == 0 else "#ffffff"
+        rows_html += f"""
+        <tr style="background:{row_bg};">
+            <td style="padding:9px 14px;border-bottom:1px solid #e8edf5;text-align:center;color:#64748b;font-size:12px;">{row_num}</td>
+            <td style="padding:9px 14px;border-bottom:1px solid #e8edf5;font-weight:600;color:#1e293b;">{word}</td>
+            <td style="padding:9px 14px;border-bottom:1px solid #e8edf5;">
+                <span style="background:{hex_color};padding:3px 10px;border-radius:20px;font-size:12px;font-weight:600;color:#1e293b;">{label_idn}</span>
+            </td>
+            <td style="padding:9px 14px;border-bottom:1px solid #e8edf5;font-size:12px;color:#64748b;text-align:center;">{score}</td>
+        </tr>"""
+        row_num += 1
+
+    if not rows_html:
+        return "<p style='color:#94a3b8;font-style:italic;padding:16px;'>Tidak ada entitas yang ditemukan.</p>"
+
+    return f"""
+    <div style="overflow-x:auto;margin-top:4px;border-radius:12px;border:1px solid #e2e8f0;box-shadow:0 2px 12px rgba(0,0,0,0.06);">
+        <table style="width:100%;border-collapse:collapse;font-size:14px;font-family:'Segoe UI',sans-serif;">
+            <thead>
+                <tr style="background:linear-gradient(135deg,#1e40af,#6d28d9);">
+                    <th style="padding:12px 14px;color:#fff;width:55px;font-weight:600;font-size:12px;letter-spacing:0.05em;">NO</th>
+                    <th style="padding:12px 14px;color:#fff;text-align:left;font-weight:600;font-size:12px;letter-spacing:0.05em;">KATA / FRASA</th>
+                    <th style="padding:12px 14px;color:#fff;text-align:left;font-weight:600;font-size:12px;letter-spacing:0.05em;">ENTITAS</th>
+                    <th style="padding:12px 14px;color:#fff;width:90px;font-weight:600;font-size:12px;letter-spacing:0.05em;">SKOR</th>
+                </tr>
+            </thead>
+            <tbody>{rows_html}</tbody>
+        </table>
+    </div>"""
+
 def run_ner_file(upload_file):
     if upload_file is None:
         return "<p style='color:#94a3b8;font-style:italic;padding:16px;'>Unggah file terlebih dahulu.</p>", ""
@@ -225,7 +276,7 @@ HERO_HTML = """
 
 
 # Gradio UI 
-with gr.Blocks(title="NER for Financial Statements", css=CUSTOM_CSS) as demo:
+with gr.Blocks(title="NER for Financial Statements") as demo:
 
     gr.HTML(HERO_HTML)
 
@@ -276,4 +327,4 @@ with gr.Blocks(title="NER for Financial Statements", css=CUSTOM_CSS) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css=CUSTOM_CSS)