Update app.py

app.py

@@ -7,11 +7,15 @@ proxycurl_api_key = os.getenv("PROXYCURL_API_KEY")  # Proxycurl API key
 groq_api_key = os.getenv("GROQ_CLOUD_API_KEY")  # Groq Cloud API key
 
 class EmailAgent:
-    def __init__(self, linkedin_url, company_name, role, word_limit):
+    def __init__(self, linkedin_url, company_name, role, word_limit, user_name, email, phone, linkedin):
         self.linkedin_url = linkedin_url
         self.company_name = company_name
         self.role = role
         self.word_limit = word_limit
+        self.user_name = user_name
+        self.email = email
+        self.phone = phone
+        self.linkedin = linkedin
         self.bio = None
         self.skills = []
         self.experiences = []
@@ -21,10 +25,6 @@ class EmailAgent:
     # Reason: Decide what information is needed
     def reason_about_data(self):
         print("Reasoning: I need LinkedIn data, company info, and role description.")
-        if not self.linkedin_url:
-            print("Warning: LinkedIn URL missing. Will proceed with default bio.")
-        if not self.company_name:
-            print("Warning: Company name missing. Will proceed with default company info.")
     
     # Action: Fetch LinkedIn data via Proxycurl
     def fetch_linkedin_data(self):
@@ -73,14 +73,12 @@ class EmailAgent:
     # Action: Fetch role description
     def fetch_role_description(self):
         print(f"Action: Fetching role description for {self.role}.")
-        self.role_description = f"The role of {self.role} at {self.company_name} involves..."
+        self.role_description = f"The role of {self.role} at {self.company_name} involves mentoring AI and technology students to develop their skills and progress their careers."
 
     # Reflection: Check if the data is sufficient to generate an email
     def reflect_on_data(self):
         print("Reflection: Do I have enough data to generate the email?")
         # Allow the email to be generated with default values if data is missing
-        if not self.bio or not self.skills or not self.company_info:
-            print("Warning: Some critical information is missing. Proceeding with default values.")
         return True
     
     # Action: Generate the email using Groq Cloud LLM
@@ -89,11 +87,21 @@ class EmailAgent:
         prompt = f"""
         Write a professional email applying for the {self.role} position at {self.company_name}.
         The candidate’s bio is: {self.bio}.
-        The candidate's LinkedIn profile highlights the following skills: {', '.join(self.skills)}.
-        The candidate has the following experiences relevant to the job: {', '.join([exp['title'] for exp in self.experiences])}.
-        The email should be professional, concise, and tailored to the company's culture.
-        Use relevant company details: {self.company_info}.
-        Highlight the candidate’s skills and experiences from LinkedIn, and map them to the job's requirements: {self.role_description}.
+        
+        Focus on relevant skills and experiences from LinkedIn, such as {', '.join(self.skills)}, 
+        that directly align with the role of {self.role}. 
+        Highlight only the skills and experiences that relate to mentoring, AI, technology, and leadership.
+        
+        The company info is: {self.company_info}.
+        The role involves: {self.role_description}.
+        
+        End the email with this signature:
+        Best regards,
+        {self.user_name}
+        Email: {self.email}
+        Phone: {self.phone}
+        LinkedIn: {self.linkedin}
+        
         The email should not exceed {self.word_limit} words.
         """
         
@@ -141,8 +149,8 @@ def gradio_ui():
     email_output = gr.Textbox(label="Generated Email", placeholder="Your generated email will appear here", lines=10)
 
     # Function to create and run the email agent
-    def create_email(name, company_name, role, email, phone, linkedin_profile_url, word_limit):
-        agent = EmailAgent(linkedin_profile_url, company_name, role, word_limit)
+    def create_email(name, company_name, role, email, phone, linkedin_url, word_limit):
+        agent = EmailAgent(linkedin_url, company_name, role, word_limit, name, email, phone, linkedin_url)
         return agent.run()
 
     # Gradio interface