Updates

app.py

@@ -109,6 +109,12 @@ def navigate_question(direction):
         st.session_state.current_index -= 1
     elif direction == "next" and st.session_state.current_index < len(st.session_state.questions) - 1:
         st.session_state.current_index += 1
+    elif direction == "new":
+        new_question = generate_new_question(st.session_state.current_module, modules[st.session_state.current_module])
+        st.session_state.questions.append(new_question)
+        st.session_state.current_index = len(st.session_state.questions) - 1
+        st.session_state.submit_enabled = True
+        st.session_state.new_enabled = False
 
 # Load all modules dynamically
 modules = load_modules()
@@ -205,17 +211,9 @@ if submit_button:
         for step in current_question['step_by_step_solution']:
             st.write(step)
 
-        # Rerun to update the button states
-        st.experimental_rerun()
-
     else:
         st.warning("Please select an option before submitting.", icon="⚠️")
 
 # Handle new question generation
 if new_button:
-    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.submit_enabled = True
-    st.session_state.new_enabled = False
-    st.experimental_rerun()
+    navigate_question("new")