Updates

app.py

@@ -25,6 +25,8 @@ if 'selected_answer' not in st.session_state:
     st.session_state.selected_answer = None
 if 'answered' not in st.session_state:
     st.session_state.answered = False
+if 'button_clicked' not in st.session_state:
+    st.session_state.button_clicked = False
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -70,7 +72,7 @@ def generate_pdf_report():
                 pdf.multi_cell(0, 10, f"Explanation: {explanation}")
                 pdf.ln(5)
                 pdf.multi_cell(0, 10, "Step-by-Step Solution:")
-                for step in step_by_step_solution:
+                for step in step_by-step_solution:
                     pdf.multi_cell(0, 10, step)
                 pdf.ln(10)
 
@@ -133,6 +135,7 @@ if selected_module != st.session_state.current_module:
     st.session_state.module_correct_count[selected_module] = 0
     st.session_state.selected_answer = None
     st.session_state.answered = False
+    st.session_state.button_clicked = False
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -190,6 +193,7 @@ if submit_button:
             # Set answered to true to disable options
             st.session_state.questions[st.session_state.current_index]['answered'] = True
             st.session_state.selected_answer = selected_answer
+            st.session_state.button_clicked = True
 
     else:
         # If already answered, move to the next question
@@ -197,6 +201,12 @@ if submit_button:
         st.session_state.questions.append(new_question)
         st.session_state.current_index = len(st.session_state.questions) - 1
         st.session_state.answered = False
+        st.session_state.button_clicked = False
+
+# Re-render the form if the button was clicked to disable the options
+if st.session_state.button_clicked:
+    st.session_state.button_clicked = False
+    st.experimental_rerun()
 
 # Show correct/incorrect feedback after submission
 if current_question.get('answered', False):