honeyangelhp's picture
Update model.py
8f5c874 verified
raw
history blame
635 Bytes
import google.generativeai as genai
import os
from dotenv import load_dotenv
from convert import ExtractPDFText
load_dotenv()
genai.configure(api_key = os.environ["GOOGLE_API_KEY"])
model = genai.GenerativeModel("gemini-pro")
def modelFeedback(ats_score,resume_data):
input_prompt = f"""
You are now an ATS Score analyzer and given ATS Score is {int(ats_score*100)}%.
Your task is to generate summaries and matching scores for each selected resume
print ATS score first.
"""
response = model.generate_content([input_prompt,resume_data],stream=True)
response.resolve()
return response