Update email_processor.py

email_processor.py

@@ -74,38 +74,49 @@ def load_email_guidelines():
         print(f"Error loading email guidelines: {e}")
         return "No guidelines available."
 
-def create_email_format_prompt(email_guidelines):
+def create_email_format_prompt(email_guidelines, identity=None):
     """
     Create the prompt template that will be sent to the language model.
     Incorporates the provided email guidelines and placeholders for user input.
     
     Args:
         email_guidelines: The guidelines text to include in the prompt
+        identity: Optional identity perspective to write from
     """
-    return """<EMAIL_GUIDELINES>
-""" + email_guidelines + """
+    identity_section = ""
+    if identity and identity.strip():
+        identity_section = f"""
+<IDENTITY>
+The email is being written from the perspective of: {identity}
+</IDENTITY>
+"""
+
+    return f"""<EMAIL_GUIDELINES>
+{email_guidelines}
 </EMAIL_GUIDELINES>
+{identity_section}
 =========================
 
 Your Task:
-1. Review the user's draft email.
-2. Apply the above style, formatting, and general content guidelines to the email.
-3. Output a reformatted email ready for use.
+1. Review the user's draft email
+2. Apply the style, formatting, and content guidelines above to improve the email
+3. Provide a polished, ready-to-send version
 
-The following email thread precedes the user's draft email:
+Previous email thread:
 <PREVIOUS_EMAILS>
-{preceding_conversation}
+{{preceding_conversation}}
 </PREVIOUS_EMAILS>
 
-The rough draft of the user's email is as follows:
+User's draft reply:
 <USER_EMAIL_DRAFT>
-{drafted_user_reply}
+{{drafted_user_reply}}
 </USER_EMAIL_DRAFT>
 
-Your task is to rewrite the user's email draft to align with the provided guidelines.
-Begin your reply with the exact string EMAIL_STARTS_HERE followed by the revised email.
-If you have additional suggestions, include them only AFTER the revised email draft, separated by the string SUGGESTIONS_START_HERE.
-Do not assume any new information or context beyond what is provided in the user's immediate draft email. Do not add any new URLs, attachments, phone numbers, or other content beyond what is provided in the user's immediate draft email. Do not add a subject line. Begin the email body immediately after EMAIL_STARTS_HERE. Include the user's typical signature at the end of the email, if they have one. Do not use info (even placeholders) that is not provided in the user's immediate draft email or in the preceding email thread (even if the guidelines call for it). Do not use markdown formatting. Assume your redrafted reply message might be sent by the user after only a cursory review."""
+Rewrite the draft to follow the guidelines while preserving the user's intent.
+Begin with EMAIL_STARTS_HERE followed by the revised email.
+If you have suggestions, add them after SUGGESTIONS_START_HERE.
+
+Don't add any information not found in the original draft or thread. Don't create new URLs, phone numbers, attachments, or subject lines. Start directly with the email body after EMAIL_STARTS_HERE. Include the user's signature if present in their draft. Don't use markdown formatting."""
 
 def process_email(
     preceding_conversation,
@@ -118,7 +129,8 @@ def process_email(
     top_p,
     password,
     clients,
-    custom_guidelines=None
+    custom_guidelines=None,
+    identity=None
 ):
     """
     Process the email and return a formatted response and suggestions.
@@ -135,6 +147,7 @@ def process_email(
         password: The actual password to verify against
         clients: Dictionary of configured API clients
         custom_guidelines: Optional custom guidelines to use instead of default
+        identity: Optional identity perspective to write from
         
     Returns:
         Tuple of (formatted_email, suggestions)
@@ -153,7 +166,7 @@ def process_email(
         email_guidelines = custom_guidelines
     
     # Get the email format prompt template
-    email_format_prompt = create_email_format_prompt(email_guidelines)
+    email_format_prompt = create_email_format_prompt(email_guidelines, identity)
     
     # Format the prompt using the template
     formatted_prompt = email_format_prompt.format(