rashid01 commited on
Commit
bab6757
·
verified ·
1 Parent(s): 30df955

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -6
app.py CHANGED
@@ -21,6 +21,20 @@ progress_data = {
21
  "tips_retrieved": 0
22
  }
23
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  def get_llm(model_choice):
25
  if model_choice == "Gemini":
26
  return ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
@@ -211,11 +225,11 @@ st.markdown(
211
  background-color: #45a049;
212
  }
213
  .output-container {
214
- border: 2px solid #2196F3;
215
- border-radius: 8px;
216
- padding: 15px;
217
- margin: 15px 0;
218
  background-color: #f9f9f9;
 
 
219
  }
220
  .sidebar {
221
  background-color: #ffffff; /* Sidebar background color */
@@ -276,7 +290,7 @@ if nav_option == "Generate Questions":
276
  st.header("📝 Generate Interview Questions")
277
 
278
  model_choice = st.selectbox("Choose Model:", ["OpenAI", "Gemini"])
279
- role = st.text_input("Role", placeholder="e.g. Software Engineer")
280
  question_type = st.selectbox("Question Type:", ["Behavioral", "Technical", "Situational", "Case Study", "Problem Solving"])
281
  num_questions = st.number_input("Number of Questions:", min_value=1, max_value=20, value=5)
282
  difficulty = st.selectbox("Difficulty Level:", ["Easy", "Medium", "Hard"])
@@ -365,4 +379,3 @@ st.markdown("""
365
  <p>&copy; 2024 TechPrep. All rights reserved.</p>
366
  </div>
367
  """, unsafe_allow_html=True)
368
-
 
21
  "tips_retrieved": 0
22
  }
23
 
24
+ # Define list of roles for dropdown
25
+ ROLE_OPTIONS = [
26
+ "Software Engineer",
27
+ "Data Scientist",
28
+ "Product Manager",
29
+ "UX Designer",
30
+ "Business Analyst",
31
+ "Project Manager",
32
+ "Marketing Specialist",
33
+ "Sales Manager",
34
+ "Customer Support Specialist",
35
+ "Other"
36
+ ]
37
+
38
  def get_llm(model_choice):
39
  if model_choice == "Gemini":
40
  return ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=GOOGLE_API_KEY)
 
225
  background-color: #45a049;
226
  }
227
  .output-container {
228
+ padding: 1em;
229
+ margin-bottom: 1em;
 
 
230
  background-color: #f9f9f9;
231
+ border-radius: 8px;
232
+ border: 1px solid #d0d0d0;
233
  }
234
  .sidebar {
235
  background-color: #ffffff; /* Sidebar background color */
 
290
  st.header("📝 Generate Interview Questions")
291
 
292
  model_choice = st.selectbox("Choose Model:", ["OpenAI", "Gemini"])
293
+ role = st.selectbox("Role", ROLE_OPTIONS)
294
  question_type = st.selectbox("Question Type:", ["Behavioral", "Technical", "Situational", "Case Study", "Problem Solving"])
295
  num_questions = st.number_input("Number of Questions:", min_value=1, max_value=20, value=5)
296
  difficulty = st.selectbox("Difficulty Level:", ["Easy", "Medium", "Hard"])
 
379
  <p>&copy; 2024 TechPrep. All rights reserved.</p>
380
  </div>
381
  """, unsafe_allow_html=True)