Spaces:
Sleeping
Sleeping
File size: 4,894 Bytes
10fbe29 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
from reportlab.pdfgen import canvas
from reportlab.pdfbase.ttfonts import TTFont
from reportlab.pdfbase import pdfmetrics
from PyPDF2 import PdfReader
from PyPDF2 import PdfWriter
import openai
import uuid
import os
import gradio as gr
# 確保字體文件和模板文件存在於同一目錄
pdfmetrics.registerFont(TTFont("ArialUnicode", "./Arial Unicode.ttf"))
def draw_data_on_template(template_path, name, email, phone, education, experience, skills, social_media):
try:
temp_pdf_path = f"./{uuid.uuid4()}_temp.pdf"
c = canvas.Canvas(temp_pdf_path)
c.setFont("ArialUnicode", 22)
c.drawString(15, 750, f"Name: {name}")
c.drawString(15, 680, f"Email: {email}")
c.drawString(15, 610, f"Phone: {phone}")
c.drawString(15, 540, f"Education: {education}")
c.drawString(15, 470, f"Work Experience: {experience}")
c.drawString(15, 340, f"Skills: {skills}")
c.drawString(15, 200, f"Social Media: {social_media}")
c.save()
# 加載模板 PDF
reader = PdfReader(template_path)
template_page = reader.pages[0]
content_reader = PdfReader(temp_pdf_path)
content_page = content_reader.pages[0]
template_page.merge_page(content_page)
merged_pdf_path = f"./{uuid.uuid4()}_merged.pdf"
writer = PdfWriter()
writer.add_page(template_page)
with open(merged_pdf_path, "wb") as output_file:
writer.write(output_file)
os.remove(temp_pdf_path)
return merged_pdf_path
except Exception as e:
return f"Error in Drawing Data on Template: {str(e)}"
def recommend_jobs(api_key, skills):
try:
openai.api_key = api_key
skills_list = [skill.strip() for skill in skills.split(",") if skill.strip()]
if not skills_list:
return "Error: No valid skills provided."
prompt = f"請用正體中文回覆，根據以下技能推薦適合的職位：{', '.join(skills_list)}"
response = openai.ChatCompletion.create(
model="gpt-4",
messages=[{"role": "user", "content": prompt}]
)
if not response.choices:
return "Error: OpenAI API returned no results."
return response.choices[0].message['content'].strip()
except Exception as e:
return f"Error in Job Recommendation: {str(e)}"
def validate_and_process(api_key, name, email, phone, education, experience, skills, social_media):
errors = []
if not name.strip():
errors.append("Name不能為空.")
if not email.strip():
errors.append("Email不能為空.")
if not phone.strip():
errors.append("Phone不能為空.")
if not education.strip():
errors.append("Education不能為空.")
if not skills.strip():
errors.append("Skills不能為空.")
if errors:
return None, "\n".join(errors), "請填寫完整後重試！"
template_path = "./resume_null.pdf" # 確保模板文件上傳至同一目錄
merged_pdf = draw_data_on_template(template_path, name, email, phone, education, experience, skills, social_media)
if isinstance(merged_pdf, str) and merged_pdf.startswith("Error"):
return None, merged_pdf, "Merging with template failed."
job_recommendations = recommend_jobs(api_key, skills)
if job_recommendations.startswith("Error"):
return merged_pdf, "Resume generated successfully!", job_recommendations
return merged_pdf, "Resume generated successfully!", job_recommendations
with gr.Blocks() as demo:
gr.Markdown("# 自動生成簡歷工具")
api_key = gr.Textbox(label="OpenAI API Key", placeholder="請輸入您的 OpenAI API Key", type="password")
name = gr.Textbox(label="Name（必填）", placeholder="例：王小明")
email = gr.Textbox(label="Email（必填）", placeholder="a1234567890@example.com")
phone = gr.Textbox(label="Phone（必填）", placeholder="0912345678")
education = gr.Textbox(label="Education（必填）", placeholder="例：學位、主修、學校")
experience = gr.Textbox(label="Work Experience", placeholder="例：實習、兼職工作")
skills = gr.Textbox(label="Skills（必填）", placeholder="例：Python, Teaching, Data Analysis (多項技能用,分隔)")
social_media = gr.Textbox(label="Social Media", placeholder="例：LinkedIn URL")
pdf_output = gr.File(label="Download Your Resume")
status = gr.Textbox(label="Status / Error", interactive=False)
job_recommendations = gr.Textbox(label="Recommended Jobs", interactive=False)
generate_button = gr.Button("Generate Resume & Recommend Jobs")
generate_button.click(
validate_and_process,
inputs=[api_key, name, email, phone, education, experience, skills, social_media],
outputs=[pdf_output, status, job_recommendations]
)
demo.launch()
|