Updates

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import importlib
 import logging
 from fpdf import FPDF
+import uuid
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -18,6 +19,10 @@ module_names = {
     "Subtraction": "subtraction_bases",
 }
 
+# Initialize unique session state
+if 'session_id' not in st.session_state:
+    st.session_state.session_id = str(uuid.uuid4())
+
 # Initialize session state variables
 if 'questions' not in st.session_state:
     st.session_state.questions = []
@@ -121,9 +126,7 @@ module_name = st.sidebar.radio("Choose a module:", list(module_names.keys()), in
 if module_name != st.session_state.current_module:
     st.session_state.current_module = module_name
     st.session_state.current_index = 0
-
-    # Generate the first question for the new module if it's a new module
-    st.session_state.questions.append(generate_new_question(module_name))
+    st.session_state.questions = [generate_new_question(module_name)]
 
 # Load the current module for title and description
 current_module = load_module(st.session_state.current_module)
@@ -153,7 +156,20 @@ with col3:
 # Display the current question
 st.write(f"**Question {st.session_state.current_index + 1}:** {current_question['question']}")
 
-selected_answer = st.radio("Choose an answer:", current_question['options'], index=current_question['options'].index(current_question['selected']) if current_question['selected'] else None, key=st.session_state.current_index)
+# Option highlighting logic
+option_styles = {
+    'correct': "background-color:#d4edda;padding:10px;border-radius:5px;",
+    'incorrect': "background-color:#f8d7da;padding:10px;border-radius:5px;",
+    'default': "padding:10px;border-radius:5px;"
+}
+
+selected_answer = st.radio(
+    "Choose an answer:", 
+    current_question['options'], 
+    index=current_question['options'].index(current_question['selected']) if current_question['selected'] else None, 
+    key=st.session_state.current_index,
+    format_func=lambda x: f"{x}"
+)
 
 if st.button("Submit"):
     if not current_question['answered']:
@@ -165,12 +181,12 @@ if st.button("Submit"):
             st.session_state.correct_count += 1
             st.session_state.module_correct_count[module_name] += 1
 
-# Highlight the correct and incorrect answers
-if current_question['answered']:
-    for option in current_question['options']:
+# Retain and highlight the options after submission
+for option in current_question['options']:
+    if current_question['answered']:
         if option == current_question['correct']:
-            st.markdown(f"<div style='background-color:#d4edda;padding:10px;border-radius:5px;'>{option}</div>", unsafe_allow_html=True)
+            st.markdown(f"<div style='{option_styles['correct']}'>{option}</div>", unsafe_allow_html=True)
         elif option == current_question['selected']:
-            st.markdown(f"<div style='background-color:#f8d7da;padding:10px;border-radius:5px;'>{option}</div>", unsafe_allow_html=True)
+            st.markdown(f"<div style='{option_styles['incorrect']}'>{option}</div>", unsafe_allow_html=True)
         else:
-            st.markdown(f"<div style='padding:10px;border-radius:5px;'>{option}</div>", unsafe_allow_html=True)
+            st.markdown(f"<div style='{option_styles['default']}'>{option}</div>", unsafe_allow_html=True)