Update app.py

app.py

@@ -28,16 +28,24 @@ def text2story(text):
         text (str): Text caption generated from the image.
     
     Returns:
-        str: Generated story suitable for kids aged 3-10.
+        str: Generated story suitable for kids aged 3-10, within 95 words.
     """
     # Add a prompt to ensure the story is kid-friendly and happy
-    prompt = f"Write a happy and fun story for kids aged 3-10 based on the following scenario: {text}."
+    prompt = f"Write a happy, fun and complete story for kids aged 3-10 based on the following scenario: {text}. Keep the story under 120 words."
     
     # Load the text generation model
     story_generator = pipeline("text-generation", model="aspis/gpt2-genre-story-generation")
     
     # Generate the story
-    story = story_generator(prompt, max_length=95, num_return_sequences=1)[0]["generated_text"]
+    story = story_generator(prompt, max_length=150, num_return_sequences=1)[0]["generated_text"]
+    
+    # Remove the prompt from the generated story
+    story = story.replace(prompt, "").strip()
+    
+    # Ensure the story is within 95 words
+    story_words = story.split()
+    if len(story_words) > 98:
+        story = " ".join(story_words[:98])
     
     return story
 
@@ -60,12 +68,21 @@ def text2audio(story_text):
     return audio_file
 
 # Main application
-st.set_page_config(page_title="Your Image to Audio Story",
-                   page_icon="🦜")
-st.header("Turn Your Image into a Fun Audio Story! 🎉")
+st.set_page_config(page_title="Picture Stories 🎨📖", page_icon="🦄")
+st.title("Picture Stories 🎨📖")
+st.markdown("### Turn your pictures into fun stories and listen to them! 🎉")
+
+# Instructions for kids
+st.markdown("""
+**How to use this app:**
+1. **Upload a picture** of something fun, like your favorite toy, a park, or your pet.
+2. Wait for the app to **create a story** from your picture.
+3. **Listen to the story** by clicking the "Play Audio" button.
+4. Enjoy your fun story! 🎧
+""")
 
 # Upload image
-uploaded_file = st.file_uploader("Select an Image...", type=["jpg", "jpeg", "png"])
+uploaded_file = st.file_uploader("📷 **Upload your picture here!**", type=["jpg", "jpeg", "png"])
 
 if uploaded_file is not None:
     # Save the uploaded file
@@ -74,24 +91,24 @@ if uploaded_file is not None:
         file.write(bytes_data)
 
     # Display the uploaded image
-    st.image(uploaded_file, caption="Uploaded Image", use_column_width=True)
+    st.image(uploaded_file, caption="Your awesome picture!", use_container_width=True)
 
     # Stage 1: Image to Text
-    st.text('Processing image to text...')
+    st.text('✨ Turning your picture into words...')
     scenario = img2text(uploaded_file.name)
-    st.write("**Caption:**", scenario)
+    st.write("**What we see:**", scenario)
 
     # Stage 2: Text to Story
-    st.text('Generating a fun story for kids...')
+    st.text('📖 Creating a fun story for you...')
     story = text2story(scenario)
-    st.write("**Story:**", story)
+    st.write("**Your story:**", story)
 
     # Stage 3: Story to Audio
-    st.text('Converting story to audio...')
+    st.text('🎧 Turning your story into audio...')
     audio_file = text2audio(story)
 
     # Play button for the generated audio
-    if st.button("Play Audio"):
+    if st.button("🎵 **Play Audio**"):
         st.audio(audio_file, format="audio/mp3")
 
     # Clean up the generated audio file