Update app.py

app.py

@@ -54,6 +54,30 @@ def get_model(model_name):
 # -----------------------
 # NER function (SAFE OUTPUT)
 # -----------------------
+
+# Function to run NER on input text
+def analyze_text(text, model_name):
+    ner = models[model_name]
+    ner_results = ner(text)
+    highlighted_text = []
+    last_idx = 0
+    for entity in ner_results:
+        start = entity["start"]
+        end = entity["end"]
+        label = entity["entity_group"]
+        # Add non-entity text
+        if start > last_idx:
+            highlighted_text.append((text[last_idx:start], None))
+        # Add entity text
+        highlighted_text.append((text[start:end], label))
+        last_idx = end
+    # Add any remaining text after the last entity
+    if last_idx < len(text):
+        highlighted_text.append((text[last_idx:], None))
+    return highlighted_text
+
+
+'''
 def analyze_text(text, model_name):
     ner = get_model(model_name)
     results = ner(text)
@@ -67,7 +91,7 @@ def analyze_text(text, model_name):
         score = ent.get("score", 0.0)
         lines.append(f"{word} -> {label} ({score:.2f})")
     return "\n".join(lines)
-
+'''
 # -----------------------
 # UI
 # -----------------------