Update app.py

app.py

@@ -44,7 +44,7 @@ else:
         img.save(buf, format="PNG")
         cap_out = caption_client(data=buf.getvalue())
 
-        # Unwrap list/dict properly
+        # Handle caption response
         if isinstance(cap_out, list) and cap_out:
             cap_text = cap_out[0].get("generated_text", "").strip()
         elif isinstance(cap_out, dict):
@@ -58,7 +58,7 @@ else:
 
     st.markdown(f"**Caption:** {cap_text}")
 
-    # 3) Build prompt
+    # 3) Build story prompt
     prompt = (
         f"Here’s an image description: “{cap_text}”.\n\n"
         "Write an 80–100 word playful story for 3–10 year-old children that:\n"
@@ -68,26 +68,31 @@ else:
         "Story:"
     )
 
-    # 4) Generate story via HF Inference API (use `params`)
+    # 4) Generate story with CORRECT parameters
     with st.spinner("✍️ Generating story..."):
-        story_out = story_client(
-            inputs=prompt,
-            parameters={             # ← must be `params`, not `parameters`
-                "max_new_tokens": 120,
-                "do_sample": True,
-                "temperature": 0.7,
-                "top_p": 0.9,
-                "top_k": 50,
-                "repetition_penalty": 1.2,
-                "no_repeat_ngram_size": 3
-            }
-        )
-        if isinstance(story_out, list) and story_out:
-            story = story_out[0].get("generated_text", "").strip()
-        elif isinstance(story_out, dict):
-            story = story_out.get("generated_text", "").strip()
-        else:
-            story = str(story_out).strip()
+        try:
+            story_out = story_client(
+                inputs=prompt,
+                max_new_tokens=150,
+                do_sample=True,
+                temperature=0.8,
+                top_p=0.9,
+                repetition_penalty=1.1,
+                no_repeat_ngram_size=2
+            )
+            
+            # Handle different response formats
+            if isinstance(story_out, list):
+                story_text = story_out[0].get("generated_text", "")
+            else:
+                story_text = story_out.get("generated_text", "")
+            
+            # Clean up the output
+            story = story_text.split("Story:")[-1].strip()
+
+        except Exception as e:
+            st.error(f"🚨 Story generation failed: {str(e)}")
+            st.stop()
 
     if not story:
         st.error("😕 Couldn’t generate a story. Please try again!")
@@ -96,11 +101,14 @@ else:
     st.markdown("**Story:**")
     st.write(story)
 
-    # 5) Text-to-Speech via gTTS
+    # 5) Text-to-Speech
     with st.spinner("🔊 Converting to speech..."):
-        tts = gTTS(text=story, lang="en")
-        tmp = tempfile.NamedTemporaryFile(suffix=".mp3", delete=False)
-        tts.write_to_fp(tmp)
-        tmp.flush()
-    st.audio(tmp.name, format="audio/mp3")
+        try:
+            tts = gTTS(text=story, lang="en")
+            with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp:
+                tts.write_to_fp(tmp)
+                tmp.seek(0)
+                st.audio(tmp.name, format="audio/mp3")
+        except Exception as e:
+            st.error(f"🔇 Audio conversion failed: {str(e)}")