Update app.py

app.py

@@ -51,15 +51,18 @@ raw = storyer(
 allowed_pattern = re.compile(r'[a-zA-Z0-9.,!?"\'-]+\b(?<!\b\w\b)')
 clean_raw = ' '.join(word for word in re.findall(allowed_pattern, raw) if len(word) > 1)
 
-# Split into words and trim to 100 words
-words = clean_raw.split()
-story = " ".join(words[:100])
+def generate_story(raw, caption, tts):  
+    # Split into words and trim to 100 words
+    words = raw.split()  
+    story = " ".join(words[:100])
+    
+    # Clean the story using clean_generated_story
+    story = clean_generated_story(raw)  
+    
+    # Display story in Streamlit
+    st.write("**📖 Your funny story: 📖**")
+    st.write(story)
     
-story = clean_generated_story(raw)
-st.write("**📖 Your funny story: 📖**")
-st.write(story)
-return story
-
     # Generate audio from cleaned story
     chunks = textwrap.wrap(story, width=200)
     audio = np.concatenate([tts(chunk)["audio"].squeeze() for chunk in chunks])