Update app.py

app.py

@@ -4,6 +4,8 @@ import PyPDF2
 import docx
 from docx import Document
 import os
+from docx2pdf import convert
+
 
 
 
@@ -23,9 +25,9 @@ messages = []
 
 latest_resume_text = ""
 desired_position = ""
-default_text_1 = "Hello, I will share a candidate's resume with you. Please provide a summary for me" 
+default_text_1 = "请帮我总结我上传的简历" 
                 
-default_text_2 = "Hello, I will share a candidate's resume with you. Please desensitize the resume by removing or redacting any personal contact information like the phone number, home address, and email address. Also remove or mask any private details like ID numbers, exact dates for education and work history, and compensation details. Double check that no other private information is included. The goal is to remove personally identifying details while still allowing this candidate's professional qualifications to be visible to potential employers."
+default_text_2 = "请帮我把我上传的简历进行脱敏处理。"
 
 default_text_3 = "请帮我阅读候选人简历并总结出以下几个部分。第一个板块是“个人信息”，请总结出此后选人的姓名，性别，工作经验（多少年），最高学历，毕业院校，专业，和毕业时间。第二个部分是此候选人的本人评价，放进一个自然段里。第三个部分是具体得工作经历。第四个部分是做过的项目经验。请根据你对于这位候选人简历的最细致的阅读排出以上几个部分。每个部分由自己的标题：五个标题为 个人信息，本人评价，工作经历，和项目经验。请把每一个小项写得细致一点，并且用数字排序！"
 
@@ -240,7 +242,12 @@ def generate_resume_document(latest_resume_text):
     doc.save(word_filename)
     return word_filename
 
-
+# create a function that automatically convert a docx file to a pdf file
+def convert_to_pdf(word_filename):
+    pdf_filename = word_filename.replace(".docx", ".pdf")
+    doc = docx.Document(word_filename)
+    doc.save(pdf_filename)
+    return pdf_filename
 
 
 
@@ -279,6 +286,7 @@ def CustomChatGPT(user_input, uploaded_file):
 
     if "请帮我阅读候选人简历" in user_input:
         word_filename = generate_resume_document(latest_resume_text)
+        pdf_filename = convert_to_pdf(word_filename)
         ChatGPT_reply = "Resume generated successfully. Please download the file using the link below."
         messages = []
     else:
@@ -286,10 +294,14 @@ def CustomChatGPT(user_input, uploaded_file):
         doc.add_paragraph(ChatGPT_reply)
         word_filename = "generated_response.docx"
         doc.save(word_filename)
+        # pdf_filename = convert_to_pdf(word_filename)
+
+        convert(word_filename, pdf_filename)
+
         # reset the messages to start a new conversation
         messages = []
 
-    return ChatGPT_reply, word_filename
+    return ChatGPT_reply, word_filename, pdf_filename
 
 
 
@@ -330,7 +342,7 @@ def main():
     """
 
     with gr.Blocks(css=custom_css) as demo:
-        gr.Markdown("# Resume Analysis Tool", elem_classes=["centered-title"])
+        gr.Markdown("# 简历分析系统", elem_classes=["centered-title"])
 
         with gr.Row():
             with gr.Column():
@@ -338,8 +350,8 @@ def main():
                 file_input = gr.File(file_count="multiple", label="Upload Resume")
 
                 with gr.Row():
-                    btn1 = gr.Button("Summary", elem_classes=["custom-button"])
-                    btn2 = gr.Button("Desensitization", elem_classes=["custom-button"])
+                    btn1 = gr.Button("总结简历", elem_classes=["custom-button"])
+                    btn2 = gr.Button("脱敏处理", elem_classes=["custom-button"])
                     btn3 = gr.Button("简历生成", elem_classes=["custom-button"])
 
 
@@ -349,26 +361,27 @@ def main():
                     
 
                 with gr.Row():
-                    submit_btn = gr.Button("Submit", elem_classes=["custom-button"])
-                    clear_btn = gr.Button("Clear", elem_classes=["custom-button"])
+                    submit_btn = gr.Button("提交", elem_classes=["custom-button"])
+                    clear_btn = gr.Button("清楚", elem_classes=["custom-button"])
                     clear_btn.click(fn=clear_inputs, inputs=[], outputs=[])
 
             with gr.Column():
                 with gr.Row():
-                    output_text = gr.Textbox(label="ChatGPT Reply", interactive=True, lines=1)
+                    output_text = gr.Textbox(label="ChatGPT回复", interactive=True, lines=1)
                 with gr.Row():
-                    output_word = gr.File(label="Download Word File")
+                    output_word = gr.File(label="下载word文件")
                     # output_pdf = gr.File(label="Download PDF File")
+                    output_pdf = gr.File(label="下载PDF文件")
 
-                    submit_btn.click(fn=CustomChatGPT, inputs=[text_input, file_input], outputs=[output_text, output_word])
+                    submit_btn.click(fn=CustomChatGPT, inputs=[text_input, file_input], outputs=[output_text, output_word, output_pdf])
 
                 with gr.Row():
                     log_text = gr.Textbox(label="Log Content", interactive=True, lines=1)
 
                 with gr.Row():
-                    view_log_button = gr.Button("View Log", elem_classes=["custom-button"])
-                    download_log_button = gr.Button("Download Log File", elem_classes=["custom-button"])
-                    clear_log_button = gr.Button("Clear Log File", elem_classes=["custom-button"])
+                    view_log_button = gr.Button("对话记录", elem_classes=["custom-button"])
+                    download_log_button = gr.Button("下载对话记录", elem_classes=["custom-button"])
+                    clear_log_button = gr.Button("清楚对话记录", elem_classes=["custom-button"])
 
                     view_log_button.click(fn=read_log_file, inputs=[], outputs=log_text)
                     download_log_button.click(fn=get_log_file, inputs=[], outputs=[])
@@ -379,9 +392,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
-
-
-
-
-
+    main()