Updates

app.py

@@ -23,8 +23,8 @@ if 'pdf_data' not in st.session_state:
     st.session_state.pdf_data = None
 if 'selected_answer' not in st.session_state:
     st.session_state.selected_answer = None
-if 'button_label' not in st.session_state:
-    st.session_state.button_label = "Submit/Next"
+if 'button_disabled' not in st.session_state:
+    st.session_state.button_disabled = False
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -107,10 +107,14 @@ def generate_new_question(module_name, module):
 def navigate_question(direction):
     if direction == "prev" and st.session_state.current_index > 0:
         st.session_state.current_index -= 1
-        st.session_state.button_label = "Submit/Next"
-    elif direction == "next" and st.session_state.current_index < len(st.session_state.questions) - 1:
-        st.session_state.current_index += 1
-        st.session_state.button_label = "Submit/Next"
+        st.session_state.button_disabled = True  # Disable the button when navigating to previous questions
+    elif direction == "next":
+        if st.session_state.current_index < len(st.session_state.questions) - 1:
+            st.session_state.current_index += 1
+        else:
+            st.session_state.questions.append(generate_new_question(st.session_state.current_module, modules[st.session_state.current_module]))
+            st.session_state.current_index = len(st.session_state.questions) - 1
+        st.session_state.button_disabled = False  # Enable the button for new questions
 
 # Load all modules dynamically
 modules = load_modules()
@@ -132,7 +136,7 @@ if selected_module != st.session_state.current_module:
     st.session_state.module_question_count[selected_module] = 0
     st.session_state.module_correct_count[selected_module] = 0
     st.session_state.selected_answer = None
-    st.session_state.button_label = "Submit/Next"
+    st.session_state.button_disabled = False
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -147,7 +151,7 @@ with col1:
     if st.button("⬅️ Prev", disabled=st.session_state.current_index == 0):
         navigate_question("prev")
 with col2:
-    if st.button("➡️ Next", disabled=st.session_state.current_index >= len(st.session_state.questions) - 1):
+    if st.button("➡️ Next"):
         navigate_question("next")
 with col3:
     if len(st.session_state.questions) > 0:
@@ -168,13 +172,14 @@ with st.form(key=f'question_form_{st.session_state.current_index}'):
         "Choose an answer:", 
         options=current_question['options'], 
         key=f"question_{st.session_state.current_index}_options",
-        index=None  # Ensure no option is pre-selected
+        index=current_question['options'].index(current_question['selected']) if current_question['answered'] else None, 
+        disabled=current_question['answered']  # Disable if the question has been answered
     )
     
-    submit_button = st.form_submit_button(label=st.session_state.button_label)
+    submit_button = st.form_submit_button(label="Submit/Next", disabled=st.session_state.button_disabled)
 
 # Handle button state and answer submission
-if submit_button and st.session_state.button_label == "Submit/Next":
+if submit_button:
     if selected_answer is not None:
         # Process the answer
         current_question['selected'] = selected_answer
@@ -186,7 +191,7 @@ if submit_button and st.session_state.button_label == "Submit/Next":
             st.session_state.module_correct_count[selected_module] += 1
 
         # Show correct/incorrect feedback and explanation
-        st.session_state.button_label = "Next Question"
+        st.session_state.button_disabled = True  # Disable the button after submission
 
 # Show correct/incorrect feedback after submission
 if current_question.get('answered', False):
@@ -203,13 +208,3 @@ if current_question.get('answered', False):
     st.write("**Step-by-Step Solution:**")
     for step in current_question['step_by_step_solution']:
         st.write(step)
-    
-    # Change the button label immediately after showing the solution
-    st.session_state.button_label = "Next Question"
-
-# This part handles switching to the next question on button click
-if submit_button and st.session_state.button_label == "Next Question":
-    new_question = generate_new_question(selected_module, modules[selected_module])
-    st.session_state.questions.append(new_question)
-    st.session_state.current_index = len(st.session_state.questions) - 1
-    st.session_state.button_label = "Submit/Next"