Update app.py

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 import openai
 from langchain_google_genai import ChatGoogleGenerativeAI
-import requests  # For Groq model
+from langchain_groq import ChatGroqGenerativeAI  # Assume this is the correct import for Groq
 from datetime import datetime, timedelta
 import time
 
@@ -10,7 +10,7 @@ GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
 OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
 GROQ_API_KEY = st.secrets["GROQ_API_KEY"]
 
-# Initialize OpenAI
+# Initialize APIs
 openai.api_key = OPENAI_API_KEY
 
 # In-memory storage for progress tracking
@@ -44,7 +44,7 @@ def get_llm(model_choice):
     elif model_choice == "OpenAI":
         return None
     elif model_choice == "Groq":
-        return GROQ_API_KEY  # Example placeholder
+        return ChatGroqGenerativeAI(api_key=GROQ_API_KEY)
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -65,9 +65,9 @@ def generate_questions(model_choice, role, question_type, num_questions, difficu
         response = llm.invoke(prompt)
         return response.content.split("\n")
     elif model_choice == "Groq":
-        headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
-        response = requests.post("https://api.groq.com/generate", headers=headers, json={"prompt": prompt})
-        return response.json().get("text", "").split("\n")
+        llm = get_llm(model_choice)
+        response = llm.invoke(prompt)
+        return response.content.split("\n")
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -85,9 +85,9 @@ def provide_feedback(model_choice, answer):
         response = llm.invoke(prompt)
         return response.content
     elif model_choice == "Groq":
-        headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
-        response = requests.post("https://api.groq.com/feedback", headers=headers, json={"prompt": prompt})
-        return response.json().get("text", "")
+        llm = get_llm(model_choice)
+        response = llm.invoke(prompt)
+        return response.content
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -105,9 +105,9 @@ def get_tips(model_choice, role):
         response = llm.invoke(prompt)
         return response.content
     elif model_choice == "Groq":
-        headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
-        response = requests.post("https://api.groq.com/tips", headers=headers, json={"prompt": prompt})
-        return response.json().get("text", "")
+        llm = get_llm(model_choice)
+        response = llm.invoke(prompt)
+        return response.content
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -185,7 +185,6 @@ def track_progress():
         <p><strong>Mock Interviews Taken:</strong> {progress_data['mock_interviews_taken']}</p>
         <p><strong>Feedback Provided:</strong> {progress_data['feedback_provided']}</p>
         <p><strong>Tips Retrieved:</strong> {progress_data['tips_retrieved']}</p>
-        <p><strong>Resources Used:</strong> {progress_data['resources_used']}</p>
     </div>
     """, unsafe_allow_html=True)
 
@@ -204,41 +203,41 @@ def connect_resources():
 
     # Form to connect with career coaches or mentors
     with st.form("contact_form"):
-        st.write("For personalized assistance, please fill out the form below:")
-        name = st.text_input("Your Name")
-        email = st.text_input("Your Email")
+        st.write("For personalized assistance, please fill out this form:")
+        name = st.text_input("Name")
+        email = st.text_input("Email")
         message = st.text_area("Message")
-        submit_button = st.form_submit_button("Send Message")
+        submit_button = st.form_submit_button("Submit")
 
         if submit_button:
-            if name and email and message:
-                st.success("Thank you for reaching out! We'll get back to you soon.")
-                progress_data["resources_used"] += 1
+            if not name or not email or not message:
+                st.error("Please fill out all fields.")
             else:
-                st.error("Please fill out all fields before submitting.")
+                st.success("Thank you for contacting us! We will get back to you soon.")
+                progress_data["resources_used"] += 1
 
-# Style for feedback and results
-def style_output(title, color):
-    return f"""
-    <div class="output-container" style="border-left: 5px solid {color};">
-        <h3>{title}</h3>
-    </div>
-    """
+def style_output(text, color):
+    return f'<div class="output-container"><span style="color: {color}; font-weight: bold;">{text}</span></div>'
 
-# Custom CSS
-st.markdown("""
+# Streamlit app layout
+st.set_page_config(page_title="TechPrep", layout="wide")
+st.markdown(
+    """
     <style>
+    body {
+        background-color: #e0f7fa; /* Light cyan background color */
+        font-family: Arial, sans-serif;
+    }
     .stButton>button {
-        background-color: #4CAF50;
+        width: 100%;
+        height: 3em;
+        font-size: 1.2em;
         color: white;
+        background-color: #4CAF50;
         border: none;
-        padding: 10px 20px;
-        text-align: center;
-        text-decoration: none;
-        display: inline-block;
-        font-size: 16px;
         border-radius: 8px;
         cursor: pointer;
+        transition: background-color 0.3s ease;
     }
     .stButton>button:hover {
         background-color: #45a049;
@@ -251,7 +250,7 @@ st.markdown("""
         border: 1px solid #d0d0d0;
     }
     .sidebar {
-        background-color: #ffffff;
+        background-color: #ffffff; /* Sidebar background color */
         padding: 1em;
     }
     .footer {
@@ -264,11 +263,10 @@ st.markdown("""
         width: 100%;
         border-top: 2px solid #ffffff;
     }
-    .progress-container p {
-        margin: 5px 0;
-    }
     </style>
-    """, unsafe_allow_html=True)
+    """,
+    unsafe_allow_html=True
+)
 
 # Show welcome message with an icon for 3-4 seconds
 welcome_message = st.empty()
@@ -285,10 +283,10 @@ with welcome_message.container():
             <i class="fa fa-smile-o" aria-hidden="true" style="font-size: 40px;"></i> Welcome to TechPrep!
         </div>
     """, unsafe_allow_html=True)
-time.sleep(4)
-welcome_message.empty()
+time.sleep(4)  # Wait for 4 seconds
+welcome_message.empty()  # Remove the welcome message
 
-# Initialize session state
+# 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:
@@ -302,7 +300,7 @@ if 'show_results' not in st.session_state:
 
 # Sidebar Navigation
 st.sidebar.title("TechPrep Navigation")
-nav_option = st.sidebar.radio("Choose an option:",
+nav_option = st.sidebar.radio("Choose an option:", 
     ["Generate Questions", "Mock Interview", "Track Progress", "Connect with Resources"])
 
 # Handling page navigation
@@ -313,7 +311,7 @@ if nav_option == "Generate Questions":
     role = st.selectbox("Role", ROLE_OPTIONS)
     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:", ["Easy", "Medium", "Hard"])
+    difficulty = st.selectbox("Difficulty Level:", ["Basic", "Medium", "Complex"])
     
     if st.button("Generate Questions", key="generate_questions"):
         with st.spinner("Generating questions..."):