Update app.py

app.py

@@ -254,11 +254,17 @@ with welcome_message.container():
 time.sleep(4)  # Wait for 4 seconds
 welcome_message.empty()  # Remove the welcome message
 
-# Initialize session state for questions and current index
+# Initialize session state for questions, answers, and current index
 if 'questions' not in st.session_state:
     st.session_state.questions = []
+if 'answers' not in st.session_state:
+    st.session_state.answers = []
+if 'feedback' not in st.session_state:
+    st.session_state.feedback = []
 if 'question_index' not in st.session_state:
     st.session_state.question_index = 0
+if 'show_results' not in st.session_state:
+    st.session_state.show_results = False
 
 # Sidebar Navigation
 st.sidebar.title("TechPrep Navigation")
@@ -279,7 +285,10 @@ if nav_option == "Generate Questions":
         with st.spinner("Generating questions..."):
             questions = generate_questions(model_choice, role, question_type, num_questions, difficulty)
             st.session_state.questions = questions
+            st.session_state.answers = ["" for _ in questions]
+            st.session_state.feedback = ["" for _ in questions]
             st.session_state.question_index = 0
+            st.session_state.show_results = False
             progress_data["questions_solved"][question_type] += num_questions
 
     # Display questions with navigation
@@ -311,10 +320,35 @@ if nav_option == "Generate Questions":
                 else:
                     with st.spinner("Providing feedback..."):
                         feedback = provide_feedback(model_choice, answer)
+                        st.session_state.answers[index] = answer
+                        st.session_state.feedback[index] = feedback
                         st.markdown(style_output("Feedback Received:", "#FF5722"), unsafe_allow_html=True)
                         st.write(feedback)
                         progress_data["feedback_provided"] += 1
 
+            # Show results and score when all questions have been answered
+            if index == len(question_list) - 1:
+                st.session_state.show_results = True
+
+        if st.session_state.show_results:
+            st.write("### Your Results")
+            total_questions = len(st.session_state.questions)
+            answered_questions = sum(1 for ans in st.session_state.answers if ans)
+            score = (answered_questions / total_questions) * 100
+            st.write(f"**Score:** {score:.2f}%")
+            
+            # Display feedback and tips
+            st.write("### Feedback Summary")
+            for i, (q, ans, fb) in enumerate(zip(st.session_state.questions, st.session_state.answers, st.session_state.feedback)):
+                st.write(f"**Question {i + 1}:** {q}")
+                st.write(f"**Your Answer:** {ans}")
+                st.write(f"**Feedback:** {fb}")
+
+            tips = get_tips(model_choice, role)
+            st.write("### Tips to Improve")
+            st.write(tips)
+            progress_data["tips_retrieved"] += 1
+
 elif nav_option == "Mock Interview":
     st.header("🎥 Mock Interview")
     schedule_mock_interview()
@@ -332,4 +366,3 @@ st.markdown("""
     </div>
     """, unsafe_allow_html=True)
 
-