Update app.py

app.py

@@ -982,7 +982,10 @@ def play_page():
         if st.button("Next", key=f"next_{gid}_{idx}"):
             # Record answer and time
             st.session_state['answers'][idx] = choice
-            st.session_state['answer_times'][idx] = time.time() - st.session_state['question_started_at']
+            start_time = st.session_state.get('question_started_at') or time.time()
+            st.session_state['answer_times'][idx] = time.time() - start_time
+
+           # st.session_state['answer_times'][idx] = time.time() - st.session_state['question_started_at']
             # Move to next question
             st.session_state['current_index'] = idx + 1
             st.session_state['question_started_at'] = time.time()
@@ -994,7 +997,7 @@ def play_page():
             if st.button("Submit All Answers", key=f"submit_{gid}_{idx}"):
                 # Record last question
                 st.session_state['answers'][idx] = choice
-                st.session_state['answer_times'][idx] = time.time() - st.session_state['question_started_at']
+                st.session_state['answer_times'][idx] = time.time() - (st.session_state.get('question_started_at') or time.time())
 
                 answers = st.session_state['answers']
                 times = st.session_state['answer_times']