Update chains.py

chains.py

@@ -18,7 +18,7 @@ class Chain:
             {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`.
+            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):
             """
@@ -32,7 +32,15 @@ class Chain:
             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"):
+    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."
+        }
+        
         prompt_email = PromptTemplate.from_template(
             """
             ### JOB DETAILS:
@@ -44,9 +52,12 @@ class Chain:
             
             ### 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 **compelling and professional** cold email to {client_name}, demonstrating how your academic background, technical expertise, and problem-solving abilities align with the role’s requirements.  
-            Effectively highlight your **skills, relevant projects, certifications, or portfolio links**: {link_list}, showcasing how you can add value to the company.  
-            Ensure the email is **concise, persuasive, and professionally structured**, making a strong impression on the hiring manager.  
+            Craft a **{email_style.lower()} 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):
             """  
         )
@@ -73,11 +84,98 @@ class Chain:
             "job_description": job_description,
             "link_list": formatted_links,
             "username": username,
-            "client_name": client_name
+            "client_name": client_name,
+            "email_style": email_style,
+            "style_instruction": style_instructions[email_style]
         })
     
         return res.content
 
 if __name__ == "__main__":
     print(os.getenv("GROQ_API_KEY"))
+
+
+
+# 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"):
+#         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 **compelling and professional** cold email to {client_name}, demonstrating how your academic background, technical expertise, and problem-solving abilities align with the role’s requirements.  
+#             Effectively highlight your **skills, relevant projects, certifications, or portfolio links**: {link_list}, showcasing how you can add value to the company.  
+#             Ensure the email is **concise, persuasive, and professionally structured**, making a strong impression on the hiring manager.  
+#             ### 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
+#         })
+    
+#         return res.content
+
+# if __name__ == "__main__":
+#     print(os.getenv("GROQ_API_KEY"))