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 'show_next_button' not in st.session_state:
-    st.session_state.show_next_button = False
+if 'button_label' not in st.session_state:
+    st.session_state.button_label = "Submit"
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -107,10 +107,10 @@ 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.show_next_button = True
+        st.session_state.button_label = "Next Question"
     elif direction == "next" and st.session_state.current_index < len(st.session_state.questions) - 1:
         st.session_state.current_index += 1
-        st.session_state.show_next_button = True
+        st.session_state.button_label = "Next Question"
 
 # Load all modules dynamically
 modules = load_modules()
@@ -132,7 +132,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.show_next_button = False
+    st.session_state.button_label = "Submit"
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -171,19 +171,16 @@ with st.form(key=f'question_form_{st.session_state.current_index}'):
         index=None  # Ensure no option is pre-selected
     )
     
-    # Determine button label
-    button_label = "Next Question" if st.session_state.show_next_button else "Submit"
-    
-    submit_button = st.form_submit_button(label=button_label)
+    submit_button = st.form_submit_button(label=st.session_state.button_label)
 
 # Handle button state and answer submission
 if submit_button:
-    if st.session_state.show_next_button:
+    if st.session_state.button_label == "Next Question":
         # Move to the 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.show_next_button = False
+        st.session_state.button_label = "Submit"
     else:
         if selected_answer is not None:
             # Process the answer
@@ -195,7 +192,7 @@ if submit_button:
                 st.session_state.correct_count += 1
                 st.session_state.module_correct_count[selected_module] += 1
 
-            st.session_state.show_next_button = True
+            st.session_state.button_label = "Next Question"
 
 # Show correct/incorrect feedback after submission
 if current_question.get('answered', False):