Update app.py

app.py

@@ -100,7 +100,8 @@ def init_game_state(llm_personality: str, info_sharing: str, participant_id: str
             'incoming_order': 0, 'order_placed': 0, 'shipment_sent': 0,
             'weekly_cost': 0, 'total_cost': 0, 'upstream_name': upstream, 'downstream_name': downstream,
         }
-    st.info(f"New game started for **{participant_id}**! AI Mode: **{llm_personality} / {info_sharing}**. You are the **{human_role}**.")
+    # [修改点 2.1]: 移除启动时显示实验条件的提示
+    st.info(f"New game started for **{participant_id}**! You are the **{human_role}**.")
 # ==============================================================================
 def get_llm_order_decision(prompt: str, echelon_name: str) -> (int, str):
     # This function remains correct.
@@ -614,7 +615,8 @@ else:
             # 默认值，如果未输入ID
             llm_personality, info_sharing = ('human_like', 'local') 
             
-        #st.info(f"You will be automatically assigned to the condition: **{llm_personality.replace('_', ' ').title()} / {info_sharing.title()}**.")
+        # [修改点 2.2]: 移除启动时显示实验条件的提示
+        # st.info(f"You will be automatically assigned to the condition: **{llm_personality.replace('_', ' ').title()} / {info_sharing.title()}**.")
         # =================================================================
         
         # 移除原有的 c1, c2 和 selectbox 
@@ -655,174 +657,172 @@ else:
         week, human_role, echelons, info_sharing = state['week'], state['human_role'], state['echelons'], state['info_sharing']
         echelon_order = ["Retailer", "Wholesaler", "Distributor", "Factory"] # Define here for UI
         
-        # =============== 2. IMAGE SIZE MODIFICATION (使用 st.columns) ===============
-        col_main, col_sidebar_image = st.columns([4, 1]) # 增大主内容区，同时保持侧边栏用于放置图表
+        # [修改点 1.1]: 移除 st.columns，恢复主内容到左侧，侧边栏不变
+        # col_main, col_sidebar_image = st.columns([4, 1]) 移除
         
-        with col_main:
-            st.header(f"Week {week} / {WEEKS}")
-            st.subheader(f"Your Role: **{human_role}** ({state['participant_id']}) | AI Mode: **{state['llm_personality'].replace('_', ' ')}** | Information: **{state['info_sharing']}**")
-            st.markdown("---")
-            st.subheader("Supply Chain Status (Start of Week State)") 
-            
-            # ... (UI LOGIC REMAINS HERE) ...
-            
-            # --- MODIFIED UI LOGIC (v12) ---
-            if info_sharing == 'full':
-                cols = st.columns(4)
-                for i, name in enumerate(echelon_order):
-                    with cols[i]:
-                        e = echelons[name]
-                        icon = "👤" if name == human_role else "🤖"
-                        if name == human_role:
-                            st.markdown(f"##### **<span style='border: 1px solid #FF4B4B; padding: 2px 5px; border-radius: 3px;'>{icon} {name} (You)</span>**", unsafe_allow_html=True)
-                        else:
-                            st.markdown(f"##### {icon} {name}")
-                                            
-                        st.metric("Inventory (Opening)", e['inventory'])
-                        st.metric("Backlog (Opening)", e['backlog'])
-                                            
-                        current_incoming_order = 0
-                        if name == "Retailer":
-                            current_incoming_order = get_customer_demand(week)
-                        else:
-                            downstream_name = e['downstream_name']
-                            if downstream_name:
-                                current_incoming_order = state['last_week_orders'].get(downstream_name, 0)
-                        st.write(f"Incoming Order (This Week): **{current_incoming_order}**")
-                                            
-                        if name == "Factory":
-                            prod_completing_next = state['last_week_orders'].get("Distributor", 0)
-                            st.write(f"Completing Next Week: **{prod_completing_next}**")
-                        else:
-                            arriving_next = 0
-                            q = e['incoming_shipments']
-                            if q: arriving_next = list(q)[0] # Read W+1
-                            st.write(f"Arriving Next Week: **{arriving_next}**")
-            else: # Local Info Mode
-                st.info("In Local Information mode, you can only see your own status dashboard.")
-                e = echelons[human_role] # Distributor
-                st.markdown(f"### 👤 **<span style='color:#FF4B4B;'>{human_role} (Your Dashboard - Start of Week State)</span>**", unsafe_allow_html=True)
-                col1, col2, col3 = st.columns(3)
-                with col1:
+        # with col_main: 移除
+        st.header(f"Week {week} / {WEEKS}")
+        # [修改点 2.3]: 移除 subheader 中 AI Mode 和 Information 的��示
+        st.subheader(f"Your Role: **{human_role}** ({state['participant_id']})") 
+        st.markdown("---")
+        st.subheader("Supply Chain Status (Start of Week State)") 
+        
+        # --- MODIFIED UI LOGIC (v12) ---
+        if info_sharing == 'full':
+            cols = st.columns(4)
+            for i, name in enumerate(echelon_order):
+                with cols[i]:
+                    e = echelons[name]
+                    icon = "👤" if name == human_role else "🤖"
+                    if name == human_role:
+                        st.markdown(f"##### **<span style='border: 1px solid #FF4B4B; padding: 2px 5px; border-radius: 3px;'>{icon} {name} (You)</span>**", unsafe_allow_html=True)
+                    else:
+                        st.markdown(f"##### {icon} {name}")
+                                        
                     st.metric("Inventory (Opening)", e['inventory'])
                     st.metric("Backlog (Opening)", e['backlog'])
