Updates

app.py

@@ -157,11 +157,13 @@ with col3:
 st.write(f"**Question {st.session_state.current_index + 1}:** {current_question['question']}")
 
 # 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;"
-}
+def get_option_style(option):
+    if current_question['answered']:
+        if option == current_question['correct']:
+            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
 
 selected_answer = st.radio(
     "Choose an answer:", 
@@ -183,10 +185,4 @@ if st.button("Submit"):
 
 # 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='{option_styles['correct']}'>{option}</div>", unsafe_allow_html=True)
-        elif option == current_question['selected']:
-            st.markdown(f"<div style='{option_styles['incorrect']}'>{option}</div>", unsafe_allow_html=True)
-        else:
-            st.markdown(f"<div style='{option_styles['default']}'>{option}</div>", unsafe_allow_html=True)
+    st.markdown(f"<div style='{get_option_style(option)}'>{option}</div>", unsafe_allow_html=True)