Update app.py

app.py

@@ -21,20 +21,33 @@ def text2story(text):
     text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
     
     # Add a strong prompt to guide the model
-    prompt = f"Write a short, happy, and fun story for kids aged 3-10 based on the following description: {text}. " \
-             "The story should be cheerful, imaginative, and suitable for young children. " \
-             "Avoid any scary or sad elements. Keep the story under 95 words and make sure it has a clear beginning, middle, and end."
+    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."
     
     # Generate the story
-    story = text_generator(prompt, max_length=95, num_return_sequences=1)[0]["generated_text"]
+    story = text_generator(prompt, max_length=150, num_return_sequences=1)[0]["generated_text"]
     
-    # Clean up the output to remove the prompt (if necessary)
+    # Clean up the output to remove the prompt
     story = story.replace(prompt, "").strip()
+    
+    # Ensure the story is exactly 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 = "story_audio.mp3"
+    audio_file = "kids_playing_audio.wav"  # Use the correct file name
     tts = gTTS(story_text, lang="en")
     tts.save(audio_file)
     return audio_file
@@ -77,7 +90,7 @@ if uploaded_file is not None:
 
     # Play button for audio
     if st.session_state.audio_file and st.button("🎧 Listen to the Story"):
-        st.audio(st.session_state.audio_file, format="audio/mp3")
+        st.audio(st.session_state.audio_file, format="audio/wav")
 
     # Clean up the generated audio file
     if st.session_state.audio_file and os.path.exists(st.session_state.audio_file):