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 'submit_click_count' not in st.session_state:
+    st.session_state.submit_click_count = 0
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -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.submit_click_count = 0  # Reset the submit click count
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -175,47 +178,52 @@ with st.form(key=f'question_form_{st.session_state.current_index}'):
 
 # Handle button state and answer submission
 if submit_button:
-    if not current_question['answered']:
-        if selected_answer is not None:
-            # Process the answer
-            current_question['selected'] = selected_answer
-            current_question['answered'] = True
-            st.session_state.module_question_count[selected_module] += 1
-
-            if selected_answer == current_question['correct_answer']:
-                st.session_state.correct_count += 1
-                st.session_state.module_correct_count[selected_module] += 1
-
-            # 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
-
-    else:
-        # If already answered, move to the next question
+    st.session_state.submit_click_count += 1
+
+    if st.session_state.submit_click_count == 1:
+        # First click: show solution
+        if not current_question['answered']:
+            if selected_answer is not None:
+                # Process the answer
+                current_question['selected'] = selected_answer
+                current_question['answered'] = True
+                st.session_state.module_question_count[selected_module] += 1
+
+                if selected_answer == current_question['correct_answer']:
+                    st.session_state.correct_count += 1
+                    st.session_state.module_correct_count[selected_module] += 1
+
+                # 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
+
+        # Show correct/incorrect feedback after submission
+        for option in current_question['options']:
+            if option == current_question['correct_answer']:
+                st.markdown(f"<span style='color:green;'>{option} ✅</span>", unsafe_allow_html=True)
+            elif option == current_question['selected']:
+                st.markdown(f"<span style='color:red;'>{option} ❌</span>", unsafe_allow_html=True)
+            else:
+                st.markdown(f"{option}")
+
+        # Show explanation and step-by-step solution with improved visuals
+        st.write(f"**Explanation:** {current_question['explanation']}")
+        st.markdown("""
+        <div style='border: 2px solid #4CAF50; padding: 10px; margin-top: 20px; background-color: #f9f9f9;'>
+        <h4 style='color: #4CAF50;'>Step-by-Step Solution:</h4>
+        <ol>
+        """, unsafe_allow_html=True)
+
+        for step in current_question['step_by_step_solution']:
+            st.markdown(f"<li style='margin-bottom: 10px;'>{step}</li>", unsafe_allow_html=True)
+
+        st.markdown("</ol></div>", unsafe_allow_html=True)
+
+    elif st.session_state.submit_click_count == 2:
+        # Second click: generate a new 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
-
-# Show correct/incorrect feedback after submission
-if current_question.get('answered', False):
-    for option in current_question['options']:
-        if option == current_question['correct_answer']:
-            st.markdown(f"<span style='color:green;'>{option} ✅</span>", unsafe_allow_html=True)
-        elif option == current_question['selected']:
-            st.markdown(f"<span style='color:red;'>{option} ❌</span>", unsafe_allow_html=True)
-        else:
-            st.markdown(f"{option}")
-    
-    # Show explanation and step-by-step solution with improved visuals
-    st.write(f"**Explanation:** {current_question['explanation']}")
-    st.markdown("""
-    <div style='border: 2px solid #4CAF50; padding: 10px; margin-top: 20px; background-color: #f9f9f9;'>
-    <h4 style='color: #4CAF50;'>Step-by-Step Solution:</h4>
-    <ol>
-    """, unsafe_allow_html=True)
-    
-    for step in current_question['step_by_step_solution']:
-        st.markdown(f"<li style='margin-bottom: 10px;'>{step}</li>", unsafe_allow_html=True)
-    
-    st.markdown("</ol></div>", unsafe_allow_html=True)
+        st.session_state.submit_click_count = 0  # Reset the click count
+        st.experimental_rerun()  # Rerun to load the new question