Update app.py

app.py

@@ -1,5 +1,5 @@
 # app.py
-# @title Beer Game Final Version (v4.3 - Added Weekly Decision Log)
+# @title Beer Game Final Version (v4.4 - Added Diagram Image)
 
 # -----------------------------------------------------------------------------
 # 1. Import Libraries
@@ -41,6 +41,7 @@ BACKLOG_COST = 1.0
 OPENAI_MODEL = "gpt-4o-mini"
 LOCAL_LOG_DIR = Path("logs")
 LOCAL_LOG_DIR.mkdir(exist_ok=True)
+IMAGE_PATH = "beer_game_diagram.png" # Path to your uploaded image
 
 # --- API & Secrets Configuration ---
 try:
@@ -144,7 +145,6 @@ def step_game(human_final_order: int, human_initial_order: int, ai_suggestion: i
     echelon_order = ["Retailer", "Wholesaler", "Distributor", "Factory"]
     llm_raw_responses = {}
     
-    # Store pre-step state for logging
     pre_step_inventory = echelons[human_role]['inventory']
     pre_step_backlog = echelons[human_role]['backlog']
 
@@ -215,7 +215,7 @@ def plot_results(df: pd.DataFrame, title: str, human_role: str):
     inventory_pivot.plot(ax=axes[0], kind='line', marker='o', markersize=4); axes[0].set_title('Inventory Levels'); axes[0].grid(True, linestyle='--'); axes[0].set_ylabel('Stock (Units)')
 
     order_pivot = plot_df.pivot(index='week', columns='echelon', values='order_placed').reindex(columns=echelons)
-    order_pivot.plot(ax=axes[1], style='--'); axes[1].plot(range(1, WEEKS + 1), [get_customer_demand(w) for w in range(1, WEEKS + 1)], label='Customer Demand', color='black', lw=2.5); axes[1].set_title('Order Quantities (The Bullwhip Effect)'); axes[1].grid(True, linestyle='--'); axes[1].legend(); axes[1].set_ylabel('Ordered (Units)')
+    order_pivot.plot(ax=axes[1], style='--'); axes[1].plot(range(1, WS + 1), [get_customer_demand(w) for w in range(1, WEEKS + 1)], label='Customer Demand', color='black', lw=2.5); axes[1].set_title('Order Quantities (The Bullwhip Effect)'); axes[1].grid(True, linestyle='--'); axes[1].legend(); axes[1].set_ylabel('Ordered (Units)')
 
     total_costs = plot_df.groupby('echelon')['total_cost'].max().reindex(echelons)
     total_costs.plot(kind='bar', ax=axes[2], rot=0); axes[2].set_title('Total Cumulative Cost'); axes[2].set_ylabel('Cost ($)')
@@ -255,6 +255,7 @@ if st.session_state.get('initialization_error'):
 else:
     # --- Game Setup & Instructions ---
     if 'game_state' not in st.session_state or not st.session_state.game_state.get('game_running', False):
+        
         st.markdown("---")
         st.header("📖 Welcome to the Beer Game!")
         st.markdown("""
@@ -266,9 +267,18 @@ else:
         - **Holding Inventory:** **$0.50 per unit per week**
         - **Backlog (Unfilled Orders):** **$1.00 per unit per week**
         """)
+
         col1, col2 = st.columns(2)
         with col1:
             st.subheader("🔗 The Supply Chain")
+            
+            # =============== IMAGE ADDED HERE ===============
+            try:
+                st.image(IMAGE_PATH, caption="Orders flow upstream (right), beer flows downstream (left).")
+            except FileNotFoundError:
+                st.warning("Image file not found. Please ensure 'beer_game_diagram.png' is uploaded to the repository.")
+            # ================================================
+                
             st.markdown("""
             You will be randomly assigned one of four roles. The other three will be controlled by AI agents.
             - **Retailer:** Fulfills end-customer demand. Orders from the Wholesaler.
@@ -279,8 +289,13 @@ else:
         with col2:
             st.subheader("⏳ The Challenge: Delays!")
             st.markdown("""
-            The key challenge is managing delays. There is a **communication delay** for orders to reach your supplier and a **shipping delay** for goods to arrive. You must order enough to meet future demand without creating a huge pile of expensive inventory.
+            The key challenge is managing delays. As shown in the diagram, there are **shipping delays** for beer to arrive and **production delays** at the factory. 
+            
+            There is also a one-week **order-passing delay** (the small boxes) for your order to reach your supplier. 
+            
+            You must order enough to meet future demand without creating a huge pile of expensive inventory.
             """)
+        
         st.subheader("🎮 How to Play This Version")
         st.markdown("""
         1.  **Configure the Game:** Choose the AI's behavior and the level of information sharing.
@@ -291,12 +306,14 @@ else:
         4.  **Advance:** Once you submit your final order, the week advances, and all AI agents make their moves.
         """)
         st.markdown("---")
+        
         st.header("⚙️ Game Configuration")
         c1, c2 = st.columns(2)
         with c1:
              llm_personality = st.selectbox("AI Agent 'Personality'", ('human_like', 'perfect_rational'), format_func=lambda x: x.replace('_', ' ').title(), help="**Human-like:** Tends to react emotionally, potentially over-ordering. **Perfect Rational:** Uses a mathematical heuristic to make stable, logical decisions.")
         with c2:
             info_sharing = st.selectbox("Information Sharing Level", ('local', 'full'), format_func=lambda x: x.title(), help="**Local:** You and the AI agents can only see your own inventory and incoming orders. **Full:** Everyone can see the entire supply chain's status and the true end-customer demand.")
+        
         if st.button("🚀 Start Game", type="primary", disabled=(client is None)):
             init_game_state(llm_personality, info_sharing)
             st.rerun()
@@ -359,40 +376,31 @@ else:
                     del st.session_state.final_order_input
                     st.rerun()
 
-        # =============== NEW SECTION STARTS HERE ===============
         st.markdown("---")
         with st.expander("📖 Your Weekly Decision Log", expanded=False):
             if not state['logs']:
                 st.write("Your weekly history will be displayed here after you complete the first week.")
             else:
                 history_df = pd.json_normalize(state['logs'])
-                
-                # Define columns to display for the human player
                 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}.weekly_cost': 'Weekly Cost',
+                    '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}.weekly_cost': 'Weekly Cost',
                 }
-                
-                # Filter and rename columns
-                display_df = history_df[list(human_cols.keys())].rename(columns=human_cols)
-                
-                # Format cost column
-                display_df['Weekly Cost'] = display_df['Weekly Cost'].apply(lambda x: f"${x:,.2f}")
-                
-                # Display dataframe, sorted with the latest week on top
-                st.dataframe(
-                    display_df.sort_values(by="Week", ascending=False),
-                    hide_index=True,
-                    use_container_width=True
-                )
-        # =============== NEW SECTION ENDS HERE =================
+                display_cols = [col for col in human_cols.keys() if col in history_df.columns]
+                display_df = history_df[display_cols].rename(columns=human_cols)
+                if 'Weekly Cost' in display_df:
+                    display_df['Weekly Cost'] = display_df['Weekly Cost'].apply(lambda x: f"${x:,.2f}")
+                st.dataframe(display_df.sort_values(by="Week", ascending=False), hide_index=True, use_container_width=True)
 
+        # =============== IMAGE ADDED TO SIDEBAR ===============
+        try:
+            st.sidebar.image(IMAGE_PATH, caption="Supply Chain Reference")
+        except FileNotFoundError:
+            st.sidebar.warning("Image file not found.")
+        # ======================================================
+            
         st.sidebar.header("Game Info")
         st.sidebar.markdown(f"**Game ID**: `{state['participant_id']}`\n\n**Current Week**: {week}")
         if st.sidebar.button("🔄 Reset Game"):