Checkpoint that includes the prompt templates for this task.

app.py

@@ -12,7 +12,8 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
-GAP_ASSESSMENT_PROMPT = """
+def get_gap_assessment_prompt(job_and_company_info, resume):
+    gap_assessment_prompt = f"""
 
 # Do a gap assessment on this job role and resume looking for gaps that can be overcome by strategy in writing the resume and cover letter. For each gap, include a strategy for remediation. Ignore and do not mention any gap that is structural and for which improved writing will not be helpful. Consider things like:
 
@@ -91,15 +92,54 @@ Remediation Strategies: Briefly explain significant gaps in the cover letter, fr
 
 # This is the role description, company information, and results of company research, and 
 
-{JOB_AND_COMPANY_INFO}
+    Company Name: {job_and_company_info['Company Name']}
+    Company URL: {job_and_company_info['Company URL']}
+    Job Title: {job_and_company_info['job_title']}
+    Role Requirements: {job_and_company_info['Role Requirements']}
+    Clean Job Description: {job_and_company_info['Clean Job Description']}
+    Company Research:
+        Company Values: {job_and_company_info['Company Research']['Company Values Summary']}
+        Corporate Culture: {job_and_company_info['Company Research']['Corporate Culture Summary']}
+        Leadership Team and Possible Key Opinion Leaders: {job_and_company_info['Company Research']['Leadership Team Summary']}
+        Recent Company News: {job_and_company_info['Company Research']['Recent News Summary']}
+        Company Competitive Advantages: {job_and_company_info['Company Research']['Competitive Advantages Summary']}
 
 
 ## This is the candidate's resume:
 
-{RESUME}
+{resume}
 
 # For this task, do not re-write their resume. Simply identify any of these described gaps or other potential gaps and strategies in re-writing and re-framing the resume which are likely to be effective in improving the odds of being offered an interview or job. 
  
+"""
+    return gap_assessment_prompt
+
+
+def get_key_accomplishments_prompt(job_and_company_info, resume):
+
+    key_skills_and_accomplaishments = f"""
+# Extract relevant 1. 1-5 key accomplishments accomplishments and 2. all transferable skills from the applicant's resume that aligns with the job description.
+
+# This is the role description, company information, and results of company research, and 
+
+    Company Name: {job_and_company_info['Company Name']}
+    Company URL: {job_and_company_info['Company URL']}
+    Job Title: {job_and_company_info['job_title']}
+    Role Requirements: {job_and_company_info['Role Requirements']}
+    Clean Job Description: {job_and_company_info['Clean Job Description']}
+    Company Research:
+        Company Values: {job_and_company_info['Company Research']['Company Values Summary']}
+        Corporate Culture: {job_and_company_info['Company Research']['Corporate Culture Summary']}
+        Leadership Team and Possible Key Opinion Leaders: {job_and_company_info['Company Research']['Leadership Team Summary']}
+        Recent Company News: {job_and_company_info['Company Research']['Recent News Summary']}
+        Company Competitive Advantages: {job_and_company_info['Company Research']['Competitive Advantages Summary']}
+
+
+## This is the candidate's resume:
+
+{resume}
+
+Please provide the following information in a concise and straightforward manner, without any extraneous comments or information that may confuse an LLM that this response will be piped to: Output the results in a simple format, with the key accomplishments first, followed by the transferable skills."
 """
 
 class Role:
@@ -110,6 +150,10 @@ class Role:
         self.clean_job_description = clean_job_description
         self.job_title = job_title
 
+class Applicant:
+    def __init__(self, resume):
+        self.resume = resume
+
 def smol_lm_jd_process(job_description, system_prompt, max_new_tokens=512):
     prompt = f"""<|im_start|>system
 {system_prompt}<|im_end|>
@@ -198,9 +242,9 @@ def process_job_description(company_name, company_url, job_description):
         }
     }
 
-    return job_and_company_info
+    return job_and_company_info, role
 
