Updates

app.py

@@ -143,14 +143,28 @@ 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)
+
+if st.button("Submit"):
+    if not current_question['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']:
+            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']:
         if option == current_question['correct']:
-            st.success(f"✔ {option}")
+            st.markdown(f"<div style='background-color:#d4edda;padding:10px;border-radius:5px;'>✔ {option}</div>", unsafe_allow_html=True)
         elif option == current_question['selected']:
-            st.error(f"✘ {option}")
+            st.markdown(f"<div style='background-color:#f8d7da;padding:10px;border-radius:5px;'>✘ {option}</div>", unsafe_allow_html=True)
         else:
             st.write(option)
+    
     st.markdown(
         f"""
         <div style='border: 2px solid #ccc; padding: 15px;'>
@@ -159,16 +173,3 @@ if current_question['answered']:
         """,
         unsafe_allow_html=True
     )
-else:
-    selected_answer = st.radio("Choose an answer:", current_question['options'], key=st.session_state.current_index)
-    if st.button("Submit"):
-        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']:
-                st.session_state.correct_count += 1
-                st.session_state.module_correct_count[module_name] += 1
-
-            # Streamlit will automatically update the UI after state changes