-                            
-                with col2:
+                                        
                     current_incoming_order = 0
-                    downstream_name = e['downstream_name'] # Wholesaler
-                    if downstream_name:
-                        current_incoming_order = state['last_week_orders'].get(downstream_name, 0)
-                    st.write(f"**Incoming Order (This Week):**\n# {current_incoming_order}")
-                                
-                with col3:
-                    arriving_next = 0
-                    q = e['incoming_shipments']
-                    if q: 
-                        arriving_next = list(q)[0]
-                    st.write(f"**Shipment Arriving (Next Week):**\n# {arriving_next}")
-            
-            # --- Decision Logic (Remainging the Same) ---
-            st.markdown("---")
-            st.header("Your Decision (Step 4)")
-            
-            # Prepare state snapshot for the AI prompt (logic remains identical)
-            all_decision_point_states = {}
-            for name in echelon_order:
-                e_curr = echelons[name] 
-                arrived = 0
-                if name == "Factory":
-                    if state['factory_production_pipeline']: arrived = list(state['factory_production_pipeline'])[0]
-                else:
-                    if e_curr['incoming_shipments']: arrived = list(e_curr['incoming_shipments'])[0]
-                            
-                inc_order_this_week = 0
-                if name == "Retailer": inc_order_this_week = get_customer_demand(week)
-                else:
-                    ds_name = e_curr['downstream_name']
-                    if ds_name: inc_order_this_week = state['last_week_orders'].get(ds_name, 0)
-                inv_after_arrival = e_curr['inventory'] + arrived
-                backlog_after_new_order = e_curr['backlog'] + inc_order_this_week
+                    if name == "Retailer":
+                        current_incoming_order = get_customer_demand(week)
+                    else:
+                        downstream_name = e['downstream_name']
+                        if downstream_name:
+                            current_incoming_order = state['last_week_orders'].get(downstream_name, 0)
+                    st.write(f"Incoming Order (This Week): **{current_incoming_order}**")
+                                        
+                    if name == "Factory":
+                        prod_completing_next = state['last_week_orders'].get("Distributor", 0)
+                        st.write(f"Completing Next Week: **{prod_completing_next}**")
+                    else:
+                        arriving_next = 0
+                        q = e['incoming_shipments']
+                        if q: arriving_next = list(q)[0] # Read W+1
+                        st.write(f"Arriving Next Week: **{arriving_next}**")
+        else: # Local Info Mode
+            st.info("In Local Information mode, you can only see your own status dashboard.")
+            e = echelons[human_role] # Distributor
+            st.markdown(f"### 👤 **<span style='color:#FF4B4B;'>{human_role} (Your Dashboard - Start of Week State)</span>**", unsafe_allow_html=True)
+            col1, col2, col3 = st.columns(3)
+            with col1:
+                st.metric("Inventory (Opening)", e['inventory'])
+                st.metric("Backlog (Opening)", e['backlog'])
+                        
+            with col2:
+                current_incoming_order = 0
+                downstream_name = e['downstream_name'] # Wholesaler
+                if downstream_name:
+                    current_incoming_order = state['last_week_orders'].get(downstream_name, 0)
+                st.write(f"**Incoming Order (This Week):**\n# {current_incoming_order}")
                             
