Update app.py

app.py

@@ -8,6 +8,7 @@ import tempfile
 import os
 from PIL import Image
 import string
+import re
 
 # Initialize pipelines with caching
 @st.cache_resource
@@ -45,12 +46,15 @@ def generate_content(image):
         return_full_text=False
     )[0]["generated_text"].strip()
 
-    # Define allowed characters to keep (removes symbols like * and ~, and digits)
-    allowed_chars = string.ascii_letters + " .,!?\"'-"
+    # Define allowed characters to keep (removes symbols like * ~ , - 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)
     
+    # 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
     words = clean_raw.split()
     if len(words) < 50:
@@ -78,14 +82,10 @@ 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.image("https://example.com/placeholder_image.jpg", caption="Upload your picture here! 📷", use_column_width=True)
+    st.warning("Please upload a picture first! 📸")
 else:
     st.image(uploaded_image, caption="Your Picture 🌟", use_column_width=True)
-
-if st.button("✨ Make My Story! ✨"):
-    if uploaded_image is None:
-        st.warning("Please upload a picture first! 📸")
-    else:
+    if st.button("✨ Make My Story! ✨"):
         with st.spinner("🔮 Creating your magical story..."):
             caption, story, audio_path = generate_content(uploaded_image)
             st.success("🎉 Your story is ready! 🎉")