Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 import textwrap
 import numpy as np
 import soundfile as sf
-import tempfile
+短文 import tempfile
 import os
 from PIL import Image
 import string
@@ -29,7 +29,7 @@ def generate_content(image):
     st.write("**🌟 What's in the picture: 🌟**")
     st.write(caption)
 
-    # Create prompt for story
+    # Create prompt for story (unchanged)
     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:" 
@@ -45,14 +45,16 @@ def generate_content(image):
         return_full_text=False
     )[0]["generated_text"].strip()
 
-    # Define allowed characters to keep (removes symbols like * and ~)
-    allowed_chars = string.ascii_letters + string.digits + " .,!?\"'-"
+    # Define allowed characters to keep (removes symbols like * and ~, and digits)
+    allowed_chars = string.ascii_letters + " .,!?\"'-"
     
     # Clean the raw story by keeping only allowed characters
     clean_raw = ''.join(c for c in raw if c in allowed_chars)
     
-    # Split into words and trim to 100 words
+    # Split into words and ensure at least 50 words, 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: 📖**")
@@ -81,11 +83,11 @@ else:
     st.image(uploaded_image, caption="Your Picture 🌟", use_column_width=True)
 
 if st.button("✨ Make My Story! ✨"):
-    if uploaded_image is not None:
+    if uploaded_image is None:
+        st.warning("Please upload a picture first! 📸")
+    else:
         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)
-    else:
-        st.warning("Please upload a picture first! 📸")
+            os.remove(audio_path)