Update app.py

app.py

@@ -39,6 +39,30 @@ FACTORY_SHIPPING_DELAY = 1
 HOLDING_COST = 0.5
 BACKLOG_COST = 1.0
 
+# --- NEW: Comprehension Questions Data ---
+COMPREHENSION_QUESTIONS = [
+    {
+        "q": "1. Which role are you playing in this supply chain experiment?",
+        "options": ["Retailer", "Wholesaler", "Distributor", "Factory"],
+        "correct_index": 2, # Distributor
+    },
+    {
+        "q": "2. What is your primary objective as a manager in this game?",
+        "options": ["Maximize on-hand inventory.", "Minimize total accumulated costs (holding + backlog).", "Maximize weekly order quantity.", "Minimize customer demand."],
+        "correct_index": 1, # Minimize total costs
+    },
+    {
+        "q": "3. Your final order quantity for the Factory will typically arrive at your location after how many weeks?",
+        "options": ["1 week", "2 weeks", "3 weeks", "4 weeks"],
+        "correct_index": 2, # 3 weeks
+    },
+    {
+        "q": "4. At the end of the week, your holding cost is $0.50 per unit, and your backlog cost is $1.00 per unit. If your final inventory is 0 units and you have a remaining Backlog of 15 units, what is your Total Weekly Cost?",
+        "options": ["$7.50", "$15.00", "$15.50", "$0.00"],
+        "correct_index": 1, # 15.00
+    },
+]
+
 # --- Model & Log Configuration ---
 OPENAI_MODEL = "gpt-4o-mini"
 LOCAL_LOG_DIR = Path("logs")
@@ -619,6 +643,9 @@ if 'consent_given' not in st.session_state:
 # --- NEW: Storage for Consent Timestamp ---
 if 'consent_timestamp' not in st.session_state:
     st.session_state['consent_timestamp'] = None
+# --- NEW: Check for Comprehension ---
+if 'comprehension_passed' not in st.session_state:
+    st.session_state['comprehension_passed'] = False
 
 # --- Initialization Check ---
 if st.session_state.get('initialization_error'):
@@ -628,13 +655,10 @@ if st.session_state.get('initialization_error'):
 elif not st.session_state['consent_given']:
     st.header("📝 Participant Consent Form")
     st.markdown("""
-
     **Lead Researcher:** Xinyu Li
     **Supervisor:** Professor Li Ding & Dr Yanlu Zhao
     **Contact Email:** xinyu.li3@durham.ac.uk
-
     Please read the following statements carefully. If you have any questions, please ask the researcher or contact them at the email address provided above.
-
     By agreeing to participate, you are confirming that you understand all of the following:
     
     - You have listened to the in-class explanation of this study and have had the opportunity to ask questions.
@@ -671,6 +695,42 @@ elif not st.session_state['consent_given']:
             else:
                 st.warning("Please select an option to continue.")
 
+# --- NEW: Comprehension Questions Section ---
+elif not st.session_state['comprehension_passed']:
+    st.header("🧠 Comprehension Check")
+    st.markdown("Before starting the experiment, please answer the following questions to ensure you understand the game rules and your role.")
+    
+    with st.form("comprehension_quiz"):
+        user_answers = {}
+        for i, q_data in enumerate(COMPREHENSION_QUESTIONS):
+            st.subheader(q_data['q'])
+            # Create a radio button for each question
+            user_answers[i] = st.radio(
+                "Select an option:",
+                q_data['options'],
+                key=f"comp_q_{i}",
+                index=None, # Require explicit selection
+                label_visibility="collapsed"
+            )
+            st.markdown("---")
+            
+        submit_quiz = st.form_submit_button("Submit Answers")
+        
+        if submit_quiz:
+            all_correct = True
+            for i, q_data in enumerate(COMPREHENSION_QUESTIONS):
+                selected = user_answers.get(i)
+                correct_option = q_data['options'][q_data['correct_index']]
+                
+                if selected != correct_option:
+                    all_correct = False
+                    st.error(f"❌ Question {i+1} is incorrect.")
+            
+            if all_correct:
+                st.success("✅ All answers are correct! Proceeding to game setup...")
+                st.session_state['comprehension_passed'] = True
+                time.sleep(1.5) # Brief pause for user to see success
+                st.rerun()
 
 # --- Main Application Flow ---
 else:
@@ -724,6 +784,7 @@ else:
             # 在重置游戏时清除所有与实验相关的时间戳
             if 'consent_timestamp' in st.session_state: del st.session_state['consent_timestamp']
             if 'consent_given' in st.session_state: del st.session_state['consent_given']
+            if 'comprehension_passed' in st.session_state: del st.session_state['comprehension_passed']
             del st.session_state.game_state
             st.rerun()