Fix?

app.py

@@ -29,18 +29,29 @@ def analyze(text):
     
     has_high_ner = False
     
-    last_idx = 0
-    highlighted_data = []
+    res_ner = sorted(res_ner, key=lambda x: x['start'])
     
-    for ent in sorted(res_ner, key=lambda x: x['start']):
-        if ent['score'] > 0.90:
-            has_high_ner = True
+    highlighted_data = []
+    last_idx = 0
 
-            highlighted_data.append((text[last_idx:ent['start']], None))
-            highlighted_data.append((text[ent['start']:ent['end']], f"{ent['entity_group']} ({ent['score']:.2%})"))
+    for ent in res_ner:
+        if ent['start'] < last_idx:
+            continue
+            
+        if ent['score'] > 0.90:
+            if ent['start'] > last_idx:
+                highlighted_data.append((text[last_idx:ent['start']], None))
+            
+            label_text = f"{ent['entity_group']} ({ent['score']:.0%})"
+            highlighted_data.append((text[ent['start']:ent['end']], label_text))
+            
             last_idx = ent['end']
+            has_high_ner = True
+        else:
+            continue
     
-    highlighted_data.append((text[last_idx:], None))
+    if last_idx < len(text):
+        highlighted_data.append((text[last_idx:], None))
 
     if has_high_ner:
         return gr.update(visible=True), highlighted_data