Update model.py

model.py

@@ -11,15 +11,15 @@ genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
 # Initialize the generative model
 model = genai.GenerativeModel("gemini-pro")
 
-def modelFeedback(ats_score, resume_data):
+def modelFeedback(ats_score, resume_data, job_description):
     # Create the input prompt template
     input_prompt_template = """
     You are an HR assistant reviewing resumes. The ATS score for the resume is {ats_score}.
-    For each resume, provide a summary(100 words), include the strengths of the resume and if the resume is fit for the job based on the job description.
+    For each resume, provide a summary(100 words), include the strengths of the resume and if the resume is fit for the job based on the {job_description}.
     Resume Content: {resume_data}
     """
     # Format the prompt with ATS score, missing skills, and resume data
-    input_prompt = input_prompt_template.format(ats_score=ats_score, resume_data=resume_data)
+    input_prompt = input_prompt_template.format(ats_score=ats_score, resume_data=resume_data, job_description=job_description)
     
     # Generate response
     response = model.generate_content(input_prompt)