Update model.py

model.py

@@ -11,20 +11,18 @@ 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, missing_skills):
     # Create the input prompt template
     input_prompt_template = """
-    You are an HR Screening for Resumes.
-    The ATS score for the resume is {ats_score}.
-    Provide a relevant summary (up to 150 words) for the following resume based on this score:
-    {resume_data}
+    You are an HR assistant reviewing resumes. The ATS score for the resume is {ats_score}.
+    Provide a professional summary for the resume and suggest improvements:
+    - Include the strengths of the resume.
+    - Highlight the following missing skills to improve ATS performance: {missing_skills}.
+    Resume Content: {resume_data}
     """
-    # Format the prompt with ATS score and resume data
-    input_prompt = input_prompt_template.format(ats_score=ats_score, resume_data=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, missing_skills=", ".join(missing_skills))
     
     # Generate response
     response = model.generate_content(input_prompt)
-    
-    # Return response text
     return response.text
-