Merge pull request #82 from Knox-AAU/80-fix-start-and-end-index-of-en…

…t-ments Fixed start and end index
Knox-AAU · Nov 30, 2023 · ada3ff2 · ada3ff2
2 parents 6d6d10e + d3744d1
commit ada3ff2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/components/GetSpacyData.py b/components/GetSpacyData.py
@@ -108,8 +108,8 @@ def GetEntities(doc) -> List[Entity]:
         entities.append(
             Entity(
                 name=entity.text,
-                startIndex=entity.start_char,
-                endIndex=entity.end_char,
+                startIndex=entity.start_char-entity.sent.start_char,
+                endIndex=entity.end_char-entity.sent.start_char,
                 sentence=entity.sent.text,
                 sentenceStartIndex=entity.sent.start_char,
                 sentenceEndIndex=entity.sent.end_char,