-                all_decision_point_states[name] = {
-                    'name': name, 'inventory': inv_after_arrival, 'backlog': backlog_after_new_order,
-                    'incoming_order': inc_order_this_week, 
-                    'incoming_shipments': e_curr['incoming_shipments'].copy() if name != "Factory" else deque()
-                }
-            human_echelon_state_for_prompt = all_decision_point_states[human_role]
-            
-            if state['decision_step'] == 'initial_order':
-                with st.form(key="initial_order_form"):
-                    st.markdown("#### **Step 4a:** Based on the dashboard, submit your **initial** order to the Factory.")
-                    initial_order = st.number_input("Your Initial Order Quantity:", min_value=0, step=1, value=None) # Start blank
-                    if st.form_submit_button("Submit Initial Order & See AI Suggestion", type="primary"):
-                        state['human_initial_order'] = int(initial_order) if initial_order is not None else 0
-                        state['decision_step'] = 'final_order'
-                        prompt_sugg = get_llm_prompt(human_echelon_state_for_prompt, week, state['llm_personality'], state['info_sharing'], all_decision_point_states)
-                        ai_suggestion, _ = get_llm_order_decision(prompt_sugg, f"{human_role} (Suggestion)")
-                        state['current_ai_suggestion'] = ai_suggestion # Store it
-                        st.rerun()
-            elif state['decision_step'] == 'final_order':
-                st.success(f"Your initial order was: **{state['human_initial_order']}** units.")
-                ai_suggestion = state.get('current_ai_suggestion', 4) # Read stored value
-                with st.form(key="final_order_form"):
-                    st.markdown(f"#### **Step 4b:** The AI suggests ordering **{ai_suggestion}** units.")
-                    st.markdown("Considering the AI's advice, submit your **final** order to end the week. (This order will arrive in 3 weeks).")
-                    st.number_input("Your Final Order Quantity:", min_value=0, step=1, key='final_order_input', value=None) # Start blank
-                                    
-                    if st.form_submit_button("Submit Final Order & Advance to Next Week"):
-                        final_order_value = st.session_state.get('final_order_input', 0)
-                        final_order_value = int(final_order_value) if final_order_value is not None else 0
-                                                    
-                        step_game(final_order_value, state['human_initial_order'], ai_suggestion)
-                                                    
-                        if 'final_order_input' in st.session_state: del st.session_state.final_order_input
-                        st.rerun()
-            
-            st.markdown("---")
-            with st.expander("📖 Your Weekly Decision Log", expanded=False):
-                if not state.get('logs'):
-                    st.write("Your weekly history will be displayed here after you complete the first week.")
-                else:
-                    try:
-                        history_df = pd.json_normalize(state['logs'])
-                        human_cols = {
-                            'week': 'Week', f'{human_role}.opening_inventory': 'Opening Inv.',
-                            f'{human_role}.opening_backlog': 'Opening Backlog', 
-                            f'{human_role}.incoming_order': 'Incoming Order', f'{human_role}.initial_order': 'Your Initial Order',
-                            f'{human_role}.ai_suggestion': 'AI Suggestion', f'{human_role}.order_placed': 'Your Final Order',
-                            f'{human_role}.arriving_next_week': 'Arriving Next Week', f'{human_role}.weekly_cost': 'Weekly Cost',
-                        }
-                        ordered_display_cols_keys = [
-                            'week', f'{human_role}.opening_inventory', f'{human_role}.opening_backlog',
-                            f'{human_role}.incoming_order',
-                            f'{human_role}.initial_order', f'{human_role}.ai_suggestion', f'{human_role}.order_placed',
-                            f'{human_role}.arriving_next_week', f'{human_role}.weekly_cost'
-                        ]
-                        final_cols_to_display = [col for col in ordered_display_cols_keys if col in history_df.columns]
-                        if not final_cols_to_display:
-                            st.write("No data columns available to display.")
-                        else:
-                            display_df = history_df[final_cols_to_display].rename(columns=human_cols)
-                            if 'Weekly Cost' in display_df.columns:
-                                display_df['Weekly Cost'] = display_df['Weekly Cost'].apply(lambda x: f"${x:,.2f}" if isinstance(x, (int, float)) else "")
-                            st.dataframe(display_df.sort_values(by="Week", ascending=False), hide_index=True, use_container_width=True)
-                    except Exception as e:
-                        st.error(f"Error displaying weekly log: {e}")
+            with col3:
+                arriving_next = 0
+                q = e['incoming_shipments']
+                if q: 
+                    arriving_next = list(q)[0]
+                st.write(f"**Shipment Arriving (Next Week):**\n# {arriving_next}")
         
