preq results page

pages/prequiz.py

@@ -22,6 +22,7 @@ for index, row in prequiz_qs.iterrows():
 
 def on_submit():
   duration = time.time() - st.session_state.prequiz_start_time
+  st.session_state.pre_quiz_answers = questions
   corr = []
   for index, row in prequiz_qs.iterrows():
     correct_answer = row[row['Correct Ans.']]
@@ -29,16 +30,12 @@ def on_submit():
       corr.append(1)
     else:
       corr.append(0)
+  st.session_state.pre_quiz_correct = corr
   prequiz_qs['Correct'] = corr
   pqq_processed = prequiz_qs.groupby('Subtopic').agg(num_correct=('Correct', 'sum'), num_questions=('Correct', 'count')).reset_index()
   st.session_state.prequiz_df = pqq_processed
   push_prequiz_data(corr, duration)
-  if st.session_state.group_id == "A":
-    st.session_state.tutor_start_time = time.time()
-    st.switch_page("pages/llm_tutor.py")
-  else:
-    st.session_state.textbook_start_time = time.time()
-    st.switch_page("pages/textbook.py")
+  st.switch_page("pages/prequiz_results.py")
   
       
 

pages/prequiz_results.py

@@ -0,0 +1,45 @@
+import streamlit as st 
+import pandas as pd
+import time
+
+df = pd.read_csv("./LSATLR_questions.csv")
+df['qid'] = df['qid'].astype(int)
+prequiz_qs = df[df['qid'] < 0].sort_values(by='qid', ascending=False).reset_index(drop=True)
+
+st.title("Pre-Quiz")
+st.write("Results:")
+
+def response_to_idx(response, row):
+  if response == row['A']:
+    return 0
+  elif response == row['B']:
+    return 1
+  elif response == row['C']:
+    return 2
+  elif response == row['D']:
+    return 3
+  elif response == row['E']:
+    return 4
+
+questions = []
+for index, row in prequiz_qs.iterrows():
+  check_or_x = '✅' if st.session_state.pre_quiz_correct[index] == 1 else '❌'
+  st.write(f"Question {index + 1}: {check_or_x}")
+  questions.append(st.radio(
+    row['Question'],
+    [row['A'], row['B'], row['C'], row['D'], row['E']],
+    key=f'prequiz{row["qid"]}',
+    index=response_to_idx(st.session_state.pre_quiz_answers[index], row),
+    disabled=True
+  ))
+  st.divider()
+
+btn = st.button("Click here when finished")
+
+if btn:
+  if st.session_state.group_id == "A":
+    st.session_state.tutor_start_time = time.time()
+    st.switch_page("pages/llm_tutor.py")
+  else:
+    st.session_state.textbook_start_time = time.time()
+    st.switch_page("pages/textbook.py")