| 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)
|
|
|
|
|
|
|
|
|
