Update app.py

app.py

@@ -38,7 +38,8 @@ genai.configure(api_key=GEMINI_API_KEY)
 
 # Set up generative model
 model = genai.GenerativeModel('gemini-pro')
-
+llm = ChatGroq(groq_api_key=api_key, model_name="gemma2-9b-it")
+llm_2 = ChatGroq(groq_api_key=api_key, model_name="gemma-7b-it")
 #st.sidebar.title("Select AI Model")
 
 # model_name = st.sidebar.selectbox("Select AI Model", [
@@ -61,6 +62,7 @@ model = genai.GenerativeModel('gemini-pro')
 
 st.title("🕵️ Recruitment AI 🤖")
 
+
 def create_job_post(job_title, location, exp,salary_range,other_input):
     prompt = (
         f"Create a job opening post for platforms like Internshala, LinkedIn, and Naukri.com. "
@@ -71,18 +73,15 @@ def create_job_post(job_title, location, exp,salary_range,other_input):
 
     try:
         # Replace 'model.generate_content' with the actual method to generate content
-        response = model.generate_content(prompt)  # assuming 'model' is defined elsewhere
-        return response.text
+        response = llm_2.invoke(prompt)  # assuming 'model' is defined elsewhere
+        return response.content
     except Exception as e:
         return f"Error generating response: {e}"
 
 
-# Streamlit app title
-
 
 with st.form("job_post_form"):
     st.write("📝 Fill in the details to create a job post ✍🏼 :")
-    st.write("  ───────────────────────────────────────────────────────────")
 
     left, right = st.columns(2)
     with left:
@@ -95,7 +94,6 @@ with st.form("job_post_form"):
 
     other_input = st.text_input("Enter any other details you want to add in the job post:")
 
-    st.write("  ───────────────────────────────────────────────────────────")
 
     # Form submission button
     submitted = st.form_submit_button("Create Job Post ✅")
@@ -115,8 +113,7 @@ job_post = create_job_post(job_title, location, exp, salary_range,other_input)
 
 # Resume scoring section
 
-llm = ChatGroq(groq_api_key=api_key, model_name="gemma2-9b-it")
-llm_2 = ChatGroq(groq_api_key=api_key, model_name="gemma-7b-it")
+
 
 
 
@@ -289,12 +286,10 @@ def process_pdfs_in_batches(files):
     )
 
 
-st.write("  ───────────────────────────────────────────────────────────")
 st.write("📂 Upload the resumes to score them 📑:")
 
 uploaded_files = st.file_uploader("Choose PDF files", type="pdf", accept_multiple_files=True)
 
-st.write("  ───────────────────────────────────────────────────────────")
 if st.button("Score Resumes✔️"):
     if uploaded_files:
         process_pdfs_in_batches(uploaded_files)