Updates

app.py

@@ -136,33 +136,51 @@ st.write(current_question["question"])
 
 # Option highlighting logic
 def get_option_style(option):
-    if current_question.get('answered', False):  # Use .get() with default False
+    if current_question.get('answered', False):
         if option == current_question['correct_answer']:
             return "background-color:#d4edda;padding:10px;border-radius:5px;"  # Green background for correct
         elif option == current_question['selected']:
             return "background-color:#f8d7da;padding:10px;border-radius:5px;"  # Red background for incorrect
     return "background-color:#f0f0f0;padding:10px;border-radius:5px;"  # Gray background by default
 
-def format_option(option, option_style):
-    return f"<label style='{option_style}'>{option}</label>"
-
-# Display options with custom formatting
-selected_answer = st.radio(
-    "Choose an answer:",
-    options=[format_option(option, get_option_style(option)) for option in current_question['options']],
-    format_func=lambda x: x,
-    key=st.session_state.current_index,
-    label_visibility="collapsed",
-)
-
-if st.button("Submit"):
-    if not current_question.get('answered', False):  # Check if the question has already been answered
-        current_question['selected'] = selected_answer
-        current_question['answered'] = True
-        st.session_state.module_question_count[module_name] += 1
-
-        if selected_answer == current_question['correct_answer']:
-            st.session_state.correct_count += 1
-            st.session_state.module_correct_count[module_name] += 1
-
-# Since we're not adding an additional section, remove any content below the options.
+# Display each option with custom styles
+selected_answer = None
+for i, option in enumerate(current_question['options']):
+    option_html = f"<div style='{get_option_style(option)}'>{option}</div>"
+    if st.button(option_html, key=f"option_{i}"):
+        selected_answer = option
+
+if selected_answer:
+    current_question['selected'] = selected_answer
+    current_question['answered'] = True
+    st.session_state.module_question_count[module_name] += 1
+
+    if selected_answer == current_question['correct_answer']:
+        st.session_state.correct_count += 1
+        st.session_state.module_correct_count[module_name] += 1
+
+    # Refresh the display after submission to show the correct/incorrect styles
+    st.experimental_rerun()
+
+# Navigation buttons
+col1, col2, col3 = st.columns([1, 1, 2])
+with col1:
+    if st.button("⬅️ Prev", disabled=st.session_state.current_index == 0):
+        navigate_question("prev")
+        st.experimental_rerun()
+
+with col2:
+    if st.button("➡️ Next"):
+        navigate_question("next")
+        st.experimental_rerun()
+
+with col3:
+    if len(st.session_state.questions) > 0:
+        pdf = generate_pdf_report()
+        st.session_state.pdf_data = pdf  # Reset PDF cache
+        st.download_button(
+            label="Download PDF Report 📄",
+            data=st.session_state.pdf_data,
+            file_name="quiz_report.pdf",
+            mime="application/pdf"
+        )