Updates

app.py

@@ -23,10 +23,6 @@ if 'pdf_data' not in st.session_state:
     st.session_state.pdf_data = None
 if 'selected_answer' not in st.session_state:
     st.session_state.selected_answer = None
-if 'submit_enabled' not in st.session_state:
-    st.session_state.submit_enabled = True
-if 'new_enabled' not in st.session_state:
-    st.session_state.new_enabled = False
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -113,8 +109,6 @@ def navigate_question(direction):
         new_question = generate_new_question(st.session_state.current_module, modules[st.session_state.current_module])
         st.session_state.questions.append(new_question)
         st.session_state.current_index = len(st.session_state.questions) - 1
-        st.session_state.submit_enabled = True
-        st.session_state.new_enabled = False
 
 # Load all modules dynamically
 modules = load_modules()
@@ -136,8 +130,6 @@ if selected_module != st.session_state.current_module:
     st.session_state.module_question_count[selected_module] = 0
     st.session_state.module_correct_count[selected_module] = 0
     st.session_state.selected_answer = None
-    st.session_state.submit_enabled = True
-    st.session_state.new_enabled = False
 
 # Load the current module's question
 current_question = st.session_state.questions[st.session_state.current_index]
@@ -167,23 +159,24 @@ with col3:
 
 st.write(current_question["question"])
 
-# Create the form for the question
-selected_answer = st.radio(
-    "Choose an answer:", 
-    options=current_question['options'], 
-    key=f"question_{st.session_state.current_index}_options",
-    index=None  # Ensure no option is pre-selected
+# Submit/New actions as radio buttons
+action = st.radio(
+    "Choose an action:",
+    ("Submit", "New"),
+    index=0 if not current_question['answered'] else 1,
+    key="action_radio",
+    horizontal=True
 )
 
-col1, col2 = st.columns(2)
-with col1:
-    submit_button = st.button(label="Submit", disabled=not st.session_state.submit_enabled)
-with col2:
-    new_button = st.button(label="New", disabled=not st.session_state.new_enabled)
+if action == "Submit":
+    selected_answer = st.radio(
+        "Choose an answer:",
+        options=current_question['options'],
+        key=f"question_{st.session_state.current_index}_options",
+        index=None  # Ensure no option is pre-selected
+    )
 
-# Handle button state and answer submission
-if submit_button:
-    if selected_answer is not None:
+    if selected_answer:
         # Process the answer
         current_question['selected'] = selected_answer
         current_question['answered'] = True
@@ -193,10 +186,6 @@ if submit_button:
             st.session_state.correct_count += 1
             st.session_state.module_correct_count[selected_module] += 1
 
-        # Immediately disable Submit and enable New
-        st.session_state.submit_enabled = False
-        st.session_state.new_enabled = True
-
         # Display correct/incorrect feedback
         for option in current_question['options']:
             if option == current_question['correct_answer']:
@@ -211,9 +200,5 @@ if submit_button:
         for step in current_question['step_by_step_solution']:
             st.write(step)
 
-    else:
-        st.warning("Please select an option before submitting.", icon="⚠️")
-
-# Handle new question generation
-if new_button:
+elif action == "New":
     navigate_question("new")