Checkpoint that includes all fields. Need to add confirmation.

app.py

@@ -142,6 +142,8 @@ def get_key_accomplishments_prompt(job_and_company_info, 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."
 """
 
+    return key_skills_and_accomplaishments
+
 class Role:
     def __init__(self, company_name, company_url, job_description, clean_job_description, job_title):
         self.company_name = company_name
@@ -169,7 +171,7 @@ def smol_lm_jd_process(job_description, system_prompt, max_new_tokens=512):
     response = response[start_idx + len("<|im_start|>assistant\n"):end_idx].strip()
     return response
 
-def process_job_description(company_name, company_url, job_description):
+def process_job_description(company_name, company_url, job_description, resume):
     
     # Step 2: Extract key qualifications, skills, and requirements
     system_prompt_requirements = "Extract key qualifications, skills, and requirements from this job description. Output as bullet points. Remove benefits/salary, bragging about the company, and other fluff not relevant to the skills, qualifications, and job requirements. ONLY INCLUDE INFORMATION THAT TELLS THE USER WHAT SKILLS THE EMPLOYER SEEKS."
@@ -242,109 +244,52 @@ def process_job_description(company_name, company_url, job_description):
         }
     }
 
-    return job_and_company_info, role
-
-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']}
-    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']}
-    """
-
-    prompt = GAP_ASSESSMENT_PROMPT.replace("{JOB_AND_COMPANY_INFO}", job_and_company_info_str).replace("{RESUME}", resume)
+    gap_assessment_prompt = get_gap_assessment_prompt(job_and_company_info, resume)
+    return job_and_company_info, role, gap_assessment_prompt
 
-    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 = 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
+def generate_key_accomplishments_and_skills_prompt(job_and_company_info, resume):
+    return get_key_accomplishments_prompt(job_and_company_info, resume)
 
 # Create the Gradio app
 demo = gr.Blocks()
 
 with demo:
-    gr.Markdown("# Job Description Input")
-    company_name = gr.Textbox(label="Company Name")
-    company_url = gr.Textbox(label="Company URL")
-    job_description = gr.TextArea(label="Paste Job Description")
+    gr.Markdown("# Job Description and Resume Input")
+    with gr.Row():
+        company_name = gr.Textbox(label="Company Name")
+        company_url = gr.Textbox(label="Company URL")
+        job_description = gr.TextArea(label="Paste Job Description")
+        resume = gr.TextArea(label="Paste Resume")
 
     job_and_company_info_output = gr.JSON(label="Job and Company Info")
     role_output = gr.State()
+    gap_assessment_prompt_output = gr.TextArea(label="Gap Assessment Prompt")
 
     gr.Button("Submit").click(
         process_job_description,
-        inputs=[company_name, company_url, job_description],
-        outputs=[job_and_company_info_output, role_output]
+        inputs=[company_name, company_url, job_description, resume],
+        outputs=[job_and_company_info_output, role_output, gap_assessment_prompt_output]
     )
 
-    gr.Markdown("# Resume Input")
-    resume = gr.TextArea(label="Paste Resume")
-
-    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)
+    gap_assessment_result = gr.TextArea(label="Paste Gap Assessment Result")
+    gr.Button("Submit Gap Assessment Result").click(
+        None,
+        inputs=[gap_assessment_result],
+        outputs=None
+    )
 
-    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,
+    key_accomplishments_and_skills_prompt_output = gr.TextArea(label="Key Accomplishments and Skills Prompt")
+    gr.Button("Generate Key Accomplishments and Skills Prompt").click(
+        generate_key_accomplishments_and_skills_prompt,
         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", 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,
+    key_accomplishments_and_skills_result = gr.TextArea(label="Paste Key Accomplishments and Skills Result")
+    gr.Button("Submit Key Accomplishments and Skills Result").click(
+        None,
         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)]
+        outputs=None
     )
 
 if __name__ == "__main__":