Updates

app.py

@@ -27,6 +27,8 @@ 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
+if 'show_next_button' not in st.session_state:
+    st.session_state.show_next_button = False
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -136,6 +138,7 @@ if selected_module != st.session_state.current_module:
     st.session_state.selected_answer = None
     st.session_state.answered = False
     st.session_state.submit_click_count = 0  # Reset the submit click count
+    st.session_state.show_next_button = False  # Reset the next button visibility
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -165,76 +168,77 @@ with col3:
 
 st.write(current_question["question"])
 
-# Determine the button label based on click count
-button_label = "Submit/Next" if st.session_state.submit_click_count == 0 else "Next"
-
-# 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, 
-        key=f"question_{st.session_state.current_index}_options"
-    )
-    
-    submit_button = st.form_submit_button(label=button_label)
-
-# Handle button state and answer submission
-if submit_button:
-    if selected_answer is None:
-        st.warning("Please select an option before submitting.")
-    else:
-        st.session_state.submit_click_count += 1
-
-        # Show the number of times the Submit/Next button is clicked
-        st.write(f"Submit/Next button clicked {st.session_state.submit_click_count} time(s).")
-
-        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
-            st.session_state.submit_click_count = 0  # Reset the click count
-            st.session_state.selected_answer = None  # Reset the selected answer
-            # Force a re-render by modifying a state variable
-            st.session_state['last_updated'] = str(uuid.uuid4())
+# Show Submit/Next button or Next button depending on the state
+if not st.session_state.show_next_button:
+    # 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, 
+            key=f"question_{st.session_state.current_index}_options"
+        )
+        
+        submit_button = st.form_submit_button(label="Submit/Next")
+
+    # Handle button state and answer submission
+    if submit_button:
+        if selected_answer is None:
+            st.warning("Please select an option before submitting.")
+        else:
+            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
+                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)
+
+                # Replace Submit/Next with Next
+                st.session_state.show_next_button = True
+else:
+    # Show the Next button
+    if st.button("Next"):
+        # 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
+        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())