Update app.py

app.py

@@ -2,6 +2,26 @@
 import streamlit as st
 from transformers import pipeline
 
+st.markdown("""
+<style>
+@import url('https://fonts.googleapis.com/css2?family=Comic+Neue:wght@700&display=swap');
+.header {
+    background-image: url('https://i.imgur.com/6GEdzVp.png');
+    background-size: cover;
+    border-radius: 15px;
+    padding: 2rem;
+    text-align: center;
+    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+}
+.header h1 {
+    color: #FF6B6B;
+    font-family: 'Comic Neue', cursive;
+    font-size: 2.5rem;
+    text-shadow: 2px 2px #FFF;
+}
+</style>
+""", unsafe_allow_html=True)
+
 # function part
 # img2text
 def img2text(url):
@@ -31,65 +51,45 @@ 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)
-            
-        # 显示图片在持久容器内
-        with result_container:
-            st.image(uploaded_file, caption="Your Magic Picture ✨", use_container_width=True)
+        st.image(uploaded_file, caption="Your Magic Picture ✨", use_container_width=True)
 
-        # 初始化进度组件
-        status_container = st.empty()
+        # 初始化状态容器（修复点：移出if块）
+        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):
+        with status_container.status("🔮 **Step 1/3**: Decoding picture magic...", expanded=True) as status:  # 保持缩进
             progress_bar.progress(33)
             scenario = img2text(uploaded_file.name)
-            with result_container:  # 将结果写入持久容器
-                st.success("🔍 **What I see:**")
-                st.write(scenario)
-            progress_bar.progress(33)
+            status.update(label="✅ Picture decoded!", state="complete")
+            st.write(f"**What I see:** {scenario}")
 
-        # Stage 2: Text to Story
-        with status_container.status("📚 **Step 2/3**: Writing your fairy tale..."):
+        #Stage 2: Text to Story
+        with status_container.status("📚 **Step 2/3**: Writing your fairy tale...", expanded=True) as status:
             progress_bar.progress(66)
             story = text2story(scenario)
-            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)
+            status.update(label="✅ Story created!", state="complete")
+            st.write(f"**Your Story:**\n{story}")
 
-        # Stage 3: Audio
-        with status_container.status("🎵 **Step 3/3**: Adding magic music..."):
+        #Stage 3: Story to Audio data
+        with status_container.status("🎵 **Step 3/3**: Adding magic music...", expanded=True) as status:
             progress_bar.progress(100)
             audio_data = text2audio(story)
-            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)
+            status.update(label="✅ All ready!", state="complete")
+
+            # Auto-play the audio
+            st.audio(audio_data['audio'], 
+                    format="audio/wav",
+                    start_time=0,
+                    sample_rate=audio_data['sampling_rate'],
+                    autoplay=True) 
 
 if __name__ == "__main__":
     main()