Update app.py

app.py

@@ -1,5 +1,5 @@
 # app.py
-# @title 啤酒游戏最终整合版 (Streamlit 交互应用 + Hugging Face 日志上传)
+# @title 啤酒游戏最终整合版 (Streamlit 交互应用 + Hugging Face 日志上传) - 已修复版本
 
 # -----------------------------------------------------------------------------
 # 1. 导入必要的库
@@ -19,6 +19,12 @@ from pathlib import Path
 from datetime import datetime
 from huggingface_hub import HfApi, upload_file
 
+# -----------------------------------------------------------------------------
+# 0. 页面配置 (必须是第一个Streamlit命令)
+# -----------------------------------------------------------------------------
+st.set_page_config(page_title="啤酒游戏-人机协作版", layout="wide")
+
+
 # -----------------------------------------------------------------------------
 # 2. 配置游戏核心参数和API密钥
 # -----------------------------------------------------------------------------
@@ -50,12 +56,14 @@ try:
     else:
         hf_api = None
 except Exception as e:
-    st.error(f"启动时读取Secrets出错: {e}")
-    st.info("请确保在Streamlit的Secrets中设置了 OPENAI_API_KEY。可选设置 HF_TOKEN 和 HF_REPO_ID 用于上传日志。")
+    # 将错误信息显示在主页面，而不是在加载配置时就调用st.error
+    st.session_state.initialization_error = f"启动时读取Secrets出错: {e}. 请确保在Streamlit的Secrets中设置了 OPENAI_API_KEY。可选设置 HF_TOKEN 和 HF_REPO_ID 用于上传日志。"
     client = None
     HF_TOKEN = None
     HF_REPO_ID = None
     hf_api = None
+else:
+    st.session_state.initialization_error = None
 
 
 # -----------------------------------------------------------------------------
@@ -311,61 +319,65 @@ def save_logs_and_upload(state: dict):
 # -----------------------------------------------------------------------------
 # 4. Streamlit UI 界面
 # -----------------------------------------------------------------------------
-st.set_page_config(page_title="啤酒游戏-人机协作版", layout="wide")
 st.title("🍺 啤酒游戏：人机协作挑战")
