Update app.py

app.py

@@ -24,30 +24,24 @@ def text2story(text):
     prompt = f"Write a short, happy, and fun story for kids aged 3-10. " \
              f"The story should be about: {text}. " \
              "Make it cheerful, imaginative, and suitable for young children. " \
-             "Avoid any scary or sad elements. The story must be exactly 95 words."
+             "Avoid any scary or sad elements. The story should be under 95 words."
     
     # Generate the story
-    story = text_generator(prompt, max_length=150, num_return_sequences=1)[0]["generated_text"]
+    story = text_generator(prompt, max_length=95, num_return_sequences=1)[0]["generated_text"]
     
     # Clean up the output to remove the prompt
     story = story.replace(prompt, "").strip()
     
-    # Ensure the story is exactly 95 words
+    # Ensure the story is under 95 words
     words = story.split()
     if len(words) > 95:
         story = " ".join(words[:95])
-    elif len(words) < 95:
-        # If the story is too short, pad it with a happy ending
-        story = " ".join(words) + " They all laughed and played together, making it the best day ever!"
-        words = story.split()
-        if len(words) > 95:
-            story = " ".join(words[:95])
     
     return story
 
 # Function to convert text to audio using gTTS
 def text2audio(story_text):
-    audio_file = "kids_playing_audio.wav"  # Use the correct file name
+    audio_file = "kids_playing_audio.wav"
     tts = gTTS(story_text, lang="en")
     tts.save(audio_file)
     return audio_file