Update app.py

app.py

@@ -159,22 +159,20 @@ def main():
         # Generation button
         if st.button("✨ Generate Story & Audio"):
             with st.spinner("Processing your request..."):
-                # Generate outputs
-                caption = generate_image_caption(image)
-                sys_prompt = "You are a fantasy writer. Create a 100-word adventure story about "
-                story = generate_story_content(sys_prompt, caption)
-                speech = generate_audio_from_story(story)
-                
-                # Display results
+                # Generate outputs and Display results
                 with col2:
                     st.subheader("🔍 Scene Description")
+                    caption = generate_image_caption(image)
                     st.write(caption)
                     
                     st.subheader("📖 Generated Story")
+                    sys_prompt = "You are a fantasy writer. Create a 100-word adventure story about "
+                    story = generate_story_content(sys_prompt, caption)
                     st.write(story)
                     
                     st.subheader("🔊 Audio Playback")
-                    st.audio(speech["audio"], sample_rate=audio_file["sampling_rate"], format='audio/wav')
+                    speech = generate_audio_from_story(story)
+                    st.audio(speech["audio"], sample_rate=speech["sampling_rate"], format='audio/wav')
                     
                     # Download button
                     with open(speech["audio"], "rb") as audio_file: