Update app.py

app.py

@@ -83,6 +83,31 @@ def create_pdf_report(report_text):
     doc.build(Story)
     return pdf_path
 
+# Function to process and optimize resume
+def process_resume(resume_file, job_title):
+    if resume_file is None:
+        return "Please upload a resume file."
+    try:
+        file_path = resume_file.name
+        file_ext = os.path.splitext(file_path)[1].lower()
+        if file_ext == ".pdf":
+            resume_text = extract_text_from_pdf(file_path)
+        elif file_ext == ".docx":
+            resume_text = extract_text_from_docx(file_path)
+        else:
+            return "Unsupported file format. Please upload a PDF or DOCX file."
+        if resume_text.startswith("Error"):
+            return resume_text
+        prompt = (
+            f"Optimize the following resume for the job title: {job_title}.\n"
+            f"Include relevant skills, experience, and keywords related to the job title.\n\n"
+            f"Resume:\n{resume_text}\n"
+        )
+        optimized_resume = client.text_generation(prompt, max_new_tokens=1024)
+        return optimized_resume
+    except Exception as e:
+        return f"Error processing resume: {e}"
+
 # Build the Gradio UI
 demo = gr.Blocks()
 with demo:
@@ -93,10 +118,17 @@ with demo:
         job_desc_input = gr.Textbox(label="Job Description", lines=5)
         extracted_text = gr.Textbox(label="Extracted CV Content", lines=10, interactive=False)
         analysis_output = gr.Textbox(label="Analysis Report", lines=10, interactive=False)
-        pdf_file = gr.File(label="Download Analysis Report PDF", interactive=True)
         analyze_button = gr.Button("Analyze CV")
+        download_pdf_button = gr.File(label="Download Analysis Report PDF", interactive=False)
+        analyze_button.click(parse_cv, [file_input, job_desc_input], [extracted_text, analysis_output, download_pdf_button])
 
-        analyze_button.click(parse_cv, [file_input, job_desc_input], [extracted_text, analysis_output, pdf_file])
+    with gr.Tab("CV Optimizer"):
+        gr.Markdown("### Upload your Resume and Enter Job Title")
+        resume_file = gr.File(label="Upload Resume (PDF or Word)", file_types=[".pdf", ".docx"])
+        job_title_input = gr.Textbox(label="Job Title", lines=1)
+        optimized_resume_output = gr.Textbox(label="Optimized Resume", lines=20)
+        optimize_button = gr.Button("Optimize Resume")
+        optimize_button.click(process_resume, [resume_file, job_title_input], [optimized_resume_output])
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()