Update app.py

app.py

@@ -28,7 +28,8 @@ def generate_content(image):
     
     # Generate a caption based on the image content
     caption = captioner(pil_image)[0]["generated_text"]
-    st.write("**Caption:**", caption)
+    st.write("**🌟 What's in the picture: 🌟**")
+    st.write(caption)
 
     # Create a prompt for generating a children's story
     prompt = (
@@ -50,7 +51,8 @@ def generate_content(image):
     # Trim the generated story to a maximum of 100 words
     words = raw.split()
     story = " ".join(words[:100])
-    st.write("**Story:**", story)
+    st.write("**📖 Your funny story: 📖**")
+    st.write(story)
 
     # Split the story into chunks of 200 characters for text-to-speech processing
     chunks = textwrap.wrap(story, width=200)
@@ -66,26 +68,26 @@ def generate_content(image):
     return caption, story, temp_file_path
 
 # Streamlit UI for the application
-st.title("Image to Children's Story and Audio")
-st.markdown("""
-    Upload an image below to generate a caption, a funny children's story, 
-    and an audio narration based on the image. The story will be tailored 
-    for children aged 3-10.
-""")
+st.title("✨ Magic Story Maker ✨")
+st.markdown("Upload a picture to make a funny story and hear it too! 📸")
 
 # File uploader for image input
-uploaded_image = st.file_uploader("Choose an image", type=["jpg", "jpeg", "png"], help="Supported formats: JPG, JPEG, PNG")
+uploaded_image = st.file_uploader("Choose your picture", type=["jpg", "jpeg", "png"], help="Pick a photo to start the magic!")
 
-if uploaded_image is not None:
-    # Display the uploaded image with a caption
-    st.image(uploaded_image, caption="Uploaded Image", use_column_width=True)
-    
-    # Button to trigger content generation
-    if st.button("Generate Story and Audio", help="Click to create the story and audio"):
-        # Show a spinner while content is being generated
-        with st.spinner("Generating your story and audio narration..."):
+# Placeholder image URL (replace with an actual URL of a child-friendly image)
+placeholder_url = "https://example.com/placeholder_image.jpg"
+
+if uploaded_image is None:
+    st.image(placeholder_url, caption="Upload your picture here! 📷", use_column_width=True)
+else:
+    st.image(uploaded_image, caption="Your Picture 🌟", use_column_width=True)
+
+if st.button("✨ Make My Story! ✨", help="Click to create your magic story"):
+    if uploaded_image is not None:
+        with st.spinner("🔮 Creating your magical story..."):
             caption, story, audio_path = generate_content(uploaded_image)
-            # Display the audio player with the generated narration
+            st.success("🎉 Your story is ready! 🎉")
             st.audio(audio_path, format="audio/wav")
-            # Remove the temporary audio file after use
-            os.remove(audio_path)
+            os.remove(audio_path)
+    else:
+        st.warning("Please upload a picture first! 📸")