Updates

app.py

@@ -27,6 +27,8 @@ if 'submit_enabled' not in st.session_state:
     st.session_state.submit_enabled = False
 if 'new_enabled' not in st.session_state:
     st.session_state.new_enabled = False
+if 'action' not in st.session_state:
+    st.session_state.action = None
 
 def reset_pdf_cache():
     st.session_state.pdf_data = None
@@ -102,6 +104,7 @@ def generate_new_question(module_name, module):
     question_data['selected'] = None
     st.session_state.submit_enabled = False  # Disable submit initially
     st.session_state.new_enabled = False  # Disable new initially
+    st.session_state.action = None  # Reset action
     return question_data
 
 def navigate_question(direction):
@@ -176,15 +179,18 @@ if selected_answer:
     st.session_state.submit_enabled = True
 
 # Submit/New actions as radio buttons
-action = st.radio(
-    "Choose an action:",
-    ("Submit", "New"),
-    index=-1,  # Do not pre-select any action
-    key="action_radio",
-    disabled=not st.session_state.submit_enabled and not st.session_state.new_enabled,
-    horizontal=True
-)
-
+if st.session_state.submit_enabled or st.session_state.new_enabled:
+    action = st.radio(
+        "Choose an action:",
+        ("Submit", "New"),
+        index=0 if st.session_state.submit_enabled else 1,
+        key="action_radio",
+        horizontal=True
+    )
+else:
+    action = None
+
+# Handle the selected action
 if action == "Submit" and st.session_state.submit_enabled:
     # Process the answer
     current_question['selected'] = selected_answer
@@ -212,8 +218,10 @@ if action == "Submit" and st.session_state.submit_enabled:
     # Disable submit and enable new
     st.session_state.submit_enabled = False
     st.session_state.new_enabled = True
+    st.session_state.action = "New"
 
 elif action == "New" and st.session_state.new_enabled:
     navigate_question("new")
     st.session_state.submit_enabled = False
     st.session_state.new_enabled = False
+    st.session_state.action = None