Update app.py

app.py

@@ -42,18 +42,35 @@ def generate_wav_header(sample_rate: int, num_channels: int, sample_width: int,
 
 def custom_split_text(text: str) -> list:
     """
-    Custom splitting: split text into chunks where each chunk doubles in size.
+    Custom splitting:
+      - Start with a chunk size of 2 words.
+      - For each chunk, if a period (".") is found in any word (except if it’s the very last word),
+        then split the chunk at that word (include words up to that word).
+      - Otherwise, use the current chunk size.
+      - For subsequent chunks, increase the chunk size by 2.
+      - If there are fewer than the desired number of words for a full chunk, add all remaining words.
     """
     words = text.split()
     chunks = []
-    chunk_size = 1
+    chunk_size = 2
     start = 0
     while start < len(words):
-        end = start + chunk_size
-        chunk = " ".join(words[start:end])
-        chunks.append(chunk)
-        start = end
-        chunk_size *= 2  # double the chunk size for the next iteration
+        candidate_end = start + chunk_size
+        if candidate_end > len(words):
+            candidate_end = len(words)
+        chunk_words = words[start:candidate_end]
+        # Look for a period in any word except the last one.
+        split_index = None
+        for i in range(len(chunk_words) - 1):
+            if '.' in chunk_words[i]:
+                split_index = i
+                break
+        if split_index is not None:
+            candidate_end = start + split_index + 1
+            chunk_words = words[start:candidate_end]
+        chunks.append(" ".join(chunk_words))
+        start = candidate_end
+        chunk_size += 2  # Increase the chunk size by 2 for the next iteration.
     return chunks