-def generate_gap_assessment_prompt(job_and_company_info, resume):
+def generate_gap_assessment_prompt(job_and_company_info, resume, role):
     job_and_company_info_str = f"""
     Company Name: {job_and_company_info['Company Name']}
     Company URL: {job_and_company_info['Company URL']}
@@ -219,13 +263,24 @@ def generate_gap_assessment_prompt(job_and_company_info, resume):
 
     return prompt
 
+def generate_key_accomplishments_and_skills_prompt(role, applicant):
+    prompt = KEY_ACCOMPLISHMENTS_AND_SKILLS.replace("{Role.clean_job_description}", role.clean_job_description).replace("{Applicant.resume}", applicant.resume)
+    return prompt
+
 def gap_assessment_interface(job_and_company_info, resume):
-    prompt = generate_gap_assessment_prompt(job_and_company_info, resume)
+    prompt = get_gap_assessment_prompt(job_and_company_info, resume)
     return gr.TextArea(label="Gap Assessment Prompt", value=prompt, interactive=False)
 
+def key_accomplishments_and_skills_interface(job_and_company_info, resume):
+    prompt = get_key_accomplishments_prompt(job_and_company_info, resume)
+    return gr.TextArea(label="Key Accomplishments and Skills Prompt", value=prompt, interactive=False)
+
 def submit_gap_assessment_result(gap_assessment_result):
     return gap_assessment_result
 
+def submit_key_accomplishments_and_skills_result(key_accomplishments_and_skills_result):
+    return key_accomplishments_and_skills_result
+
 # Create the Gradio app
 demo = gr.Blocks()
 
@@ -236,33 +291,62 @@ with demo:
     job_description = gr.TextArea(label="Paste Job Description")
 
     job_and_company_info_output = gr.JSON(label="Job and Company Info")
+    role_output = gr.State()
 
     gr.Button("Submit").click(
         process_job_description,
         inputs=[company_name, company_url, job_description],
-        outputs=job_and_company_info_output
+        outputs=[job_and_company_info_output, role_output]
     )
 
     gr.Markdown("# Resume Input")
     resume = gr.TextArea(label="Paste Resume")
 
-    gap_assessment_prompt_output = gr.TextArea(label="Gap Assessment Prompt")
+    gap_assessment_prompt_output = gr.TextArea(label="Gap Assessment Prompt", visible=False)
+    key_accomplishments_and_skills_prompt_output = gr.TextArea(label="Key Accomplishments and Skills Prompt", visible=False)
 
-    gr.Button("Generate Gap Assessment Prompt").click(
+    gr.Button("Generate Prompts").click(
         gap_assessment_interface,
         inputs=[job_and_company_info_output, resume],
         outputs=gap_assessment_prompt_output
+    ).then(
+        key_accomplishments_and_skills_interface,
+        inputs=[job_and_company_info_output, resume],
+        outputs=key_accomplishments_and_skills_prompt_output
+    ).then(
+        lambda: [gr.update(visible=True), gr.update(visible=True)],
+        outputs=[gap_assessment_prompt_output, key_accomplishments_and_skills_prompt_output]
     )
 
-    gr.Markdown("# Gap Assessment Result")
-    gap_assessment_result = gr.TextArea(label="Paste Gap Assessment Result")
-
-    gr.Button("Submit Gap Assessment Result").click(
+    gr.Markdown("# Gap Assessment Result", visible=False)
+    gap_assessment_result = gr.TextArea(label="Paste Gap Assessment Result", visible=False)
+    gr.Button("Submit Gap Assessment Result", visible=False).click(
         submit_gap_assessment_result,
         inputs=[gap_assessment_result],
         outputs=gr.TextArea(label="Gap Assessment Result")
     )
 
+    gr.Markdown("# Key Accomplishments and Skills Result", visible=False)
+    key_accomplishments_and_skills_result = gr.TextArea(label="Paste Key Accomplishments and Skills Result", visible=False)
+    gr.Button("Submit Key Accomplishments and Skills Result", visible=False).click(
+        submit_key_accomplishments_and_skills_result,
+        inputs=[key_accomplishments_and_skills_result],
+        outputs=gr.TextArea(label="Key Accomplishments and Skills Result")
+    )
+
+    def show_results():
+        return [gr.update(visible=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)]
+
+    gr.Button("Generate Prompts").click(
+        show_results,
+        outputs=[gr.Markdown("# Gap Assessment Result", visible=False),
+                 gap_assessment_result,
+                 gr.Button("Submit Gap Assessment Result", visible=False),
+                 gr.Markdown("# Key Accomplishments and Skills Result", visible=False),
+                 key_accomplishments_and_skills_result,
+                 gr.Button("Submit Key Accomplishments and Skills Result", visible=False)]
+    )
+
 if __name__ == "__main__":
     demo.launch()