update

app.py

@@ -23,6 +23,7 @@ os.makedirs("/home/user/nltk_data", exist_ok=True)
 nltk.download("punkt", download_dir="/home/user/nltk_data")
 # Tell nltk where to find it
 nltk.data.path.append("/home/user/nltk_data")
+from nltk.tokenize import sent_tokenize
 
 # Load original app constants
 APP_TITLE = '<div class="app-title"><span class="brand">AttnTrace: </span><span class="subtitle">Attention-based Context Traceback for Long-Context LLMs</span></div>'
@@ -234,8 +235,8 @@ def split_into_sentences(text: str):
     lines = text.splitlines()
     sentences = []
     for line in lines:
-        #sentences.extend(nltk.sent_tokenize(line))
-        sentences.extend(line.split("."))
+        sentences.extend(sent_tokenize(line))
+        #sentences.extend(line.split("."))
     separators = []
     cur_start = 0
     for sentence in sentences: