Update app.py

app.py

@@ -14,22 +14,21 @@ openai_api_key = os.environ.get("OPENAI_API_KEY")
 
 llm = OpenAI(temperature=0.9)
 
-def split_text(text, max_length):
-    chunks = []
-    current_chunk = ''
+def split_text_into_sentences(text):
+    sentences = []
+    current_sentence = ''
     words = text.split()
 
     for word in words:
-        if len(current_chunk) + len(word) <= max_length:
-            current_chunk += ' ' + word
-        else:
-            chunks.append(current_chunk.strip())
-            current_chunk = word
+        current_sentence += ' ' + word
+        if word.endswith('.'):
+            sentences.append(current_sentence.strip())
+            current_sentence = ''
 
-    if current_chunk:
-        chunks.append(current_chunk.strip())
+    if current_sentence:
+        sentences.append(current_sentence.strip())
 
-    return chunks
+    return sentences
 
 def join_wav_files(input_files, output_file):
     # Open the first input file to get its parameters
@@ -53,7 +52,9 @@ def generate_story(text):
         template=""" 
         You are a fun and seasoned storyteller. 
         Generate a short story for a 5 years old audience about {text}.
-                 """
+        Use short sentences. The story is not too long, but not too short either.
+        Always finish your story with "The End".
+        """
     )
     story = LLMChain(llm=llm, prompt=prompt)
     story_result = story.run(text=text)
@@ -64,8 +65,8 @@ def generate_story(text):
     —
     """)
     input_waves = []
-    max_length = 250
-    text_chunks = split_text(story_result, max_length)
+    
+    text_chunks = split_text_into_sentences(story_result)
     for chunk in text_chunks:
         print(chunk)
         result = eleven.predict(