Update app.py

app.py

@@ -5,6 +5,7 @@ from huggingface_hub import InferenceApi
 from gtts import gTTS
 import tempfile
 
+# —––––––– Page Config —–––––––
 st.set_page_config(page_title="Magic Story Generator", layout="centered")
 st.title("📖✨ Turn Images into Children's Stories")
 
@@ -19,6 +20,7 @@ def load_clients():
 
 caption_client, story_client = load_clients()
 
+# —––––––– Helper: Generate Caption —–––––––
 def generate_caption(img):
     """
     Runs the BLIP caption model on a PIL.Image and returns the generated text.
@@ -34,6 +36,7 @@ def generate_caption(img):
         st.error(f"Caption generation error: {e}")
         return ""
 
+# —––––––– Helper: Process Image —–––––––
 def process_image(uploaded_file):
     try:
         img = Image.open(uploaded_file).convert("RGB")
@@ -44,19 +47,21 @@ def process_image(uploaded_file):
         st.error(f"Image processing error: {e}")
         st.stop()
 
+# —––––––– Main App Flow —–––––––
 uploaded = st.file_uploader("Upload an image:", type=["jpg", "png", "jpeg"])
 if uploaded:
     img = process_image(uploaded)
     st.image(img, use_container_width=True)
 
-    with st.spinner("Generating caption..."):
+    # Generate Caption
+    with st.spinner("🔍 Discovering image secrets..."):
         caption = generate_caption(img)
         if not caption:
             st.error("😢 Couldn't understand this image. Try another one!")
             st.stop()
     st.success(f"**Caption:** {caption}")
 
-    # Story Generation Prompt
+    # Prepare Story Prompt
     story_prompt = (
         f"Image description: {caption}\n\n"
         "Write a 50-100 word children's story that:\n"
@@ -72,22 +77,19 @@ if uploaded:
         try:
             story_response = story_client(
                 story_prompt,
-                max_new_tokens=200,
-                temperature=0.8,
-                top_p=0.95,
-                repetition_penalty=1.15,
-                do_sample=True,
-                no_repeat_ngram_size=2
+                parameters={
+                    "max_new_tokens":       200,
+                    "temperature":          0.8,
+                    "top_p":                0.95,
+                    "repetition_penalty":   1.15,
+                    "do_sample":            True,
+                    "no_repeat_ngram_size": 2
+                }
             )
-            
-            # Process response
-            full_text = story_response[0]['generated_text']
+            full_text = story_response[0].get("generated_text", "")
             story = full_text.split("Story:")[-1].strip()
-            
-            # Ensure clean ending
             if "." in story:
                 story = story.rsplit(".", 1)[0] + "."
-            
         except Exception as e:
             st.error(f"🚨 Story magic failed: {str(e)}")
             st.stop()
@@ -106,4 +108,6 @@ if uploaded:
         except Exception as e:
             st.warning("⚠️ Couldn't make audio version: " + str(e))
 
+# Footer
 st.markdown("---\n*Made with ❤️ by your friendly story wizard*")
+