Update app.py

app.py

@@ -1,11 +1,16 @@
 import streamlit as st
+from langchain_groq import ChatGroq
+from langchain_core.prompts import PromptTemplate
 import fitz  # PyMuPDF
 import requests
 from bs4 import BeautifulSoup
-import openai
 
-# Set your OpenAI API key
-openai.api_key = 'YOUR_OPENAI_API_KEY'  # Replace with your OpenAI API key
+# Initialize the LLM with your Groq API key
+llm = ChatGroq(
+    temperature=0,
+    groq_api_key='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',  # Replace with your Groq API key
+    model_name="llama-3.1-70b-versatile"
+)
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(pdf_file):
@@ -21,45 +26,59 @@ def extract_job_description(job_link):
         response = requests.get(job_link)
         response.raise_for_status()
         soup = BeautifulSoup(response.text, 'html.parser')
-        # You might need to adjust this depending on the website's structure
-        job_description = soup.get_text(separator=' ')
+        job_description = soup.get_text(separator='\n')
         return job_description.strip()
     except Exception as e:
         st.error(f"Error fetching job description: {e}")
         return ""
 
-# Function to generate an email using OpenAI's GPT
-def generate_email(job_description, resume_text):
-    prompt = f"""
+# Function to extract requirements from the job description using ChatGroq
+def extract_requirements(job_description):
+    prompt_text = f"""
+    The following is a job description:
+
+    {job_description}
+
+    Extract the list of job requirements, qualifications, and skills from the job description. Provide them as a numbered list.
+
+    Requirements:
+    """
+
+    prompt = PromptTemplate.from_template(prompt_text)
+    chain = prompt | llm
+    response = chain.invoke({})
+
+    requirements = response.content.strip()
+    return requirements
+
+# Function to generate an email using ChatGroq
+def generate_email(job_description, requirements, resume_text):
+    prompt_text = f"""
     You are Mohan, a business development executive at AtliQ, an AI & Software Consulting company dedicated to facilitating the seamless integration of business processes through automated tools. AtliQ has empowered numerous enterprises with tailored solutions, fostering scalability, process optimization, cost reduction, and heightened overall efficiency.
 
     Given the following job description:
 
     {job_description}
 
+    And the following extracted requirements:
+
+    {requirements}
+
     And the following resume text:
 
     {resume_text}
 
-    Write a cold email to the client regarding the job mentioned above, describing the capability of AtliQ in fulfilling their needs. Remember you are Mohan, BDE at AtliQ. Do not provide a preamble.
+    Write a cold email to the client regarding the job mentioned above, describing the capability of AtliQ in fulfilling their needs, and how your skills align with their requirements. Remember you are Mohan, BDE at AtliQ. Do not provide a preamble.
 
     Email:
     """
-    try:
-        response = openai.Completion.create(
-            engine="text-davinci-003",  # You can use "gpt-3.5-turbo" if available
-            prompt=prompt,
-            max_tokens=500,
-            temperature=0.7,
-            top_p=1.0,
-            n=1,
-            stop=None
-        )
-        email_text = response.choices[0].text.strip()
-        return email_text
-    except Exception as e:
-        st.error(f"Error generating email: {e}")
-        return ""
+
+    prompt = PromptTemplate.from_template(prompt_text)
+    chain = prompt | llm
+    response = chain.invoke({})
+
+    email_text = response.content.strip()
+    return email_text
 
 # Streamlit App
 def main():
@@ -88,6 +107,12 @@ def main():
                 st.error("Failed to extract job description.")
                 return
 
+            # Extract requirements
+            requirements = extract_requirements(job_description)
+            if not requirements:
+                st.error("Failed to extract requirements.")
+                return
+
             # Extract resume text
             resume_text = extract_text_from_pdf(uploaded_file)
             if not resume_text:
@@ -95,7 +120,7 @@ def main():
                 return
 
             # Generate email
-            email_text = generate_email(job_description, resume_text)
+            email_text = generate_email(job_description, requirements, resume_text)
             if email_text:
                 st.subheader("Generated Email:")
                 st.write(email_text)