Update app_hug.py

app_hug.py

@@ -114,6 +114,11 @@ class GenerateMessageRequest(BaseModel):
     message_type: MessageType
     send_email: bool = False
     past_conversation: Optional[str] = Field(None, description="(Optional) Previous messages with candidate")
+    job_location: Optional[str] = Field(None, description="(Optional) Job location (e.g., 'Sterling, VA')")
+    company_blurb: Optional[str] = Field(None, description="(Optional) Brief description of company mission/industry")
+    culture: Optional[str] = Field(None, description="(Optional) Company culture description (e.g., 'fast-moving, tight-knit team')")
+    job_responsibilities: Optional[str] = Field(None, description="(Optional) Key responsibilities and duties for the role")
+    candidate_experience: Optional[str] = Field(None, description="(Optional) Specific skills and experience of the candidate")
 
 class FeedbackRequest(BaseModel):
     message: str
@@ -276,14 +281,23 @@ def refine_message_with_feedback(
         lines = content.split('\n')
         body_found = False
         body_lines = []
+        
         for line in lines:
-            if line.strip().startswith('SUBJECT:'):
-                subject_line = line.replace('SUBJECT:', '').strip()
-            elif line.strip().startswith('BODY:'):
+            line_stripped = line.strip()
+            # Handle both plain text and markdown formats
+            if line_stripped.startswith('SUBJECT:') or line_stripped.startswith('**SUBJECT:**'):
+                subject_line = line_stripped.replace('SUBJECT:', '').replace('**SUBJECT:**', '').replace('**', '').strip()
+            elif line_stripped.startswith('BODY:') or line_stripped.startswith('**BODY:**'):
                 body_found = True
-            elif body_found and line.strip():
+            elif body_found and line_stripped:
                 body_lines.append(line)
+        
         body_content = '\n'.join(body_lines).strip()
+        
+        # Clean up any remaining markdown formatting from body
+        if body_content:
+            body_content = body_content.replace('**', '')
+        
         if not subject_line:
             subject_line = "Recruitment Opportunity - Updated"
         if not body_content:
@@ -302,7 +316,12 @@ def generate_recruitment_message_with_subject(
     current_pos: str,
     evaluation: Optional[str] = None,
     feedback: Optional[str] = None,
-    past_conversation: Optional[str] = None
+    past_conversation: Optional[str] = None,
+    job_location: Optional[str] = None,
+    company_blurb: Optional[str] = None,
+    culture: Optional[str] = None,
+    job_responsibilities: Optional[str] = None,
+    candidate_experience: Optional[str] = None
 ) -> Tuple[str, str]:
     api_key = os.getenv("OPENAI_API_KEY")
     llm = ChatOpenAI(
@@ -333,6 +352,18 @@ def generate_recruitment_message_with_subject(
         - Candidate: {candidate}
         - Candidate's current position: {current_pos}
         """
+        
+        # Add optional additional details if provided
+        if job_location:
+            base_prompt += f"- Job location: {job_location}\n"
+        if company_blurb:
+            base_prompt += f"- Company description: {company_blurb}\n"
+        if culture:
+            base_prompt += f"- Company culture: {culture}\n"
+        if job_responsibilities:
+            base_prompt += f"- Job responsibilities: {job_responsibilities}\n"
+        if candidate_experience:
+            base_prompt += f"- Candidate's specific experience: {candidate_experience}\n"
         if evaluation:
             base_prompt += f"- Evaluation: {evaluation}\n"
         prompt = base_prompt
@@ -394,14 +425,23 @@ def generate_recruitment_message_with_subject(
         lines = content.split('\n')
         body_found = False
         body_lines = []
+        
         for line in lines:
-            if line.strip().startswith('SUBJECT:'):
-                subject_line = line.replace('SUBJECT:', '').strip()
-            elif line.strip().startswith('BODY:'):
+            line_stripped = line.strip()
+            # Handle both plain text and markdown formats
+            if line_stripped.startswith('SUBJECT:') or line_stripped.startswith('**SUBJECT:**'):
+                subject_line = line_stripped.replace('SUBJECT:', '').replace('**SUBJECT:**', '').replace('**', '').strip()
+            elif line_stripped.startswith('BODY:') or line_stripped.startswith('**BODY:**'):
                 body_found = True
-            elif body_found and line.strip():
+            elif body_found and line_stripped:
                 body_lines.append(line)
+        
         body_content = '\n'.join(body_lines).strip()
+        
+        # Clean up any remaining markdown formatting from body
+        if body_content:
+            body_content = body_content.replace('**', '')
+        
         if not subject_line:
             subject_line = f"Opportunity at {company} - {role_title}"
         if not body_content:
@@ -599,7 +639,12 @@ async def generate_message(
     job_evaluation: Optional[str] = Query(None, description="Recruiter's evaluation of candidate for the job"),
     reply_to_email: Optional[str] = Query(None, description="Recruiter's email for reply-to header"),
     send_email: bool = Query(False, description="Whether to send the email"),
-    past_conversation: Optional[str] = Query(None, description="Previous messages with candidate")
+    past_conversation: Optional[str] = Query(None, description="Previous messages with candidate"),
+    job_location: Optional[str] = Query(None, description="Job location (e.g., 'Sterling, VA')"),
+    company_blurb: Optional[str] = Query(None, description="Brief description of company mission/industry"),
+    culture: Optional[str] = Query(None, description="Company culture description (e.g., 'fast-moving, tight-knit team')"),
+    job_responsibilities: Optional[str] = Query(None, description="Key responsibilities and duties for the role"),
+    candidate_experience: Optional[str] = Query(None, description="Specific skills and experience of the candidate")
 ):
     try:
         current_position = f"{current_role} at {current_company}"
@@ -612,7 +657,12 @@ async def generate_message(
             candidate=candidate_name,
             current_pos=current_position,
             evaluation=job_evaluation,
-            past_conversation=past_conversation
+            past_conversation=past_conversation,
+            job_location=job_location,
+            company_blurb=company_blurb,
+            culture=culture,
+            job_responsibilities=job_responsibilities,
+            candidate_experience=candidate_experience
         )
         email_sent = False
         if send_email:
@@ -756,4 +806,8 @@ async def get_documentation():
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=8000)
+
+
+
+# Additional Infor: Job Location, Company Blurb, Culture, Job Responsibilities, Candidate Experience