Update app.py

app.py

@@ -708,6 +708,87 @@ def process_pdf(file):
         error_message = f"Error processing PDF: {str(e)}"
         return error_message, error_message
 
+from docx import Document
+
+def generate_word_document(result):
+    doc = Document()
+
+    # Add a title
+    doc.add_heading('Job Description Analysis', level=1)
+
+    # Add file name
+    doc.add_heading('File Name', level=2)
+    doc.add_paragraph(result["file"])
+
+    # Add responsibilities
+    doc.add_heading('Responsibilities', level=2)
+    doc.add_paragraph(result["responsibilities"])
+
+    # Add job family
+    doc.add_heading('Classified Job Family', level=2)
+    doc.add_paragraph(result["classified_job_family"])
+
+    # Add qualifications
+    doc.add_heading('Qualification', level=2)
+    doc.add_paragraph("\n".join(result["qualification"]))
+
+    # Add CCOG Levels
+    doc.add_heading('CCOG Levels', level=2)
+    for i in range(1, 5):
+        for field in ["code", "name", "desc"]:
+            key = f"Level_{i}_CCOG_{field}"
+            if key in result:
+                doc.add_paragraph(f"{key}: {result[key]}")
+
+    # Add interview questions
+    doc.add_heading('Interview Questions', level=2)
+    doc.add_paragraph("\n".join(result["interview"]))
+
+    # Add skills
+    doc.add_heading('Skills', level=2)
+    for skill in result["skills"]["skills"]:
+        doc.add_paragraph(f"Skill Name: {skill['skill_name']}")
+        doc.add_paragraph(f"Description: {skill['skill_description']}")
+        doc.add_paragraph(f"Code: {skill['skill_code']}")
+        doc.add_paragraph(f"Importance: {skill.get('importance', 'N/A')}")
+        doc.add_paragraph(f"Type: {skill.get('type', 'N/A')}")
+        doc.add_paragraph(f"Proficiency Level: {skill.get('proficiency_level', 'N/A')}")
+        doc.add_paragraph(f"Distinctive Elements: {skill.get('distinctive_elements', 'N/A')}")
+        doc.add_paragraph(f"Resume Signals: {skill.get('resume_signals', 'N/A')}")
+        doc.add_paragraph(f"Assessment Method: {skill.get('assessment_method', 'N/A')}")
+        doc.add_paragraph("")  # Add an empty line for separation
+
+    # Add ESCO Levels if available
+    if "skills_esco" in result and result["skills_esco"]:
+        doc.add_heading('ESCO Levels', level=2)
+        for i in range(1, 6):
+            for field in ["code", "name", "desc"]:
+                key = f"Level_{i}_ESCO_{field}"
+                if key in result:
+                    doc.add_paragraph(f"{key}: {result[key]}")
+
+        # Add ESCO Skills
+        doc.add_heading('ESCO Skills', level=2)
+        for skill in result["skills_esco"]["skills"]:
+            doc.add_paragraph(f"Skill Name: {skill['skill_name']}")
+            doc.add_paragraph(f"Description: {skill['skill_description']}")
+            doc.add_paragraph(f"Code: {skill['skill_code']}")
+            doc.add_paragraph(f"Importance: {skill.get('importance', 'N/A')}")
+            doc.add_paragraph(f"Type: {skill.get('type', 'N/A')}")
+            doc.add_paragraph(f"Proficiency Level: {skill.get('proficiency_level', 'N/A')}")
+            doc.add_paragraph(f"Distinctive Elements: {skill.get('distinctive_elements', 'N/A')}")
+            doc.add_paragraph(f"Resume Signals: {skill.get('resume_signals', 'N/A')}")
+            doc.add_paragraph(f"Assessment Method: {skill.get('assessment_method', 'N/A')}")
+            doc.add_paragraph("")  # Add an empty line for separation
+
+    # Save the document to a temporary file
+    temp_file_path = "job_description_analysis.docx"
+    doc.save(temp_file_path)
+
+    return temp_file_path
+
+
+
 with gr.Blocks() as demo:
     gr.Markdown("# Standardize Job Description!")
     gr.Markdown("Identify Job Family, Occupation, Qualification, match Skills and suggest interview questions.")
@@ -727,7 +808,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown("## CCOG Levels")
-            ccoq_levels_output = gr.JSON(label="CCOG Levels", interactive=False)
+            ccoq_levels_output = gr.JSON(label="CCOG Levels")
 
     with gr.Row():
         with gr.Column():
@@ -737,17 +818,21 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
             gr.Markdown("## Skills")
-            skills_output = gr.JSON(label="Skills", interactive=False)
+            skills_output = gr.JSON(label="Skills")
 
     with gr.Row():
         with gr.Column():
             gr.Markdown("## ESCO Levels")
-            esco_levels_output = gr.JSON(label="ESCO Levels", interactive=False)
+            esco_levels_output = gr.JSON(label="ESCO Levels")
 
     with gr.Row():
         with gr.Column():
             gr.Markdown("## ESCO Skills")
-            esco_skills_output = gr.JSON(label="ESCO Skills", interactive=False)
+            esco_skills_output = gr.JSON(label="ESCO Skills")
+
+    with gr.Row():
+        with gr.Column():
+            download_btn = gr.Button("Download Word Document")
 
     if DEBUG:
         with gr.Row():
@@ -777,5 +862,14 @@ with gr.Blocks() as demo:
         ]
     )
 
+    download_btn.click(
+        fn=generate_word_document,
+        inputs=[file_name_output, responsibilities_output, job_family_output, qualification_output,
+                ccoq_levels_output, interview_output, skills_output, esco_levels_output, esco_skills_output],
+        outputs=gr.File(label="Download Word Document")
+    )
+
 if __name__ == "__main__":
     demo.launch(show_error=True, debug=DEBUG)
+
+