Update app.py

app.py

@@ -117,9 +117,39 @@ excluding the title, author name, source information, chapter number, annotation
     return resp.choices[0].message.content.strip()
 
 
-def split_pages(text, words=300):
-    w = text.split()
-    return [" ".join(w[i:i+words]) for i in range(0, len(w), words)] or [text]
+import re
+
+def split_pages(text, max_words=300):
+    """
+    文単位でページを分割する。
+    - 文の途中でページを分割しない
+    - max_words の上限を超えないようにする
+    """
+    # 文に分割（. ? ! のあとに改行やスペースが続くパターン）
+    sentences = re.split(r'(?<=[.!?])\s+', text.strip())
+    pages = []
+    current_page = []
+    current_word_count = 0
+
+    for sentence in sentences:
+        words = sentence.split()
+        sentence_len = len(words)
+
+        # 次の文を追加すると max_words を超える場合 → 新しいページを作る
+        if current_word_count + sentence_len > max_words:
+            if current_page:
+                pages.append(" ".join(current_page))
+            current_page = [sentence]
+            current_word_count = sentence_len
+        else:
+            current_page.append(sentence)
+            current_word_count += sentence_len
+
+    # 最後のページを追加
+    if current_page:
+        pages.append(" ".join(current_page))
+
+    return pages or [text]
 
 
 # ======================================================