Update app.py

app.py

@@ -72,7 +72,10 @@ def generate_tagged_sentence(sentence, entity_tags):
     for tag in entity_tags:
         start = tag['start']
         end = tag['end']
-        token = sentence[start - 1:end]  # Adjust for 0-based indexing
+        if end<len(sentence)-1:
+            token = sentence[start:end]  # Adjust for 0-based indexing
+        else:
+            token = sentence[start:end+1]
         tag_name = f"[{tag['entity_group']}]"
 
         tagged_tokens.append(f"{token} {tag_name}")
@@ -89,7 +92,10 @@ def replace_pp_with_pause(sentence, entity_tags):
     for tag in entity_tags:
         start = tag['start']
         end = tag['end']
-        token = sentence[start - 1:end]  # Adjust for 0-based indexing
+        if end<len(sentence)-1:
+            token = sentence[start:end]  # Adjust for 0-based indexing
+        else:
+            token = sentence[start:end+1]
         tag_name = f"[{tag['entity_group']}]"
 
         if tag['entity_group'] == 'PP':