Update app.py

app.py

@@ -118,16 +118,6 @@ def text_to_speech(text, processor, model, vocoder, embeddings_dataset):
         raise
 
 def main():
-    # 初始化会话状态
-    required_states = {
-        'generated': False,
-        'audio': None,
-        'story': ""
-    }
-    for key, val in required_states.items():
-        if key not in st.session_state:
-            st.session_state[key] = val
-
     # 界面配置
     st.set_page_config(
         page_title="Magic Story Box",
@@ -139,49 +129,18 @@ def main():
     st.markdown("---")
     st.write("Upload an image to get your magical story!")
     
-    # 加载模型
-    try:
-        (blip_proc, blip_model, story_gen, 
-         tts_proc, tts_model, vocoder, embeddings) = load_models()
-    except:
-        return
+    # 初始化会话状态
+    if 'generated' not in st.session_state:
+        st.session_state.generated = False
     
-    # 文件上传组件
-    uploaded_file = st.file_uploader(
-        "Choose your magic image",
-        type=["jpg", "png", "jpeg"],
-        help="Upload photos of pets, toys or adventures!",
-        key="uploader"
-    )
+    # 加载模型（保持不变...）
+    
+    # 文件上传组件（保持不变...）
     
-    # 处理上传文件
-    if uploaded_file and not st.session_state.generated:
-        try:
-            image = Image.open(uploaded_file).convert("RGB")
-            st.image(image, caption="Your Magic Picture ✨", use_container_width=True)
-            
-            with st.status("Creating Magic...", expanded=True) as status:
-                # 生成故事
-                st.write("🔍 Reading the image...")
-                story = generate_story(image, blip_proc, blip_model, story_gen)
-                
-                # 生成语音
-                st.write("🔊 Adding sounds...")
-                audio_array, sr = text_to_speech(story, tts_proc, tts_model, vocoder, embeddings)
-                
-                # 保存结果
-                st.session_state.story = story
-                st.session_state.audio = (audio_array, sr)
-                status.update(label="Ready!", state="complete", expanded=False)
-            
-            st.session_state.generated = True
-            st.rerun()
-            
-        except Exception as e:
-            st.error(f"Magic failed: {str(e)}")
+    # 处理上传文件（保持不变...）
     
     # 显示结果
-    if st.session_state.get('generated', False):
+    if st.session_state.generated:
         st.markdown("---")
         st.subheader("Your Story 📖")
         st.markdown(f'<div style="background:#fff3e6; padding:20px; border-radius:10px;">{st.session_state.story}</div>', 
@@ -189,19 +148,8 @@ def main():
         
         st.markdown("---")
         st.subheader("Listen 🎧")
-        if st.session_state.audio is not None:
-            audio_data, sr = st.session_state.audio
-            st.audio(audio_data, sample_rate=sr)
-        else:
-            st.warning("Audio not available")
-        
-        st.markdown("---")
-        if st.button("Create New Story", use_container_width=True):
-            # 安全重置状态
-            st.session_state.generated = False
-            st.session_state.audio = None
-            st.session_state.story = ""
-            st.rerun()
+        audio_data, sr = st.session_state.audio
+        st.audio(audio_data, sample_rate=sr)
 
 if __name__ == "__main__":
     main()