Update app.py

app.py

@@ -31,47 +31,65 @@ def main():
         <h4 style="color:#4ECDC4;">Upload any photo to Get a fairy tale!</h4>
     </div>
     """, unsafe_allow_html=True)
+    
+    # 新增持久化容器（关键修改）
+    result_container = st.container()  # 用于存放所有生成内容
+    
     uploaded_file = st.file_uploader("🌈 Choose your magic picture...", type=["jpg", "png"])
 
     if uploaded_file is not None:
-        # 保存上传文件（原有逻辑）
+        # 清空旧内容（可选）
+        result_container.empty()
+        
+        # 保存上传文件
         bytes_data = uploaded_file.getvalue()
         with open(uploaded_file.name, "wb") as file:
             file.write(bytes_data)
-        st.image(uploaded_file, caption="Your Magic Picture ✨", use_container_width=True)
+            
+        # 显示图片在持久容器内
+        with result_container:
+            st.image(uploaded_file, caption="Your Magic Picture ✨", use_container_width=True)
 
-        # 初始化状态容器（修复点：移出if块）
-        status_container = st.empty()  # 移动到此处
+        # 初始化进度组件
+        status_container = st.empty()
         progress_bar = st.progress(0)
 
         # Stage 1: Image to Text
-        with status_container.status("🔮 **Step 1/3**: Decoding picture magic...", expanded=True) as status:  # 保持缩进
+        with status_container.status("🔮 **Step 1/3**: Decoding picture magic...", expanded=True):
             progress_bar.progress(33)
             scenario = img2text(uploaded_file.name)
-            status.update(label="✅ Picture decoded!", state="complete")
-            st.write(f"**What I see:** {scenario}")
+            with result_container:  # 将结果写入持久容器
+                st.success("🔍 **What I see:**")
+                st.write(scenario)
+            progress_bar.progress(33)
 
-        #Stage 2: Text to Story
-        with status_container.status("📚 **Step 2/3**: Writing your fairy tale...", expanded=True) as status:
+        # Stage 2: Text to Story
+        with status_container.status("📚 **Step 2/3**: Writing your fairy tale..."):
             progress_bar.progress(66)
             story = text2story(scenario)
-            status.update(label="✅ Story created!", state="complete")
-            st.write(f"**Your Story:**\n{story}")
+            with result_container:  # 持久化故事内容
+                st.success("📖 **Your Story:**")
+                st.markdown(f"""
+                <div style="
+                    background: #f0f9ff;
+                    padding: 1rem;
+                    border-radius: 10px;
+                    font-size: 18px;
+                ">{story}</div>
+                """, unsafe_allow_html=True)
+            progress_bar.progress(66)
 
-        #Stage 3: Story to Audio data
-        with status_container.status("🎵 **Step 3/3**: Adding magic music...", expanded=True) as status:
+        # Stage 3: Audio
+        with status_container.status("🎵 **Step 3/3**: Adding magic music..."):
             progress_bar.progress(100)
             audio_data = text2audio(story)
-            status.update(label="✅ All ready!", state="complete")
-
-
-            # 自动播放（移除按钮）
-            st.audio(audio_data['audio'], 
-                    format="audio/wav",
-                    start_time=0,
-                    sample_rate=audio_data['sampling_rate'],
-                    autoplay=True)  # 新增自动播放参数
-
+            with result_container:  # 持久化音频控件
+                st.success("🎧 **Listen to your story!**")
+                st.audio(audio_data['audio'], 
+                        format="audio/wav",
+                        start_time=0,
+                        sample_rate=audio_data['sampling_rate'],
+                        autoplay=True)
 
 if __name__ == "__main__":
     main()