Update app.py

app.py

@@ -2,89 +2,54 @@ import streamlit as st
 from transformers import pipeline
 from PIL import Image
 
-# Creates a short description from an uploaded picture using a smart model
+# Makes a short description of a picture using a clever model
 def generate_caption(image):
-    # Loads the BLIP model to analyze the image and describe it
+    # Sets up the BLIP model to look at the picture 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
 
-# Turns the picture’s description into a fun story
+# Creates a fun story from the picture’s description
 def generate_story(caption):
-    # Uses a story-generating model to craft a tale based on the description
+    # Uses a story model to write a tale based on the description
     pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
     story = pipe(caption)[0]['generated_text']
     return story
 
-# Makes the story into a spoken audio clip
+# Turns the story into a spoken audio clip
 def generate_audio(story):
-    # Employs a text-to-speech model to read the story aloud
+    # Uses a speech model to read the story out loud
     pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
     audio = pipe(story)
     return audio
 
-# Streamlit UI: Builds a colorful, kid-friendly interface for storytelling fun
+# Streamlit UI: Sets up a simple, fun interface for kids
 
-# 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)
+# Shows a bright title to excite young users
+st.title("🌟 Picture Story Magic! 🌟")
 
-# Adds a fun message to explain what the app does
-st.markdown(
-    """
-    <p style='text-align: center; font-size: 16px;'>
-    🦄 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
-)
+# Explains the app in a fun way
+st.write("Hey, kids! Upload a picture and get a cool story! Perfect for ages 3-10. Let’s make magic! 🪄")
 
-# 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!")
+# Lets kids upload a picture
+uploaded_file = st.file_uploader("📸 Pick a fun picture!", type=["png", "jpg", "jpeg"])
 
-# Checks if a picture has been uploaded
 if uploaded_file is not None:
-    # Opens and shows the picture on the screen
+    # Shows the uploaded picture
     image = Image.open(uploaded_file)
-    st.image(image, caption="Your Super Cool Picture! 😎", use_column_width=True)
-
-    # Creates a description of the picture
-    with st.spinner("🕵️‍♂️ Exploring your picture..."):
-        image_caption = generate_caption(image)
-    
-    st.subheader("🌈 What’s Your Picture About?")
-    st.write(f"{image_caption}")
-    st.markdown("<p style='font-size: 14px;'>This is the start of your adventure! Let’s build a story! 🏰</p>", unsafe_allow_html=True)
-
-    # Makes a story from the picture’s description
-    with st.spinner("✍️ Crafting a fantastic tale..."):
-        story_telling = generate_story(image_caption)
+    st.image(image, caption="Your Awesome Picture!", use_container_width=True)
     
-    st.subheader("🎉 Your Awesome Story!")
-    st.write(f"{story_telling}")
-    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)
-
-    # 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'], 
-            format="audio/wav", 
-            start_time=0, 
-            sample_rate=audio['sampling_rate']
-        )
-        st.markdown("<p style='font-size: 14px;'>Wow, your story sounds amazing! Listen again anytime! 🌟</p>", unsafe_allow_html=True)
-
-# Adds a cheerful note to encourage more storytelling
-st.markdown(
-    """
-    <hr>
-    <p style='text-align: center; font-size: 14px;'>
-    📸 Ready for another tale? Upload a new picture for a fresh adventure! 
-    Created with sparkles for awesome kids! 💖
-    </p>
-    """, 
-    unsafe_allow_html=True
-)
+    # Gets the picture’s description
+    image_caption = generate_caption(image)
+    st.subheader("What’s in Your Picture?")
+    st.write(image_caption)
+
+    # Makes a story from the description
+    story_telling = generate_story(image_caption)
+    st.subheader("Your Super Story!")
+    st.write(story_telling)
+
+    # Creates audio for the story
+    audio = generate_audio(story_telling)
+    if st.button("🎵 Listen to Your Story!"):
+        st.audio(audio['audio'], format="audio/wav", start_time=0, sample_rate=audio['sampling_rate'])