Create chains.py

chains.py

@@ -0,0 +1,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"))