|
|
import os
|
|
|
from dotenv import load_dotenv
|
|
|
import json
|
|
|
import google.genai as genai
|
|
|
import time
|
|
|
import logging
|
|
|
|
|
|
|
|
|
load_dotenv()
|
|
|
class Model:
|
|
|
def __init__(self,PROMPT,model_inp):
|
|
|
API_KEY = os.getenv("GOOGLE_API_KEY")
|
|
|
self.client = genai.Client(api_key=API_KEY)
|
|
|
self.chat = self.client.chats.create(model = model_inp)
|
|
|
self.model_name = model_inp
|
|
|
response = self.chat.send_message(PROMPT)
|
|
|
|
|
|
def json(self,text):
|
|
|
if not text.strip():
|
|
|
return ""
|
|
|
|
|
|
s_idx = text.find("{")
|
|
|
e_idx = text.rfind("}") + 1
|
|
|
|
|
|
if s_idx == -1 or e_idx == -1 :
|
|
|
return ""
|
|
|
|
|
|
try:
|
|
|
output = json.loads(text[s_idx:e_idx])
|
|
|
except Exception as e:
|
|
|
raise ValueError(f"Error Parsing {e}")
|
|
|
|
|
|
return output if output else ""
|
|
|
|
|
|
def send(self, content, max_retries=2):
|
|
|
for attempt in range(max_retries + 1):
|
|
|
try:
|
|
|
start_time = time.time()
|
|
|
logging.info(f"Sending request to {self.model_name} (attempt {attempt + 1})")
|
|
|
|
|
|
response = self.chat.send_message(content)
|
|
|
|
|
|
elapsed_time = time.time() - start_time
|
|
|
logging.info(f"API call completed in {elapsed_time:.2f} seconds")
|
|
|
|
|
|
result = self.json(response.text)
|
|
|
return result
|
|
|
|
|
|
except Exception as e:
|
|
|
logging.error(f"API call failed (attempt {attempt + 1}): {str(e)}")
|
|
|
if attempt == max_retries:
|
|
|
|
|
|
logging.warning("All retries failed, returning basic structure")
|
|
|
return {
|
|
|
"name": "Unknown",
|
|
|
"email": "unknown@example.com",
|
|
|
"phone": None,
|
|
|
"linkedin": None,
|
|
|
"location": None,
|
|
|
"summary": "Resume parsing failed",
|
|
|
"education": [],
|
|
|
"experience": [],
|
|
|
"projects": [],
|
|
|
"skills": {
|
|
|
"languages": [],
|
|
|
"frameworks": [],
|
|
|
"libraries": [],
|
|
|
"tools": [],
|
|
|
"platforms": [],
|
|
|
"design": []
|
|
|
},
|
|
|
"certifications": [],
|
|
|
"awards": []
|
|
|
}
|
|
|
time.sleep(2)
|
|
|
|
|
|
|
|
|
|
|
|
|
