Spaces:
Build error
Build error
File size: 4,161 Bytes
d04d842 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 |
import os
from langchain_groq import ChatGroq
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import JsonOutputParser
from langchain_core.exceptions import OutputParserException
from dotenv import load_dotenv
load_dotenv()
class Chain:
def __init__(self):
self.llm = ChatGroq(temperature=0, groq_api_key=os.getenv("GROQ_API_KEY"), model_name="llama-3.3-70b-versatile")
def extract_jobs(self, cleaned_text):
prompt_extract = PromptTemplate.from_template(
"""
### SCRAPED TEXT FROM WEBSITE:
{page_data}
### INSTRUCTION:
The scraped text is from the career's page of a website.
Your job is to extract the job postings and return them in JSON format containing the following keys: `role`, `experience`, `skills`, and `description`.
Only return the valid JSON.
### VALID JSON (NO PREAMBLE):
"""
)
chain_extract = prompt_extract | self.llm
res = chain_extract.invoke(input={"page_data": cleaned_text})
try:
json_parser = JsonOutputParser()
res = json_parser.parse(res.content)
except OutputParserException:
raise OutputParserException("Context too big. Unable to parse jobs.")
return res if isinstance(res, list) else [res]
def write_mail(self, job, links, username="Thamani", client_name="Hiring Manager", email_style="Formal"):
"""Generates cold emails based on the selected style."""
style_instructions = {
"Formal": "Maintain a professional and polished tone. Focus on achievements and qualifications.",
"Casual": "Use a friendly, engaging tone. Keep it light while still showcasing strengths.",
"Persuasive": "Be compelling and assertive. Highlight why you are the perfect fit with strong language."
}
# Convert style name to lowercase for prompt clarity
email_style_lower = email_style.lower()
prompt_email = PromptTemplate.from_template(
"""
### JOB DETAILS:
- Role: {job_title}
- Company: {company_name}
- Experience Required: {experience}
- Skills: {skills}
- Description: {job_description}
### INSTRUCTION:
You are {username}, a motivated MCA graduate with strong technical and analytical skills, seeking an opportunity to contribute to {company_name} as a {job_title}.
Craft a **{email_style} cold email** to {client_name} demonstrating your skills, projects, and value.
Style Instruction: {style_instruction}
Highlight relevant projects, certifications, or portfolio links: {link_list}.
### EMAIL (NO PREAMBLE):
"""
)
chain_email = prompt_email | self.llm
# Extract job details, handling missing fields
job_title = job.get("role", "the position")
company_name = job.get("company", "the company")
experience = job.get("experience", "not specified")
skills = ", ".join(job.get("skills", [])) or "not mentioned"
job_description = job.get("description", "No description provided.")
# Filter out empty links
valid_links = [link for link in links if link]
formatted_links = "\n".join(f"- {link}" for link in valid_links) if valid_links else "No portfolio links provided."
# Generate email
res = chain_email.invoke({
"job_title": job_title,
"company_name": company_name,
"experience": experience,
"skills": skills,
"job_description": job_description,
"link_list": formatted_links,
"username": username,
"client_name": client_name,
"email_style": email_style_lower, # Fixed Issue
"style_instruction": style_instructions[email_style]
})
return res.content
if __name__ == "__main__":
print(os.getenv("GROQ_API_KEY")) |