Update app.py

app.py

@@ -22,10 +22,8 @@ progress_data = {
 }
 
 def get_llm(model_choice):
-    if model_choice == "Gemini":
+    if model_choice == "Gemini" or model_choice == "Groq":
         return ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
-    elif model_choice == "Groq":
-        return ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)  # Groq uses Gemini AI
     elif model_choice == "OpenAI":
         return None
     else:
@@ -43,7 +41,7 @@ def generate_questions(model_choice, role, question_type, num_questions, difficu
             max_tokens=150
         )
         return response.choices[0].text.strip().split("\n")
-    elif model_choice in ["Gemini", "Groq"]:
+    elif model_choice == "Gemini" or model_choice == "Groq":
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
         return response.content.split("\n")
@@ -59,7 +57,7 @@ def provide_feedback(model_choice, answer):
             max_tokens=150
         )
         return response.choices[0].text.strip()
-    elif model_choice in ["Gemini", "Groq"]:
+    elif model_choice == "Gemini" or model_choice == "Groq":
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
         return response.content
@@ -75,7 +73,7 @@ def get_tips(model_choice, role):
             max_tokens=150
         )
         return response.choices[0].text.strip()
-    elif model_choice in ["Gemini", "Groq"]:
+    elif model_choice == "Gemini" or model_choice == "Groq":
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
         return response.content
@@ -197,10 +195,6 @@ st.markdown(
     body {
         background-color: #e0f7fa; /* Light cyan background color */
         font-family: Arial, sans-serif;
-        background-image: url('https://example.com/your-watermark-image.png'); /* URL to your watermark image */
-        background-repeat: no-repeat;
-        background-position: center;
-        background-size: cover;
     }
     .stButton>button {
         width: 100%;
@@ -209,20 +203,14 @@ st.markdown(
         color: white;
         background-color: #4CAF50;
         border: none;
-        border-radius: 5px;
+        border-radius: 8px;
         cursor: pointer;
+        transition: background-color 0.3s ease;
     }
     .stButton>button:hover {
         background-color: #45a049;
     }
     .output-container {
-        border: 2px solid #2196F3;
-        border-radius: 8px;
-        padding: 15px;
-        margin: 15px 0;
-        background-color: #f1f1f1;
-    }
-    .progress-container {
         border: 2px solid #2196F3;
         border-radius: 8px;
         padding: 15px;
@@ -266,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")
@@ -282,11 +276,7 @@ if nav_option == "Generate Questions":
     st.header("📝 Generate Interview Questions")
     
     model_choice = st.selectbox("Choose Model:", ["OpenAI", "Gemini", "Groq"])
-    role = st.selectbox("Role", [
-        "Software Engineer", "Data Scientist", "ML Engineer", "GenAI Specialist",
-        "Product Manager", "UX Designer", "DevOps Engineer", "Cloud Architect",
-        "Business Analyst", "Database Administrator", "System Administrator", "Network Engineer"
-    ])
+    role = st.selectbox("Role:", ["GenAI", "ML", "DevOps", "Software Engineer", "Data Scientist", "Product Manager", "Designer", "Business Analyst"])
     question_type = st.selectbox("Question Type:", ["Behavioral", "Technical", "Situational", "Case Study", "Problem Solving"])
     num_questions = st.number_input("Number of Questions:", min_value=1, max_value=20, value=5)
     difficulty = st.selectbox("Difficulty Level:", ["Basic", "Medium", "Complex"])
@@ -295,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
@@ -303,28 +296,22 @@ if nav_option == "Generate Questions":
         question_list = st.session_state.questions
         index = st.session_state.question_index
         
-        # Debugging information
-        st.write(f"**Current Index:** {index}")
-        st.write(f"**Total Questions:** {len(question_list)}")
-        
         if index < len(question_list):
             st.write(f"**Question {index + 1}:** {question_list[index]}")
             
             # Answer input box
             answer = st.text_area("Your Answer", key="text_area_answer")
-            
+
             col1, col2 = st.columns(2)
             with col1:
                 if index > 0:
                     if st.button("Previous"):
                         st.session_state.question_index -= 1
-                        st.experimental_rerun()  # Re-run to update display
             
             with col2:
                 if index < len(question_list) - 1:
                     if st.button("Next"):
                         st.session_state.question_index += 1
-                        st.experimental_rerun()  # Re-run to update display
 
             # Submit answer and provide feedback
             if st.button("Submit Answer"):
@@ -333,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()
@@ -353,3 +365,4 @@ st.markdown("""
         <p>&copy; 2024 TechPrep. All rights reserved.</p>
     </div>
     """, unsafe_allow_html=True)
+