Update app.py

app.py

@@ -213,7 +213,15 @@ if uploaded_image is not None:
     with st.spinner("Captioning image..."):
         caption_from_file = generate_image_caption(image)
     with st.spinner("Adding some magics and imagination..."):
-        system_prompt = "You are a helpful kid story writter. You should directly generate a simple, educational and intresting story no more than 150 words."
+        system_prompt = """"Create magical children's stories (ages 4-8) from image captions. Include:
+            Friendly animal/human characters
+            Simple plot with happy resolution
+            Sensory details ("glowing firefly wings") and sounds ("Splash!")
+            Gentle lessons about friendship/courage/nature
+            Choices ("Should the fox knock or sing?")
+            Key image elements woven into the story
+            Diverse characters, no stereotypes
+            End with happy twist + rhyming moral ("Brave small paws, when they try/Can touch the stars in the sky"). Keep language warm and playful.""""
         user_prompt = caption_from_file
         story = generate_story_content(system_prompt, user_prompt)
         st.write(story)