Update app.py

app.py

@@ -8,7 +8,6 @@ import tempfile
 import os
 from PIL import Image
 import string
-import re
 
 # Initialize pipelines with caching
 @st.cache_resource
@@ -30,35 +29,31 @@ def generate_content(image):
     st.write("**🌟 What's in the picture: 🌟**")
     st.write(caption)
 
-    # Create prompt for story (unchanged)
+    # Create prompt for story
     prompt = (
         f"Write a funny, warm children's story for ages 3-10, 50–100 words, "
-        f"Completely and precisely centered on this scene {caption}\nStory:" 
+        f"Completely and precisely centered on this scene {caption}\nStory:"
     )
     
     # Generate raw story
     raw = storyer(
         prompt,
         max_new_tokens=150,
+        min_new_tokens=50
         temperature=0.7,
         top_p=0.9,
         no_repeat_ngram_size=2,
         return_full_text=False
     )[0]["generated_text"].strip()
 
-    # Define allowed characters to keep (removes symbols like * ~ , - and digits)
-    allowed_chars = string.ascii_letters + " .!?\"'"
+    # Define allowed characters to keep (removes symbols like * and ~)
+    allowed_chars = string.ascii_letters + string.digits + " .,!?\"'-"
     
     # Clean the raw story by keeping only allowed characters
     clean_raw = ''.join(c for c in raw if c in allowed_chars)
     
-    # Remove multiple spaces and trailing punctuation
-    clean_raw = re.sub(r'\s+', ' ', clean_raw).strip('.!? ')
-    
-    # Split into words and ensure at least 50 words, trim to 100 words
+    # Split into words and trim to 100 words
     words = clean_raw.split()
-    if len(words) < 50:
-        words.extend("The children laughed and played happily, making new friends in the sunny park.".split())
     story = " ".join(words[:100])
     
     st.write("**📖 Your funny story: 📖**")
@@ -82,12 +77,16 @@ st.markdown("Upload a picture to make a funny story and hear it too! 📸")
 uploaded_image = st.file_uploader("Choose your picture", type=["jpg", "jpeg", "png"])
 
 if uploaded_image is None:
-    st.warning("Please upload a picture first! 📸")
+    st.image("https://example.com/placeholder_image.jpg", 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! ✨"):
+
+if st.button("✨ Make My Story! ✨"):
+    if uploaded_image is not None:
         with st.spinner("🔮 Creating your magical story..."):
             caption, story, audio_path = generate_content(uploaded_image)
             st.success("🎉 Your story is ready! 🎉")
             st.audio(audio_path, format="audio/wav")
-            os.remove(audio_path)
+            os.remove(audio_path)
+    else:
+        st.warning("Please upload a picture first! 📸")