Update app_hug.py

app_hug.py

@@ -55,7 +55,7 @@ class MessageType(str, Enum):
     FOLLOWUP = "followup"
 
 class GenerateMessageRequest(BaseModel):
-    job_evaluation: str
+    job_evaluation: Optional[str] = Field(None, description="(Optional) Recruiter's evaluation of candidate for the job")
     sender_email: EmailStr
     reply_to_email: Optional[EmailStr] = Field(None, description="Recruiter's email for reply-to header")
     recipient_email: EmailStr
@@ -68,6 +68,7 @@ class GenerateMessageRequest(BaseModel):
     organisation: str
     message_type: MessageType
     send_email: bool = False
+    past_conversation: Optional[str] = Field(None, description="(Optional) Previous messages with candidate")
 
 class FeedbackRequest(BaseModel):
     message: str
@@ -178,7 +179,6 @@ def create_email_message(sender: str, to: str, subject: str, message_text: str,
     message["Subject"] = subject
     if reply_to:
         message["Reply-To"] = reply_to
-        message["Cc"] = reply_to 
     raw_message = base64.urlsafe_b64encode(message.as_bytes()).decode()
     return {"raw": raw_message}
 
@@ -255,8 +255,9 @@ def generate_recruitment_message_with_subject(
     org: str,
     candidate: str,
     current_pos: str,
-    evaluation: str,
-    feedback: Optional[str] = None
+    evaluation: Optional[str] = None,
+    feedback: Optional[str] = None,
+    past_conversation: Optional[str] = None
 ) -> Tuple[str, str]:
     api_key = os.getenv("OPENAI_API_KEY")
     llm = ChatOpenAI(
@@ -265,44 +266,64 @@ def generate_recruitment_message_with_subject(
         max_tokens=600,
         openai_api_key=api_key
     )
-    base_prompt = f"""
-    Generate a professional recruitment {msg_type} with the following details:
-    - Company hiring: {company}
-    - Role: {role_title}
-    - Recruiter: {recruiter} from {org}
-    - Candidate: {candidate}
-    - Candidate's current position: {current_pos}
-    - Evaluation: {evaluation}
-    """
+    # Outreach: Only request consent
     if msg_type == "outreach":
-        prompt = base_prompt + """
-        Create an initial outreach message that:
-        - Introduces the recruiter and organization
-        - Mentions the specific role and company
-        - Expresses interest in discussing the opportunity
-        - Keeps it short and to the point.
-        - Do not include any placeholders like [Candidate Name] or [Role Title] in the email.
-        """
-    elif msg_type == "introductory":
-        prompt = base_prompt + """
-        Create an introductory message that:
-        - Thanks the candidate for their initial response
-        - Provides more details about the role and company
-        - Explains why this opportunity aligns with their background
-        - Suggests next steps (like a call or meeting)
-        - Maintains a warm, professional tone
+        prompt = f"""
+        Generate a professional initial outreach message to a candidate.
+        - Introduce yourself as {recruiter} from {org}
+        - Clearly state you are reaching out about an open role ({role_title}) at {company}
+        - Ask if they are open to learning more or interested in a quick chat.
+        - Do NOT discuss any job evaluation or judgment.
+        - Explicitly request their consent to share more details. E.g., 'Would you be open to hearing more about this opportunity?'
+        - Keep it short and friendly.
+        - No placeholders like [Candidate Name] or [Role Title] in the output.
         """
-    else:  # followup
-        prompt = base_prompt + """
-        Create a follow-up message that:
-        - References previous communication
-        - Reiterates interest in the candidate
-        - Addresses any potential concerns
-        - Provides additional compelling reasons to consider the role
-        - Includes a clear call to action
+    else:
+        base_prompt = f"""
+        Generate a professional recruitment {msg_type} with the following details:
+        - Company hiring: {company}
+        - Role: {role_title}
+        - Recruiter: {recruiter} from {org}
+        - Candidate: {candidate}
+        - Candidate's current position: {current_pos}
         """
+        if evaluation:
+            base_prompt += f"- Evaluation: {evaluation}\n"
+        prompt = base_prompt
+        if msg_type == "introductory":
+            prompt += """
+            Create an introductory message that:
+            - Thanks the candidate for their initial response
+            - Provides more details about the role and company
+            - Explains why this opportunity aligns with their background
+            - Suggests next steps (like a call or meeting)
+            - Maintains a warm, professional tone
+            """
+        else:  # followup
+            prompt += """
+            Create a follow-up message that:
+            - References previous communication
+            - Reiterates interest in the candidate
+            - Addresses any potential concerns
+            - Provides additional compelling reasons to consider the role
+            - Includes a clear call to action
+            """
+
+    # Use feedback if provided
     if feedback:
         prompt += f"\n\nPlease modify the message based on this feedback: {feedback}"
+
+    # Use past conversation if provided
+    if past_conversation:
+        prompt += f"""
+        Use the following past conversation as context to inform your reply:
+
+        PAST CONVERSATION:
+        {past_conversation}
+
+        Write a reply message to the candidate, maintaining professionalism and continuity.
+        """
+
     prompt += """
 
     Please provide your response in the following format:
@@ -311,6 +332,7 @@ def generate_recruitment_message_with_subject(
     BODY:
     [Your email body content here]
     """
+
     try:
         messages = [
             SystemMessage(content="You are a professional recruitment message writer. Generate both an email subject line and body content. Follow the exact format requested."),
@@ -398,17 +420,17 @@ async def send_message(request: SendMessageRequest):
         
 
         formatted_html = format_email_html(
-    request.email_body
-)
+            request.email_body
+        )
         # Create the email
         email_message = create_email_message(
-    sender=request.sender_email,
-    to=request.recipient_email,
-    subject=request.subject,
-    message_text=formatted_html,
-    reply_to=request.reply_to_email,
-    is_html=True
-)
+            sender=request.sender_email,
+            to=request.recipient_email,
+            subject=request.subject,
+            message_text=formatted_html,
+            reply_to=request.reply_to_email,
+            is_html=True
+        )
         # Send the email
         email_sent = send_gmail_message(service, "me", email_message)
         if email_sent:
@@ -445,7 +467,8 @@ async def generate_message(request: GenerateMessageRequest, background_tasks: Ba
             org=request.organisation,
             candidate=request.candidate_name,
             current_pos=current_position,
-            evaluation=request.job_evaluation
+            evaluation=request.job_evaluation,
+            past_conversation=request.past_conversation
         )
         email_sent = False
         if request.send_email:
@@ -464,18 +487,18 @@ async def generate_message(request: GenerateMessageRequest, background_tasks: Ba
             service = authenticate_gmail(request.sender_email)
             if service:
                 formatted_html = format_email_html(
-    generated_message,
-    sender_name=request.recruiter_name,
-    sender_org=request.organisation
-)
+                    generated_message,
+                    sender_name=request.recruiter_name,
+                    sender_org=request.organisation
+                )
                 email_message = create_email_message(
-    sender=request.sender_email,
-    to=request.recipient_email,
-    subject=email_subject,
-    message_text=formatted_html,
-    reply_to=request.reply_to_email,
-    is_html=True
-)
+                    sender=request.sender_email,
+                    to=request.recipient_email,
+                    subject=email_subject,
+                    message_text=formatted_html,
+                    reply_to=request.reply_to_email,
+                    is_html=True
+                )
                 email_sent = send_gmail_message(service, "me", email_message)
                 if not email_sent:
                     return MessageResponse(