Update app.py

app.py

@@ -1,12 +1,14 @@
 import streamlit as st
 import openai
 from langchain_google_genai import ChatGoogleGenerativeAI
+import requests  # For Groq model
 from datetime import datetime, timedelta
 import time
 
 # API keys
 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
 openai.api_key = OPENAI_API_KEY
@@ -18,7 +20,8 @@ progress_data = {
     },
     "mock_interviews_taken": 0,
     "feedback_provided": 0,
-    "tips_retrieved": 0
+    "tips_retrieved": 0,
+    "resources_used": 0
 }
 
 # Define list of roles for dropdown
@@ -40,6 +43,8 @@ def get_llm(model_choice):
         return ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
     elif model_choice == "OpenAI":
         return None
+    elif model_choice == "Groq":
+        return GROQ_API_KEY  # Example placeholder
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -59,6 +64,10 @@ def generate_questions(model_choice, role, question_type, num_questions, difficu
         llm = get_llm(model_choice)
         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")
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -75,6 +84,10 @@ def provide_feedback(model_choice, answer):
         llm = get_llm(model_choice)
         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", "")
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -91,6 +104,10 @@ def get_tips(model_choice, role):
         llm = get_llm(model_choice)
         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", "")
     else:
         raise ValueError("Unsupported model choice.")
 
@@ -168,6 +185,7 @@ 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)
 
@@ -186,40 +204,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 this form:")
-        name = st.text_input("Name")
-        email = st.text_input("Email")
+        st.write("For personalized assistance, please fill out the form below:")
+        name = st.text_input("Your Name")
+        email = st.text_input("Your Email")
         message = st.text_area("Message")
-        submit_button = st.form_submit_button("Submit")
+        submit_button = st.form_submit_button("Send Message")
 
         if submit_button:
-            if not name or not email or not message:
-                st.error("Please fill out all fields.")
+            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
             else:
-                st.success("Thank you for contacting us! We will get back to you soon.")
+                st.error("Please fill out all fields before submitting.")
 
-def style_output(text, color):
-    return f'<div class="output-container"><span style="color: {color}; font-weight: bold;">{text}</span></div>'
-
-# Streamlit app layout
-st.set_page_config(page_title="TechPrep", layout="wide")
-st.markdown(
+# 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>
     """
+
+# Custom CSS
+st.markdown("""
     <style>
-    body {
-        background-color: #e0f7fa; /* Light cyan background color */
-        font-family: Arial, sans-serif;
-    }
     .stButton>button {
-        width: 100%;
-        height: 3em;
-        font-size: 1.2em;
-        color: white;
         background-color: #4CAF50;
+        color: white;
         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;
@@ -232,7 +251,7 @@ st.markdown(
         border: 1px solid #d0d0d0;
     }
     .sidebar {
-        background-color: #ffffff; /* Sidebar background color */
+        background-color: #ffffff;
         padding: 1em;
     }
     .footer {
@@ -245,10 +264,11 @@ 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()
@@ -265,10 +285,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)  # Wait for 4 seconds
-welcome_message.empty()  # Remove the welcome message
+time.sleep(4)
+welcome_message.empty()
 
-# Initialize session state for questions, answers, and current index
+# Initialize session state
 if 'questions' not in st.session_state:
     st.session_state.questions = []
 if 'answers' not in st.session_state:
@@ -282,14 +302,14 @@ 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
 if nav_option == "Generate Questions":
     st.header("📝 Generate Interview Questions")
     
-    model_choice = st.selectbox("Choose Model:", ["OpenAI", "Gemini"])
+    model_choice = st.selectbox("Choose Model:", ["OpenAI", "Gemini", "Groq"])
     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)