Update app.py

app.py

@@ -26,10 +26,16 @@ from huggingface_hub import upload_file, HfApi
 # CONFIG
 # ---------------------------
 DEFAULT_WEEKS = 36          # 24 或 36 可选，默认 36
-TRANSPORT_DELAY = 2
+
+# Lead times
+ORDER_LEAD_TIME = 1         # Time for orders to reach supplier
+SHIPPING_LEAD_TIME = 2      # Time for shipments to arrive
+PRODUCTION_LEAD_TIME = 2    # Time for factory to produce goods
+
 INITIAL_INVENTORY = 12
 INITIAL_BACKLOG = 0
 
+
 OPENAI_MODEL = "gpt-4o-mini"   
 
 LOCAL_LOG_DIR = Path("logs")
@@ -82,15 +88,15 @@ def ensure_state_compat(state: dict):
     state.setdefault("inventory", {r: INITIAL_INVENTORY for r in roles})
     state.setdefault("backlog", {r: INITIAL_BACKLOG for r in roles})
 
-    # pipeline: ensure lists and proper length >= TRANSPORT_DELAY
+    # pipeline: ensure lists and proper length >= SHIPPING_LEAD_TIME
     if "pipeline" not in state:
-        state["pipeline"] = {r: [0] * TRANSPORT_DELAY for r in roles}
+        state["pipeline"] = {r: [4] * SHIPPING_LEAD_TIME for r in roles}
     else:
         for r in roles:
-            state["pipeline"].setdefault(r, [0] * TRANSPORT_DELAY)
-            # pad if shorter than TRANSPORT_DELAY
-            if len(state["pipeline"][r]) < TRANSPORT_DELAY:
-                state["pipeline"][r] = state["pipeline"][r] + [0] * (TRANSPORT_DELAY - len(state["pipeline"][r]))
+            state["pipeline"].setdefault(r, [4] * SHIPPING_LEAD_TIME)
+            # pad if shorter than SHIPPING_LEAD_TIME
+            if len(state["pipeline"][r]) < SHIPPING_LEAD_TIME:
+                state["pipeline"][r] = state["pipeline"][r] + [4] * (SHIPPING_LEAD_TIME - len(state["pipeline"][r]))
 
     # incoming_orders, orders_history, shipments_history
     state.setdefault("incoming_orders", {r: 0 for r in roles})
@@ -192,7 +198,7 @@ def init_game(weeks=DEFAULT_WEEKS):
         "roles": roles,
         "inventory": {r: INITIAL_INVENTORY for r in roles},
         "backlog": {r: INITIAL_BACKLOG for r in roles},
-        "pipeline": {r: [0] * TRANSPORT_DELAY for r in roles},
+        "pipeline": {r: [4] * SHIPPING_LEAD_TIME for r in roles},
         "incoming_orders": {r: 0 for r in roles},
         "orders_history": {r: [] for r in roles},
         "shipments_history": {r: [] for r in roles},
@@ -277,7 +283,7 @@ def step_game(state: dict, distributor_order: int):
         elif role == "wholesaler":
             state["incoming_orders"]["factory"] = order_val
 
-    # 6) place orders into pipelines (will arrive after TRANSPORT_DELAY)
+    # 6) place orders into pipelines (will arrive after SHIPPING_LEAD_TIME)
     downstream_map = {"factory": "wholesaler", "wholesaler": "distributor", "distributor": "retailer", "retailer": None}
     for role in roles:
         placed_order = state["orders_history"][role][-1] if state["orders_history"].get(role) else 0
@@ -285,9 +291,20 @@ def step_game(state: dict, distributor_order: int):
             placed_order = int(distributor_order)
         downstream = downstream_map.get(role)
         if downstream:
-            state["pipeline"].setdefault(downstream, [0]*TRANSPORT_DELAY)
+            state["pipeline"].setdefault(downstream, [0]*SHIPPING_LEAD_TIME)
             state["pipeline"][downstream].append(placed_order)
 
+    # 6.5) cost calculation
+    if "cost" not in state:
+        state["cost"] = {r: 0.0 for r in roles}
+
+    for r in roles:
+        inv = state["inventory"].get(r, 0)
+        backlog = state["backlog"].get(r, 0)
+        inv_cost = inv * 0.5     # 每单位库存成本
+        back_cost = backlog * 1.0  # 每单位缺货成本
+        state["cost"][r] = state["cost"].get(r, 0) + inv_cost + back_cost
+        
     # 7) logging
     log_entry = {
         "timestamp": now_iso(),
@@ -298,6 +315,7 @@ def step_game(state: dict, distributor_order: int):
         "orders_submitted": {r: (state["orders_history"].get(r, [None])[-1] if state["orders_history"].get(r) else None) for r in roles},
         "inventory": {r: state["inventory"].get(r, 0) for r in roles},
         "backlog": {r: state["backlog"].get(r, 0) for r in roles},
+        "cost": {r: state["cost"].get(r, 0) for r in roles},
         "info_sharing": state.get("info_sharing", False),
         "info_history_weeks": state.get("info_history_weeks", 0),
         "llm_raw": {k: v["raw"] for k, v in llm_outputs.items()}