Updates

app.py

@@ -25,6 +25,8 @@ if 'submit_disabled' not in st.session_state:
     st.session_state.submit_disabled = True
 if 'next_disabled' not in st.session_state:
     st.session_state.next_disabled = True
+if 'selected_answer' not in st.session_state:
+    st.session_state.selected_answer = None
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -133,6 +135,7 @@ if selected_module != st.session_state.current_module:
     st.session_state.module_correct_count[selected_module] = 0
     st.session_state.submit_disabled = True
     st.session_state.next_disabled = True
+    st.session_state.selected_answer = None
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -174,19 +177,20 @@ with st.form(key=f'question_form_{st.session_state.current_index}'):
     )
     submit_button = st.form_submit_button(label="Submit", disabled=st.session_state.submit_disabled)
 
-if selected_answer and st.session_state.submit_disabled:
+if selected_answer:
     st.session_state.submit_disabled = False
+    st.session_state.selected_answer = selected_answer
 
 if submit_button and not current_question.get('answered', False):
-    if selected_answer is None:
+    if st.session_state.selected_answer is None:
         st.warning("Please select an answer before submitting.")
     else:
         # Process the answer
-        current_question['selected'] = selected_answer
+        current_question['selected'] = st.session_state.selected_answer
         current_question['answered'] = True
         st.session_state.module_question_count[selected_module] += 1
 
-        if selected_answer == current_question['correct_answer']:
+        if st.session_state.selected_answer == current_question['correct_answer']:
             st.session_state.correct_count += 1
             st.session_state.module_correct_count[selected_module] += 1