Update app.py

app.py

@@ -2,77 +2,72 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 
-# Function to generate a caption from an uploaded image
-# Uses the Hugging Face pipeline with the Salesforce BLIP model to analyze the image and produce a descriptive caption
+# Creates a short description from an uploaded picture using a smart model
 def generate_caption(image):
+    # Loads the BLIP model to analyze the image and describe it
     image_to_text = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
     caption = image_to_text(image)[0]["generated_text"]
     return caption
 
-# Function to generate a story based on the image caption
-# Uses the pranavpsv/genre-story-generator-v2 model to create a creative story inspired by the caption
+# Turns the picture’s description into a fun story
 def generate_story(caption):
+    # Uses a story-generating model to craft a tale based on the description
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story = pipe(caption, max_length=150, do_sample=True)[0]['generated_text']  # Added max_length for shorter, kid-friendly stories
+    story = pipe(caption)[0]['generated_text']
     return story
 
-# Function to convert the story into audio
-# Uses the facebook/mms-tts-eng model to generate a spoken version of the story
+# Makes the story into a spoken audio clip
 def generate_audio(story):
+    # Employs a text-to-speech model to read the story aloud
     pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
     audio = pipe(story)
     return audio
 
-# Streamlit UI: Creating a fun and interactive interface for kids
-# The app is designed to be simple, colorful, and engaging for young users
+# Streamlit UI: Builds a colorful, kid-friendly interface for storytelling fun
 
-# Display a vibrant, centered title with emojis to grab attention
-st.markdown("<h1 style='text-align: center; color: purple;'>🌟 Picture to Story Magic! 🌟</h1>", unsafe_allow_html=True)
+# Shows a big, exciting title to welcome young users
+st.markdown("<h1 style='text-align: center; color: orange;'>✨ Magic Story Creator! ✨</h1>", unsafe_allow_html=True)
 
-# Add a welcoming description to explain the app’s purpose in kid-friendly language
+# Adds a fun message to explain what the app does
 st.markdown(
     """
     <p style='text-align: center; font-size: 16px;'>
-    🎉 Hello, young adventurers! Upload a picture, and I’ll create a magical story just for you! 
-    This app is perfect for kids aged 3-10 who love fun and creative storytelling. 
-    Let’s make some story magic together! 🧙‍♂️
+    🦄 Hey there, story explorers! Share a picture, and I’ll weave a super cool story for you! 
+    This app is made for kids aged 3-10 who love adventures and imagination. 
+    Ready to dive in? Let’s go! 🚀
     </p>
     """, 
     unsafe_allow_html=True
 )
 
-# File uploader for images, with a playful prompt
-uploaded_file = st.file_uploader("📸 Pick a cool picture!", type=["png", "jpg", "jpeg"], help="Choose a fun image to start your story adventure!")
+# Lets kids upload a picture to start their story
+uploaded_file = st.file_uploader("🖼️ Share a fun picture!", type=["png", "jpg", "jpeg"], help="Pick a picture to spark a story!")
 
-# Check if an image has been uploaded
+# Checks if a picture has been uploaded
 if uploaded_file is not None:
-    # Load and display the uploaded image
-    # The image is opened using Pillow and displayed with a caption
+    # Opens and shows the picture on the screen
     image = Image.open(uploaded_file)
-    st.image(image, caption="Your Awesome Picture! 😊", use_column_width=True)  # Changed to use_column_width for better scaling
+    st.image(image, caption="Your Super Cool Picture! 😎", use_column_width=True)
 
-    # Generate and display the image caption
-    # The caption describes what’s in the image, serving as the story’s starting point
-    with st.spinner("🔍 Looking at your picture..."):
+    # Creates a description of the picture
+    with st.spinner("🕵️‍♂️ Exploring your picture..."):
         image_caption = generate_caption(image)
     
-    st.subheader("✨ What’s in Your Picture?")
+    st.subheader("🌈 What’s Your Picture About?")
     st.write(f"{image_caption}")
-    st.markdown("<p style='font-size: 14px;'>This is what I see in your picture! It’s the start of our story! 📖</p>", unsafe_allow_html=True)
+    st.markdown("<p style='font-size: 14px;'>This is the start of your adventure! Let’s build a story! 🏰</p>", unsafe_allow_html=True)
 
-    # Generate and display the story
-    # The story is created based on the caption, designed to be short and engaging for kids
-    with st.spinner("🖌️ Writing a magical story..."):
+    # Makes a story from the picture’s description
+    with st.spinner("✍️ Crafting a fantastic tale..."):
         story_telling = generate_story(image_caption)
     
-    st.subheader("📚 Your Magical Story!")
+    st.subheader("🎉 Your Awesome Story!")
     st.write(f"{story_telling}")
-    st.markdown("<p style='font-size: 14px;'>Wow, what a fun story! Want to hear it out loud? Click below! 🎶</p>", unsafe_allow_html=True)
+    st.markdown("<p style='font-size: 14px;'>How cool is that? Want to hear your story come to life? Tap below! 🎤</p>", unsafe_allow_html=True)
 
-    # Generate and play audio for the story
-    # Audio is generated only when the user clicks the button, keeping the app responsive
-    if st.button("🎙️ Listen to Your Story!"):
-        with st.spinner("🎵 Turning your story into sound..."):
+    # Turns the story into audio when requested
+    if st.button("🔊 Hear Your Story!"):
+        with st.spinner("🎧 Creating story audio..."):
             audio = generate_audio(story_telling)
         st.audio(
             audio['audio'], 
@@ -80,15 +75,15 @@ if uploaded_file is not None:
             start_time=0, 
             sample_rate=audio['sampling_rate']
         )
-        st.markdown("<p style='font-size: 14px;'>Amazing! You can listen to your story anytime! 😄</p>", unsafe_allow_html=True)
+        st.markdown("<p style='font-size: 14px;'>Wow, your story sounds amazing! Listen again anytime! 🌟</p>", unsafe_allow_html=True)
 
-# Add a footer to encourage users to try again
+# Adds a cheerful note to encourage more storytelling
 st.markdown(
     """
     <hr>
     <p style='text-align: center; font-size: 14px;'>
-    🖼️ Want another adventure? Upload a new picture and create a brand-new story! 
-    Made with love for curious kids! ❤️
+    📸 Ready for another tale? Upload a new picture for a fresh adventure! 
+    Created with sparkles for awesome kids! 💖
     </p>
     """, 
     unsafe_allow_html=True