Update app.py

app.py

@@ -272,16 +272,26 @@ def step_game(state: dict, distributor_order: int):
         demand_history_visible = state["customer_demand"][start_idx:(week - 1)]
 
     llm_outputs = {}
-    for role in ["retailer", "wholesaler", "factory"]:
-        order_val, raw = call_llm_for_order(role, state_snapshot_for_prompt(state), state.get("info_sharing", False), demand_history_visible)
+    for role in ["retailer", "wholesaler", "factory", "distributor"]:
+        order_val, raw = call_llm_for_order(
+            role, 
+            state_snapshot_for_prompt(state), 
+            state.get("info_sharing", False), 
+            demand_history_visible
+        )
         order_val = max(0, int(order_val))
-        state["orders_history"][role].append(order_val)
         llm_outputs[role] = {"order": order_val, "raw": raw}
-        # set incoming_orders for upstream parties (visible next week)
-        if role == "retailer":
-            state["incoming_orders"]["distributor"] = order_val
-        elif role == "wholesaler":
-            state["incoming_orders"]["factory"] = order_val
+
+        if role != "distributor":  # AI 决策直接生效
+            state["orders_history"][role].append(order_val)
+            if role == "retailer":
+                state["incoming_orders"]["distributor"] = order_val
+            elif role == "wholesaler":
+                state["incoming_orders"]["factory"] = order_val
+
+    # 人类 distributor 的真实 order 后面会覆盖
+    state["orders_history"]["distributor"].append(int(distributor_order))
+    state["incoming_orders"]["wholesaler"] = int(distributor_order)
 
     # 6) place orders into pipelines (will arrive after SHIPPING_LEAD_TIME)
     downstream_map = {"factory": "wholesaler", "wholesaler": "distributor", "distributor": "retailer", "retailer": None}
@@ -418,14 +428,23 @@ with col_main:
     st.markdown("---")
     # Distributor form
     with st.form(key=f"order_form_{participant_id}", clear_on_submit=False):
-        st.write("### Your (Distributor) decision this week")
-        default_val = state["incoming_orders"].get("distributor", 4) or 4
-        distributor_order = st.number_input("Order to place to upstream (Wholesaler):", min_value=0, step=1, value=default_val)
-        submitted = st.form_submit_button("Submit Order (locks your decision)")
-        if submitted:
-            st.session_state.setdefault("pending_orders", {})
-            st.session_state["pending_orders"][participant_id] = int(distributor_order)
-            st.success(f"Order submitted: {distributor_order}. Now click 'Next Week' to process the week.")
+    st.write("### Your (Distributor) decision this week")
+
+    # 如果有 LLM 给的建议，就显示出来
+    last_log = state["logs"][-1] if state.get("logs") else None
+    if last_log and "llm_raw" in last_log and "distributor" in last_log["llm_raw"]:
+        suggestion = last_log["llm_raw"]["distributor"]
+        st.info(f"💡 AI suggests you order: **{suggestion}** units (you can follow or override)")
+    else:
+        st.info("💡 AI suggestion will appear after the first processed week.")
+
+    default_val = state["incoming_orders"].get("distributor", 4) or 4
+    distributor_order = st.number_input("Order to place to upstream (Wholesaler):", min_value=0, step=1, value=default_val)
+    submitted = st.form_submit_button("Submit Order (locks your decision)")
+    if submitted:
+        st.session_state.setdefault("pending_orders", {})
+        st.session_state["pending_orders"][participant_id] = int(distributor_order)
+        st.success(f"Order submitted: {distributor_order}. Now click 'Next Week' to process the week.")
 
     st.markdown("---")
     pending = st.session_state.get("pending_orders", {}).get(participant_id, None)