Updates

app.py

@@ -117,6 +117,11 @@ if selected_module != st.session_state.current_module:
     st.session_state.current_index = len(st.session_state.questions)  # Continue numbering from previous questions
     st.session_state.selected_answer = None
     st.session_state.button_label = "Submit/New"
+    # Initialize question count and correct count if not already done
+    if selected_module not in st.session_state.module_question_count:
+        st.session_state.module_question_count[selected_module] = 0
+    if selected_module not in st.session_state.module_correct_count:
+        st.session_state.module_correct_count[selected_module] = 0
     # Generate a new question without adding it to the answered list yet
     st.session_state.current_question = generate_new_question(selected_module, modules[selected_module])