import google.generativeai as genai
import os
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Google Gemini
genai.configure(api_key=os.environ["GOOGLE_API_KEY"])

# Initialize the generative model
model = genai.GenerativeModel("gemini-pro")

def modelFeedback(ats_score, resume_data, missing_skills):
    # Create the input prompt template
    input_prompt_template = """
    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, 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