Update app.py

app.py

@@ -69,7 +69,7 @@ class AutonomousEmailAgent:
         - Candidate's Skills: {', '.join(self.skills)}
         - Candidate's Experiences: {', '.join([exp['title'] for exp in self.experiences])}
 
-        Generate the email using this structure and make it compelling and tailored to the company and role.
+        Generate the email using this structure and ensure it is within {self.word_limit} words.
         """
         
         return self.send_request_to_llm(reasoning_prompt)
@@ -112,19 +112,30 @@ class AutonomousEmailAgent:
             return "Error: Unable to generate response."
 
     def format_email(self, llm_response):
-        # Remove unnecessary tabs and spaces and format consistently
-        lines = [line.strip() for line in llm_response.split("\n")]
-        formatted_email = "\n\n".join([line for line in lines if line])
-        
-        signature = f"""
-        Best regards,
-        {self.user_name}
-        Email: {self.email}
-        Phone: {self.phone}
-        LinkedIn: {self.linkedin}
-        """
-        
-        return f"{formatted_email}\n\n{signature.strip()}"
+        # Clean and format the email
+        lines = [line.strip() for line in llm_response.split("\n") if line.strip()]
+        formatted_email = "\n\n".join(lines)
+
+        # Truncate the email if it exceeds the word limit
+        word_count = len(formatted_email.split())
+        if word_count > self.word_limit:
+            truncated_email = " ".join(formatted_email.split()[:self.word_limit])
+            formatted_email = truncated_email + "..."
+
+        # Prepare the signature
+        signature = (
+            f"Best regards,\n"
+            f"{self.user_name}\n"
+            f"Email: {self.email}\n"
+            f"Phone: {self.phone}\n"
+            f"LinkedIn: {self.linkedin}"
+        )
+
+        # Ensure only one "Best regards" section
+        if "Best regards" in formatted_email:
+            formatted_email = formatted_email.split("Best regards")[0].strip()
+
+        return f"{formatted_email}\n\n{signature}"
 
 # Gradio UI setup remains unchanged
 def gradio_ui():