app

app.py

@@ -36,18 +36,11 @@ def prepare_entities_for_highlight(text, results):
                 entity_text = text[
                     entity["start"] : entity["end"]
                 ].strip()  # Ensure we're working with the correct portion of the text
-                print(
-                    f"Entity text: {entity_text}, Start: {entity['start']}, End: {entity['end']}, Type: {entity['entity']}"
-                )
-                entities.append(
-                    {
-                        "text": entity_text,
-                        "score": entity["score"],
-                        "start": entity["start"],
-                        "end": entity["end"],
-                        "label": entity["entity"],
-                    }
-                )
+                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"])