Updates

app.py

@@ -107,10 +107,14 @@ def generate_new_question(module_name, module):
 def navigate_question(direction):
     if direction == "prev" and st.session_state.current_index > 0:
         st.session_state.current_index -= 1
-        st.session_state.button_label = "Next Question"
-    elif direction == "next" and st.session_state.current_index < len(st.session_state.questions) - 1:
-        st.session_state.current_index += 1
-        st.session_state.button_label = "Next Question"
+        st.session_state.button_label = "Submit"
+    elif direction == "next":
+        if st.session_state.current_index < len(st.session_state.questions) - 1:
+            st.session_state.current_index += 1
+        else:
+            st.session_state.questions.append(generate_new_question(st.session_state.current_module, modules[st.session_state.current_module]))
+            st.session_state.current_index = len(st.session_state.questions) - 1
+        st.session_state.button_label = "Submit"
 
 # Load all modules dynamically
 modules = load_modules()
@@ -147,7 +151,7 @@ with col1:
     if st.button("⬅️ Prev", disabled=st.session_state.current_index == 0):
         navigate_question("prev")
 with col2:
-    if st.button("➡️ Next", disabled=st.session_state.current_index >= len(st.session_state.questions) - 1):
+    if st.button("➡️ Next"):
         navigate_question("next")
 with col3:
     if len(st.session_state.questions) > 0: