app

app.py

@@ -24,6 +24,8 @@ def prepare_entities_for_highlight(text, results):
     # Print debug info about tokenization
     print(f"Original text: {text}")
 
+    # 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"])
@@ -39,9 +41,11 @@ def prepare_entities_for_highlight(text, results):
                 )
                 entities.append(
                     {
+                        "text": entity_text,
+                        "score": entity["score"],
                         "start": entity["start"],
                         "end": entity["end"],
-                        "label": f"{entity['entity']}",
+                        "label": entity["entity"],
                     }
                 )