-st.markdown("你将扮演供应链中的一个角色，与另外三个由大语言模型（LLM）驱动的AI代理合作。")
-
-# --- 游戏设置和初始化 ---
-if 'game_state' not in st.session_state or not st.session_state.game_state.get('game_running', False):
-    st.header("🎮 开始新游戏")
-    col1, col2 = st.columns(2)
-    with col1:
-        llm_personality = st.selectbox("AI '性格'", ('human_like', 'perfect_rational'), format_func=lambda x: x.replace('_', ' ').title())
-    with col2:
-        info_sharing = st.selectbox("信息共享", ('local', 'full'), format_func=lambda x: x.title())
-    if st.button("🚀 开始游戏", type="primary"):
-        init_game_state(llm_personality, info_sharing)
-        st.rerun()
-
-# --- 游戏主界面 ---
-elif 'game_state' in st.session_state and st.session_state.game_state.get('game_running'):
-    state = st.session_state.game_state
-    week, human_role, echelons = state['week'], state['human_role'], state['echelons']
-    st.header(f"第 {week} 周 / 共 {WEEKS} 周")
-    st.subheader(f"你的角色: **{human_role}** | AI模式: **{state['llm_personality'].replace('_', ' ')}** | 信息: **{state['info_sharing']}**")
-    cols = st.columns(4)
-    for i, name in enumerate(["Retailer", "Wholesaler", "Distributor", "Factory"]):
-        with cols[i]:
-            e, title_icon = echelons[name], "👤" if name == human_role else "🤖"
-            st.markdown(f"### {title_icon} {name} {'(你)' if name == human_role else '(AI)'}")
-            st.metric("库存", e['inventory']); st.metric("缺货/积压", e['backlog'])
-            st.write(f"本周收到订单: **{e['incoming_order']}**")
-            st.write(f"下周到货: **{list(e['incoming_shipments'])[0] if e['incoming_shipments'] else 0}**")
-    st.markdown("---")
-    st.header("你的决策")
-    human_echelon_state = echelons[human_role]
-    prompt_sugg = get_llm_prompt(human_echelon_state, week, state['llm_personality'], state['info_sharing'], echelons)
-    ai_suggestion, _ = get_llm_order_decision(prompt_sugg, f"{human_role} (Suggestion)", week, state['llm_personality'])
-    st.info(f"💡 AI建议你 ({human_role}) 本周向上游订购 **{ai_suggestion}** 单位。")
-    with st.form(key="order_form"):
-        final_order = st.number_input("请输入你的最终订单数量:", min_value=0, step=1, value=ai_suggestion)
-        if st.form_submit_button(label="✅ 提交订单并进入下一周"):
-            step_game(int(final_order)); st.rerun()
-    st.sidebar.header("游戏信息")
-    st.sidebar.markdown(f"**游戏ID**: `{state['participant_id']}`")
-    st.sidebar.markdown(f"**当前周**: {week-1} (已完成)")
-    if st.sidebar.button("🔄 重置游戏"):
-        del st.session_state.game_state; st.rerun()
-
-# --- 游戏结束界面 ---
-if 'game_state' in st.session_state and not st.session_state.game_state.get('game_running', False) and st.session_state.game_state['week'] > WEEKS:
-    st.header("🎉 游戏结束！")
-    state = st.session_state.game_state
-    logs_df = pd.json_normalize(state['logs'])
-    title = f"Beer Game (Human: {state['human_role']})\n(AI: {state['llm_personality'].replace('_', ' ').title()} | Info: {state['info_sharing'].title()})"
-    fig = plot_results(logs_df, title)
-    st.pyplot(fig)
-    # 保存并上传日志
-    save_logs_and_upload(state)
-    if st.button("✨ 开始一局新游戏"):
-        del st.session_state.game_state; st.rerun()
+
+# 检查初始化时是否有错误
+if st.session_state.get('initialization_error'):
+    st.error(st.session_state.initialization_error)
+else:
+    st.markdown("你将扮演供应链中的一个角色，与另外三个由大语言模型（LLM）驱动的AI代理合作。")
+
+    # --- 游戏设置和初始化 ---
+    if 'game_state' not in st.session_state or not st.session_state.game_state.get('game_running', False):
+        st.header("🎮 开始新游戏")
+        col1, col2 = st.columns(2)
+        with col1:
+            llm_personality = st.selectbox("AI '性格'", ('human_like', 'perfect_rational'), format_func=lambda x: x.replace('_', ' ').title())
+        with col2:
+            info_sharing = st.selectbox("信息共享", ('local', 'full'), format_func=lambda x: x.title())
+        if st.button("🚀 开始游戏", type="primary", disabled=(client is None)):
+            init_game_state(llm_personality, info_sharing)
+            st.rerun()
+
+    # --- 游戏主界面 ---
+    elif 'game_state' in st.session_state and st.session_state.game_state.get('game_running'):
+        state = st.session_state.game_state
+        week, human_role, echelons = state['week'], state['human_role'], state['echelons']
+        st.header(f"第 {week} 周 / 共 {WEEKS} 周")
+        st.subheader(f"你的角色: **{human_role}** | AI模式: **{state['llm_personality'].replace('_', ' ')}** | 信息: **{state['info_sharing']}**")
+        cols = st.columns(4)
+        for i, name in enumerate(["Retailer", "Wholesaler", "Distributor", "Factory"]):
+            with cols[i]:
+                e, title_icon = echelons[name], "👤" if name == human_role else "🤖"
+                st.markdown(f"### {title_icon} {name} {'(你)' if name == human_role else '(AI)'}")
+                st.metric("库存", e['inventory']); st.metric("缺货/积压", e['backlog'])
+                st.write(f"本周收到订单: **{e['incoming_order']}**")
+                st.write(f"下周到货: **{list(e['incoming_shipments'])[0] if e['incoming_shipments'] else 0}**")
+        st.markdown("---")
+        st.header("你的决策")
+        human_echelon_state = echelons[human_role]
+        prompt_sugg = get_llm_prompt(human_echelon_state, week, state['llm_personality'], state['info_sharing'], echelons)
+        ai_suggestion, _ = get_llm_order_decision(prompt_sugg, f"{human_role} (Suggestion)", week, state['llm_personality'])
+        st.info(f"💡 AI建议你 ({human_role}) 本周向上游订购 **{ai_suggestion}** 单位。")
+        with st.form(key="order_form"):
+            final_order = st.number_input("请输入你的最终订单数量:", min_value=0, step=1, value=ai_suggestion)
+            if st.form_submit_button(label="✅ 提交订单并进入下一周"):
+                step_game(int(final_order)); st.rerun()
+        st.sidebar.header("游戏信息")
+        st.sidebar.markdown(f"**游戏ID**: `{state['participant_id']}`")
+        st.sidebar.markdown(f"**当前周**: {week-1} (已完成)")
+        if st.sidebar.button("🔄 重置游戏"):
+            del st.session_state.game_state; st.rerun()
+
+    # --- 游戏结束界面 ---
+    if 'game_state' in st.session_state and not st.session_state.game_state.get('game_running', False) and st.session_state.game_state['week'] > WEEKS:
+        st.header("🎉 游戏结束！")
+        state = st.session_state.game_state
+        logs_df = pd.json_normalize(state['logs'])
+        title = f"Beer Game (Human: {state['human_role']})\n(AI: {state['llm_personality'].replace('_', ' ').title()} | Info: {state['info_sharing'].title()})"
+        fig = plot_results(logs_df, title)
+        st.pyplot(fig)
+        # 保存并上传日志
+        save_logs_and_upload(state)
+        if st.button("✨ 开始一局新游戏"):
+            del st.session_state.game_state; st.rerun()