-        # 将原来的侧边栏内容移入 col_sidebar_image
-        with col_sidebar_image:
-            st.header("Game Info")
-            st.markdown(f"**Game ID**: `{state['participant_id']}`\n\n**Current Week**: {week}")
-            # --- 放大图片 ---
-            try: st.image(IMAGE_PATH, caption="Supply Chain Reference", use_column_width=True) 
-            except FileNotFoundError: st.warning("Image file not found.")
-            # --- 放大图片 ---
-            
-            if st.button("🔄 Reset Game"):
-                if 'final_order_input' in st.session_state: del st.session_state.final_order_input
-                if 'current_ai_suggestion' in st.session_state.game_state: del st.session_state.game_state['current_ai_suggestion']
-                del st.session_state.game_state
-                st.rerun()
+        # --- Decision Logic (Remainging the Same) ---
+        st.markdown("---")
+        st.header("Your Decision (Step 4)")
+        
+        # Prepare state snapshot for the AI prompt (logic remains identical)
+        all_decision_point_states = {}
+        for name in echelon_order:
+            e_curr = echelons[name] 
+            arrived = 0
+            if name == "Factory":
+                if state['factory_production_pipeline']: arrived = list(state['factory_production_pipeline'])[0]
+            else:
+                if e_curr['incoming_shipments']: arrived = list(e_curr['incoming_shipments'])[0]
+                        
+            inc_order_this_week = 0
+            if name == "Retailer": inc_order_this_week = get_customer_demand(week)
+            else:
+                ds_name = e_curr['downstream_name']
+                if ds_name: inc_order_this_week = state['last_week_orders'].get(ds_name, 0)
+            inv_after_arrival = e_curr['inventory'] + arrived
+            backlog_after_new_order = e_curr['backlog'] + inc_order_this_week
+                        
+            all_decision_point_states[name] = {
+                'name': name, 'inventory': inv_after_arrival, 'backlog': backlog_after_new_order,
+                'incoming_order': inc_order_this_week, 
+                'incoming_shipments': e_curr['incoming_shipments'].copy() if name != "Factory" else deque()
+            }
+        human_echelon_state_for_prompt = all_decision_point_states[human_role]
+        
+        if state['decision_step'] == 'initial_order':
+            with st.form(key="initial_order_form"):
+                st.markdown("#### **Step 4a:** Based on the dashboard, submit your **initial** order to the Factory.")
+                initial_order = st.number_input("Your Initial Order Quantity:", min_value=0, step=1, value=None) # Start blank
+                if st.form_submit_button("Submit Initial Order & See AI Suggestion", type="primary"):
+                    state['human_initial_order'] = int(initial_order) if initial_order is not None else 0
+                    state['decision_step'] = 'final_order'
+                    prompt_sugg = get_llm_prompt(human_echelon_state_for_prompt, week, state['llm_personality'], state['info_sharing'], all_decision_point_states)
+                    ai_suggestion, _ = get_llm_order_decision(prompt_sugg, f"{human_role} (Suggestion)")
+                    state['current_ai_suggestion'] = ai_suggestion # Store it
+                    st.rerun()
+        elif state['decision_step'] == 'final_order':
+            st.success(f"Your initial order was: **{state['human_initial_order']}** units.")
+            ai_suggestion = state.get('current_ai_suggestion', 4) # Read stored value
+            with st.form(key="final_order_form"):
+                st.markdown(f"#### **Step 4b:** The AI suggests ordering **{ai_suggestion}** units.")
+                st.markdown("Considering the AI's advice, submit your **final** order to end the week. (This order will arrive in 3 weeks).")
+                st.number_input("Your Final Order Quantity:", min_value=0, step=1, key='final_order_input', value=None) # Start blank
+                                
+                if st.form_submit_button("Submit Final Order & Advance to Next Week"):
+                    final_order_value = st.session_state.get('final_order_input', 0)
+                    final_order_value = int(final_order_value) if final_order_value is not None else 0
+                                        
+                    step_game(final_order_value, state['human_initial_order'], ai_suggestion)
+                                        
+                    if 'final_order_input' in st.session_state: del st.session_state.final_order_input
+                    st.rerun()
+        
+        st.markdown("---")
+        with st.expander("📖 Your Weekly Decision Log", expanded=False):
+            if not state.get('logs'):
+                st.write("Your weekly history will be displayed here after you complete the first week.")
+            else:
+                try:
+                    history_df = pd.json_normalize(state['logs'])
+                    human_cols = {
+                        'week': 'Week', f'{human_role}.opening_inventory': 'Opening Inv.',
+                        f'{human_role}.opening_backlog': 'Opening Backlog', 
+                        f'{human_role}.incoming_order': 'Incoming Order', f'{human_role}.initial_order': 'Your Initial Order',
+                        f'{human_role}.ai_suggestion': 'AI Suggestion', f'{human_role}.order_placed': 'Your Final Order',
+                        f'{human_role}.arriving_next_week': 'Arriving Next Week', f'{human_role}.weekly_cost': 'Weekly Cost',
+                    }
+                    ordered_display_cols_keys = [
+                        'week', f'{human_role}.opening_inventory', f'{human_role}.opening_backlog',
+                        f'{human_role}.incoming_order',
+                        f'{human_role}.initial_order', f'{human_role}.ai_suggestion', f'{human_role}.order_placed',
+                        f'{human_role}.arriving_next_week', f'{human_role}.weekly_cost'
+                    ]
+                    final_cols_to_display = [col for col in ordered_display_cols_keys if col in history_df.columns]
+                    if not final_cols_to_display:
+                        st.write("No data columns available to display.")
+                    else:
+                        display_df = history_df[final_cols_to_display].rename(columns=human_cols)
+                        if 'Weekly Cost' in display_df.columns:
+                            display_df['Weekly Cost'] = display_df['Weekly Cost'].apply(lambda x: f"${x:,.2f}" if isinstance(x, (int, float)) else "")
+                        st.dataframe(display_df.sort_values(by="Week", ascending=False), hide_index=True, use_container_width=True)
+                except Exception as e:
+                    st.error(f"Error displaying weekly log: {e}")
+        
+        # [修改点 1.2]: 恢复 Game Info 和图片到 st.sidebar
+        # 移除 with col_sidebar_image: 
+        st.sidebar.header("Game Info")
+        st.sidebar.markdown(f"**Game ID**: `{state['participant_id']}`\n\n**Current Week**: {week}")
+        # [修改点 1.3]: 恢复 st.sidebar.image 但使用 use_column_width=True 保持图片放大
+        try: st.sidebar.image(IMAGE_PATH, caption="Supply Chain Reference", use_column_width=True) 
+        except FileNotFoundError: st.sidebar.warning("Image file not found.")
+        
+        if st.sidebar.button("🔄 Reset Game"):
+            if 'final_order_input' in st.session_state: del st.session_state.final_order_input
+            if 'current_ai_suggestion' in st.session_state.game_state: del st.session_state.game_state['current_ai_suggestion']
+            del st.session_state.game_state
+            st.rerun()
 
     # --- Game Over Interface ---
     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:
@@ -832,7 +832,8 @@ else:
             logs_df = pd.json_normalize(state['logs'])
             fig = plot_results(
                 logs_df,
-                f"Beer Game (Human: {state['human_role']})\n(AI: {state['llm_personality']} | Info: {state['info_sharing']})",
+                # [修改点 2.4]: 移除报告标题中的实验条件提示
+                f"Beer Game (Human: {state['human_role']})",
                 state['human_role']
             )
             st.pyplot(fig)