Update app.py

app.py

@@ -33,6 +33,7 @@ def text2audio(story_text):
     tts.save(audio_file.name)
     return audio_file.name
 
+    
 def main():
     st.set_page_config(page_title="Image to Story")
     st.header("Upload Your Image")
@@ -40,29 +41,44 @@ def main():
     uploaded_file = st.file_uploader("Choose image", type=["jpg", "png", "jpeg"])
     
     if uploaded_file:
-        # 注意：从这里开始统一缩进（4个空格）
         temp_img = os.path.join(tempfile.gettempdir(), uploaded_file.name)
         with open(temp_img, "wb") as f:
             f.write(uploaded_file.getvalue())
         
         st.image(uploaded_file)
         
-        # 生成描述（确保缩进一致）
-        scenario = img2text(temp_img)  # 这是错误指向的行
+        # 图片转文字处理状态
+        with st.status("🖼️ Processing img2text...", expanded=True) as status:
+            scenario = img2text(temp_img)
+            status.update(label="✅ Image analysis completed!", state="complete")
+        
         st.write("Image Caption:", scenario)
         
-        # 生成故事
-        story = text2story(scenario)
+        # 故事生成处理状态
+        with st.status("📝 Generating a story...", expanded=True) as status:
+            story = text2story(scenario)
+            status.update(label="📖 Story generation completed!", state="complete")
+        
         st.subheader("Story")
         st.write(story)
         
-        # 生成音频
-        audio_path = text2audio(story)
+        # 音频生成处理状态
+        with st.status("🔊 Converting to audio...", expanded=True) as status:
+            audio_path = text2audio(story)
+            status.update(label="🎧 Audio conversion completed!", state="complete")
+        
         st.audio(audio_path)
         
-        # 清理文件
         os.unlink(temp_img)
         os.unlink(audio_path)
 
+
+        if st.button("🎵 Play Audio Story"):
+            st.audio(
+            audio_data["audio"],
+            format="audio/mp3",
+            start_time=0,
+            sample_rate=audio_data["sampling_rate"]
+            )
 if __name__ == "__main__":
     main()