Explicitly type each token to string to correctly calculate its length

ufal · Feb 3, 2025 · e53fbf9 · e53fbf9
1 parent 33f4c34
commit e53fbf9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/factgenie/static/js/span-annotator.js b/factgenie/static/js/span-annotator.js
@@ -128,13 +128,14 @@ class SpanAnnotator {
         let currentIndex = 0;
         if (this.granularity === 'words') {
             const parts = text.split(/(\s+)/);
+
             return parts.map((part, arrayIndex) => {
                 if (arrayIndex % 2 === 1) {
                     return ''; // Remove standalone whitespace spans
                 }
 
                 const whitespace = arrayIndex < parts.length - 1 ? parts[arrayIndex + 1] : '';
-                const fullContent = part + whitespace;
+                const fullContent = String(part) + whitespace;
                 const span = `<span class="annotatable" 
                     data-index="${currentIndex}" 
                     data-content="${part}"
@@ -356,7 +357,7 @@ class SpanAnnotator {
         const [min, max] = [Math.min(startIdx, endIdx), Math.max(startIdx, endIdx)];
 
         // Get the actual end position by adding length of the last token
-        const maxWithLength = max + $end.data('content').length - 1;
+        const maxWithLength = max + String($end.data('content')).length - 1;
 
         // Check for overlap if not allowed
         if (!this.overlapAllowed && this._hasExistingAnnotations(doc, min, maxWithLength)) {
@@ -409,7 +410,7 @@ class SpanAnnotator {
             $('.whitespace', $span).removeClass('whitespace-hidden');
 
             if (spanAnnotations.length > 0) {
-                const content = $span.data('content');
+                const content = String($span.data('content'));
                 const isLastInAnyAnnotation = spanAnnotations.some(ann =>
                     position + content.length >= ann.start + ann.text.length);
                 const hasMultipleAnnotations = spanAnnotations.length > 1;