Updates

app.py

@@ -166,10 +166,10 @@ with col3:
             mime="application/pdf"
         )
 
-st.write(current_question["question"])
-
-# Show Submit/Next button or Next button depending on the state
+# Show the question only if the "Next" button hasn't been clicked yet
 if not st.session_state.show_next_button:
+    st.write(current_question["question"])
+
     # Create the form for the question
     with st.form(key=f'question_form_{st.session_state.current_index}'):
         selected_answer = st.radio(
@@ -189,7 +189,7 @@ if not st.session_state.show_next_button:
             st.session_state.submit_click_count += 1
 
             if st.session_state.submit_click_count == 1:
-                # First click: show solution and replace Submit/Next with Next
+                # First click: show solution and remove question
                 if not current_question['answered']:
                     if selected_answer is not None:
                         # Process the answer
@@ -227,10 +227,12 @@ if not st.session_state.show_next_button:
 
                 st.markdown("</ol></div>", unsafe_allow_html=True)
 
-                # Replace Submit/Next with Next
+                # Replace question and Submit/Next with Next button
                 st.session_state.show_next_button = True
-else:
-    # Show the Next button
+                st.experimental_rerun()
+
+# Show the Next button to generate a new question
+if st.session_state.show_next_button:
     if st.button("Next"):
         # Generate a new question
         new_question = generate_new_question(selected_module, modules[selected_module])
@@ -240,5 +242,4 @@ else:
         st.session_state.submit_click_count = 0  # Reset the click count
         st.session_state.selected_answer = None  # Reset the selected answer
         st.session_state.show_next_button = False  # Hide the Next button
-        # Force a re-render by modifying a state variable
-        st.session_state['last_updated'] = str(uuid.uuid4())
+        st.experimental_rerun()