Updates

app.py

@@ -25,8 +25,6 @@ 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
@@ -72,7 +70,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:  # Corrected variable name
+                for step in step_by_step_solution:
                     pdf.multi_cell(0, 10, step)
                 pdf.ln(10)
 
@@ -135,7 +133,6 @@ 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]
@@ -165,13 +162,16 @@ with col3:
 
 st.write(current_question["question"])
 
+# Disable the radio buttons if the question is already answered
+disabled_options = current_question['answered']
+
 # Create the form for the question
 with st.form(key=f'question_form_{st.session_state.current_index}'):
     selected_answer = st.radio(
         "Choose an answer:", 
         options=current_question['options'], 
         index=current_question['options'].index(current_question['selected']) if current_question['answered'] else None, 
-        disabled=current_question['answered'],  # Disable if the question has been answered
+        disabled=disabled_options,  # Disable if the question has been answered
         key=f"question_{st.session_state.current_index}_options"
     )
     
@@ -193,20 +193,6 @@ 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
-        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.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):
@@ -223,3 +209,7 @@ if current_question.get('answered', False):
     st.write("**Step-by-Step Solution:**")
     for step in current_question['step_by_step_solution']:
         st.write(step)
+
+# If submit button was pressed and question was answered, update the UI to disable options
+if current_question.get('answered', False) and not disabled_options:
+    st.experimental_rerun()