Update app.py

app.py

@@ -112,15 +112,15 @@ class AutonomousEmailAgent:
             return "Error: Unable to generate response."
 
     def format_email(self, llm_response):
-        # Clean and format the email into paragraphs
+        # Split the response into paragraphs for better formatting
         paragraphs = [line.strip() for line in llm_response.split("\n") if line.strip()]
         formatted_email = "\n\n".join(paragraphs)
 
-        # Truncate the email if it exceeds the word limit
+        # Enforce word limit without breaking the structure
         words = formatted_email.split()
         if len(words) > self.word_limit:
-            truncated_email = " ".join(words[:self.word_limit]) + "..."
-            formatted_email = truncated_email
+            truncated_email = " ".join(words[:self.word_limit])
+            formatted_email = truncated_email + "..."
 
         # Add the closing section with a call to action
         closing_section = (
@@ -138,11 +138,11 @@ class AutonomousEmailAgent:
             f"LinkedIn: {self.linkedin}"
         )
 
-        # Ensure only one "Best regards" section and remove any duplicate signatures
+        # Ensure the signature and closing are not duplicated and formatted neatly
         if "Best regards" in formatted_email:
             formatted_email = formatted_email.split("Best regards")[0].strip()
 
-        return f"{formatted_email}\n{closing_section}\n{signature}"
+        return f"{formatted_email}{closing_section}\n{signature}"
 
 # Gradio UI setup remains unchanged
 def gradio_ui():