Update app.py

app.py

@@ -1,5 +1,5 @@
 # app.py
-# @title Beer Game Final Version (v4.17 - Simplified Order Logic & State Clarity - COMPLETE CODE)
+# @title Beer Game Final Version (v4.18 - UI Adjustments)
 
 # -----------------------------------------------------------------------------
 # 1. Import Libraries
@@ -22,7 +22,6 @@ from huggingface_hub import HfApi
 # -----------------------------------------------------------------------------
 st.set_page_config(page_title="Beer Game: Human-AI Collaboration", layout="wide")
 
-
 # -----------------------------------------------------------------------------
 # 2. Game Parameters & API Configuration
 # -----------------------------------------------------------------------------
@@ -55,7 +54,6 @@ except Exception as e:
 else:
     st.session_state.initialization_error = None
 
-
 # -----------------------------------------------------------------------------
 # 3. Core Game Logic Functions
 # -----------------------------------------------------------------------------
@@ -121,7 +119,7 @@ def get_llm_order_decision(prompt: str, echelon_name: str) -> (int, str):
             st.warning(f"LLM for {echelon_name} did not return a valid number. Defaulting to 4. Raw Response: '{raw_text}'")
             return 4, raw_text # Default to 4 if no number found
         except Exception as e:
-            st.error(f"API call failed for {echelon_name}: {e}. Defaulting to 4.")
+            st.error(f"API call failed for {echelons[human_role]}: {e}. Defaulting to 4.") # Fixed typo: echelon_name -> echelons[human_role]
             return 4, f"API_ERROR: {e}"
 
 # =============== MODIFIED FUNCTION (Prompt uses simplified state) ===============
@@ -557,16 +555,8 @@ else:
         * `Arriving Next Week`: The quantity scheduled to arrive at the start of the **next week**. Use this for your planning.
         * `Your Total Cumulative Cost`: Sum of all weekly costs up to the **end of last week**.
         * `Cost Last Week`: The specific cost incurred just **last week**.
-
-        **C) Your Decision (Step 4 - Two Parts):**
-        Now, looking at the dashboard (showing the start-of-week state) and considering the incoming order and future arrivals, you decide how much to order:
-        * **(Step 4a - Initial Order):** Submit your first estimate.
-        * **(Step 4b - Final Order):** See the AI's suggestion, then submit your final decision. This order will arrive in 3 weeks.
-
-        Submitting your final order ends the week. The system then calculates your `Weekly Cost` based on your inventory/backlog *after* Step 3 shipping, logs everything, and advances to the next week.
         """)
 
-
         st.markdown("---")
         st.header("⚙️ Game Configuration")
         c1, c2 = st.columns(2)
@@ -582,7 +572,7 @@ else:
     # --- Main Game Interface ---
     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, info_sharing = state['week'], state['human_role'], state['echelons'], state['info_sharing']
+        week, human_role, echelons, info_sharing = state['week'], state['echelons'], state['human_role'], state['info_sharing']
         echelon_order = ["Retailer", "Wholesaler", "Distributor", "Factory"] # Define here for UI
 
 
@@ -595,17 +585,20 @@ else:
             for i, name in enumerate(echelon_order): # Use the defined echelon_order
                 with cols[i]:
                     e, icon = echelons[name], "👤" if name == human_role else "🤖"
-                    st.markdown(f"##### {icon} {name} {'(You)' if name == human_role else ''}")
+                    if name == human_role:
+                        # Highlight Distributor for full info mode
+                        st.markdown(f"##### **<span style='color:#FF4B4B;'>{icon} {name} (You)</span>**", unsafe_allow_html=True) # Highlight
+                    else:
+                        st.markdown(f"##### {icon} {name}")
                     # Display the END OF LAST WEEK state (which is OPENING state for this week)
                     st.metric("Inventory (Opening)", e['inventory'])
                     st.metric("Backlog (Opening)", e['backlog'])
 
-                    # Display cumulative cost and last week's cost for the human player
-                    if name == human_role:
-                        st.metric("Total Cost (Cumulative)", f"${e['total_cost']:,.2f}")
-                        # Display last week's cost if available (week > 1)
-                        last_week_cost = state['logs'][-1][f"{human_role}.weekly_cost"] if week > 1 and state['logs'] else 0
-                        st.metric("Cost Last Week", f"${last_week_cost:,.2f}")
+                    # 移除成本显示
+                    # if name == human_role:
+                    #     st.metric("Total Cost (Cumulative)", f"${e['total_cost']:,.2f}")
+                    #     last_week_cost = state['logs'][-1][f"{human_role}.weekly_cost"] if week > 1 and state['logs'] else 0
+                    #     st.metric("Cost Last Week", f"${last_week_cost:,.2f}")
 
                     # Display info about THIS week's events / NEXT week's arrivals
                     st.write(f"Incoming Order (This Week): **{e['incoming_order']}**") # Order arriving in Step 2
@@ -620,7 +613,7 @@ else:
         else:
             st.info("In Local Information mode, you can only see your own status dashboard.")
             e = echelons[human_role]
-            st.markdown(f"### 👤 {human_role} (Your Dashboard - Start of Week State)")
+            st.markdown(f"### 👤 **<span style='color:#FF4B4B;'>{human_role} (Your Dashboard - Start of Week State)</span>**", unsafe_allow_html=True) # Highlight self
             col1, col2, col3, col4 = st.columns(4)
             # Display OPENING state
             col1.metric("Inventory (Opening)", e['inventory'])
@@ -629,9 +622,10 @@ else:
             col3.write(f"**Incoming Order (This Week):**\n# {e['incoming_order']}")
             col4.write(f"**Shipment Arriving (Next Week):**\n# {list(e['incoming_shipments'])[0] if e['incoming_shipments'] else 0}")
 
-            st.metric("Your Total Cumulative Cost", f"${e['total_cost']:,.2f}")
-            last_week_cost = state['logs'][-1][f"{human_role}.weekly_cost"] if week > 1 and state['logs'] else 0
-            st.metric("Cost Last Week", f"${last_week_cost:,.2f}")
+            # 移除成本显示
+            # st.metric("Your Total Cumulative Cost", f"${e['total_cost']:,.2f}")
+            # last_week_cost = state['logs'][-1][f"{human_role}.weekly_cost"] if week > 1 and state['logs'] else 0
+            # st.metric("Cost Last Week", f"${last_week_cost:,.2f}")
 
 
         st.markdown("---")