Update app.py

app.py

@@ -24,6 +24,8 @@ progress_data = {
 def get_llm(model_choice):
     if model_choice == "Gemini":
         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:
@@ -41,7 +43,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 == "Gemini":
+    elif model_choice in ["Gemini", "Groq"]:
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
         return response.content.split("\n")
@@ -57,7 +59,7 @@ def provide_feedback(model_choice, answer):
             max_tokens=150
         )
         return response.choices[0].text.strip()
-    elif model_choice == "Gemini":
+    elif model_choice in ["Gemini", "Groq"]:
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
         return response.content
@@ -73,7 +75,7 @@ def get_tips(model_choice, role):
             max_tokens=150
         )
         return response.choices[0].text.strip()
-    elif model_choice == "Gemini":
+    elif model_choice in ["Gemini", "Groq"]:
         llm = get_llm(model_choice)
         response = llm.invoke(prompt)
         return response.content
@@ -207,9 +209,8 @@ st.markdown(
         color: white;
         background-color: #4CAF50;
         border: none;
-        border-radius: 8px;
+        border-radius: 5px;
         cursor: pointer;
-        transition: background-color 0.3s ease;
     }
     .stButton>button:hover {
         background-color: #45a049;
@@ -280,8 +281,12 @@ nav_option = st.sidebar.radio("Choose an option:",
 if nav_option == "Generate Questions":
     st.header("📝 Generate Interview Questions")
     
-    model_choice = st.selectbox("Choose Model:", ["OpenAI", "Gemini"])
-    role = st.selectbox("Role", ["Software Engineer", "Data Scientist", "ML Engineer", "GenAI Specialist"])
+    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"
+    ])
     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"])