app

app.py

@@ -51,21 +51,20 @@ def prepare_entities_for_highlight(text, results):
     print("Results:", results)
     # it should look like:
     # [{'entity': 'org.ent.pressagency.Reuters', 'score': np.float32(98.47), 'index': 78, 'word': 'Reuters', 'start': 440, 'end': 447}]
-    for category, entity_list in results.items():
-        for entity in entity_list:
-            entity_span = (entity["start"], entity["end"])
-
-            # Only add non-overlapping entities
-            if entity_span not in seen_spans:
-                seen_spans.add(entity_span)
-                entity_text = text[
-                    entity["start"] : entity["end"]
-                ].strip()  # Ensure we're working with the correct portion of the text
-                entity["text"] = entity_text
-                entity.pop("word")
-                print(f"Entity text: {entity}")
-
-                entities.append(entity)
+    for entity in results.items():
+        entity_span = (entity["start"], entity["end"])
+
+        # Only add non-overlapping entities
+        if entity_span not in seen_spans:
+            seen_spans.add(entity_span)
+            entity_text = text[
+                entity["start"] : entity["end"]
+            ].strip()  # Ensure we're working with the correct portion of the text
+            entity["text"] = entity_text
+            entity.pop("word")
+            print(f"Entity text: {entity}")
+
+            entities.append(entity)
 
     # Sort entities by their start position
     entities = sorted(entities, key=lambda x: x["start"])