Updates

app.py

@@ -195,6 +195,12 @@ if submit_button:
 
             # Show correct/incorrect feedback and explanation
             st.session_state.button_label = "Next"
+        else:
+            # Generate a new question and reset the button label
+            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"
 
 # Show correct/incorrect feedback after submission
 if current_question.get('answered', False):
@@ -211,10 +217,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)
-    
-    # Handle switching to the next question on button click
-    if submit_button and st.session_state.button_label == "Next":
-        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"