Update app.py

app.py

@@ -23,9 +23,15 @@ def text2audio(story_text):
 
 
 def main():
-    st.set_page_config(page_title="Your Image to Audio Story", page_icon="🦜")
-    st.header("Turn Your Image to Audio Story")
-    uploaded_file = st.file_uploader("Select an Image...")
+    st.set_page_config(page_title="Magic Story Box", page_icon="🧚")
+    # 新版标题区
+    st.markdown("""
+    <div class="header">
+        <h1>🪄 Magic Picture Story Box 🎧</h1>
+        <h4 style="color:#4ECDC4;">Upload any photo → Get a fairy tale!</h4>
+    </div>
+    """, unsafe_allow_html=True)
+    uploaded_file = st.file_uploader("🌈 Choose your magic picture...", type=["jpg", "png"])
 
     if uploaded_file is not None:
         print(uploaded_file)
@@ -36,25 +42,32 @@ def main():
 
 
         #Stage 1: Image to Text
-        st.text('🔍 Analyzing image...')
-        scenario = img2text(uploaded_file.name)
-        st.write(scenario)
+        with status_container.status("🔮 **Step 1/3**: Decoding picture magic...", expanded=True) as status:
+            progress_bar.progress(33)
+            scenario = img2text(uploaded_file.name)
+            status.update(label="✅ Picture decoded!", state="complete")
+            st.write(f"**What I see:** {scenario}")
 
         #Stage 2: Text to Story
-        st.text('📖 Generating story...')
-        story = text2story(scenario)
-        st.write(story)
+        with status_container.status("📚 **Step 2/3**: Writing your fairy tale...", expanded=True) as status:
+            progress_bar.progress(66)
+            story = text2story(scenario)
+            status.update(label="✅ Story created!", state="complete")
+            st.write(f"**Your Story:**\n{story}")
 
         #Stage 3: Story to Audio data
-        st.text('🎧 Converting to audio...')
-        audio_data =text2audio(story)
+        with status_container.status("🎵 **Step 3/3**: Adding magic music...", expanded=True) as status:
+            progress_bar.progress(100)
+            audio_data = text2audio(story)
+            status.update(label="✅ All ready!", state="complete")
 
-        # Play button
-        if st.button("Play Audio Story"):
+
+            # 自动播放（移除按钮）
             st.audio(audio_data['audio'], 
-                     format="audio/wav", 
-                     start_time=0, 
-                     sample_rate = audio_data['sampling_rate'])
+                    format="audio/wav",
+                    start_time=0,
+                    sample_rate=audio_data['sampling_rate'],
+                    autoplay=True)  # 新增自动播放参数
 
 
 if __name__ == "__main__":