Update app.py

app.py

@@ -78,6 +78,8 @@ if uploaded_image:
     story_prompt = (
         f"<|im_start|>system\n"
         f"You're a children's author. Create a short story (100-150 words) based on: {image_caption}\n"
+        f"Use simple language and include a moral lesson.<|im_end|>\n"
+        f"<|im_start|>assistant\n"
     )
 
     # Generate story with progress
@@ -111,4 +113,40 @@ if uploaded_image:
             
             # Format story text
             sentences = []
-            for sent in re
+            for sent in re.split(r'(?<=[.!?]) +', clean_story):
+                sent = sent.strip()
+                if sent:
+                    if len(sent) > 1 and not sent.endswith(('.','!','?')):
+                        sent += '.'
+                    sentences.append(sent[0].upper() + sent[1:])
+            
+            final_story = ' '.join(sentences)[:600]  # Limit length
+
+            update_progress(5)
+            time.sleep(0.5)  # Final progress pause
+
+    except Exception as e:
+        st.error(f"❌ Story generation failed: {str(e)}")
+        st.stop()
+
+    finally:
+        progress_bar.empty()
+        status_text.empty()
+
+    # Display story
+    st.subheader("✨ Your Magical Story")
+    st.write(final_story)
+
+    # Audio conversion
+    with st.spinner("🔊 Creating audio version..."):
+        try:
+            audio = gTTS(text=final_story, lang="en", slow=False)
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
+                audio.save(tmp_file.name)
+                st.audio(tmp_file.name, format="audio/mp3")
+        except Exception as e:
+            st.error(f"❌ Audio conversion failed: {str(e)}")
+
+# Footer
+st.markdown("---")
+st.markdown("📚 Made with ♥ by The Story Wizard • [Report Issues](https://example.com)")