Update app.py

app.py

@@ -6,21 +6,22 @@ from weasyprint import HTML
 from markitdown import MarkItDown
 from cerebras.cloud.sdk import Cerebras
 
-# Ambil API key dari environment variable
+# Pastikan Anda memiliki file style.css di direktori yang sama
+# atau sesuaikan path 'stylesheets' di fungsi export_resume
+
+# Dapatkan API key dari environment variables
 api_key = os.environ.get("CEREBRAS_API_KEY")
 
-# Inisialisasi MarkItDown untuk konversi file (pdf/docx/dll) ke Markdown
+# Inisialisasi MarkItDown
 md_converter = MarkItDown()
 
 def create_prompt(resume_string: str, jd_string: str) -> str:
     """
-    Membuat prompt detail agar AI melakukan optimasi resume berdasarkan job description.
-    Di sini kita tambahkan instruksi khusus tentang formatting, agar Work Experience 
-    berbentuk bullet list.
+    Membuat prompt detail untuk AI agar melakukan optimasi resume
+    berdasarkan job description.
     """
     return f"""
-You are a professional resume optimization expert specializing in tailoring resumes to specific job descriptions. 
-Your goal is to optimize my resume and provide actionable suggestions for improvement to align with the target role.
+You are a professional resume optimization expert specializing in tailoring resumes to specific job descriptions. Your goal is to optimize my resume and provide actionable suggestions for improvement to align with the target role.
 
 ### Guidelines:
 1. **Relevance**:  
@@ -43,11 +44,7 @@ Your goal is to optimize my resume and provide actionable suggestions for improv
 
 5. **Formatting**:  
    - Output the tailored resume in **clean Markdown format**.  
-   - Use "##" for main headings (WORK EXPERIENCE, EDUCATION, etc.).
-   - Use "###" for each role in WORK EXPERIENCE or each education entry.
-   - For each role in WORK EXPERIENCE, use bullet points ("- ") in separate lines to describe responsibilities/achievements.
    - Include an **"Additional Suggestions"** section at the end with actionable improvement recommendations.  
-   - Resume should not exceed one page if possible.
 
 ---
 
@@ -97,30 +94,26 @@ def remove_unwanted_headings(markdown_text: str) -> str:
     """
     Menghapus heading apa pun yang mengandung kata 'resume' atau 'optimized'
     (dalam berbagai huruf besar/kecil).
-    Contoh heading yang akan dihapus: '# Resume', '## optimized', dsb.
+    Contoh heading yang akan dihapus:
+    # Resume
+    ## optimized
+    ### Optimized Resume
+    dsb.
     """
     pattern = r'^#+.*\b(?:[Rr]esume|[Oo]ptimized)\b.*$'
     return re.sub(pattern, '', markdown_text, flags=re.MULTILINE)
 
-def fix_work_experience_bullets(text: str) -> str:
-    """
-    Mencari pola ' - ' di tengah kalimat (yang menandakan bullet),
-    lalu memecahnya ke baris baru agar menjadi bullet list Markdown yang valid.
-    """
-    # Ganti " - " dengan "\n- " agar bullet pindah ke baris baru
-    return re.sub(r'\s-\s', '\n- ', text)
-
 def process_resume(resume, jd_string):
     """
-    Memproses file resume (pdf, docx, dll) dan job description,
-    lalu menghasilkan resume yang telah dioptimasi + saran perbaikan.
+    Memproses file resume yang di-upload dan job description, lalu
+    menghasilkan resume yang telah dioptimasi + saran perbaikan.
     """
-    # Ekstensi yang didukung
+    # Format yang didukung
     supported_extensions = ('.pptx', '.docx', '.pdf', '.jpg', '.jpeg', '.png', '.xlsx')
     
     # Cek apakah file resume memiliki ekstensi yang didukung
     if resume.name.lower().endswith(supported_extensions):
-        # Konversi file ke Markdown
+        # Konversi file ke Markdown menggunakan MarkItDown
         result = md_converter.convert(resume.name)
         resume_string = result.text_content  # konten Markdown hasil konversi
     else:
@@ -128,20 +121,25 @@ def process_resume(resume, jd_string):
     
     # Buat prompt untuk AI
     prompt = create_prompt(resume_string, jd_string)
-    
+
     # Dapatkan response dari AI
     response_string = get_resume_response(prompt, api_key)
 
     # Pisahkan response menjadi "optimized resume" dan "additional suggestions"
     response_list = response_string.split("## Additional Suggestions")
     new_resume = response_list[0].strip()
+    new_resume = re.sub(r'^\* ', '- ', new_resume, flags=re.MULTILINE)
     suggestions = "## Additional Suggestions\n\n" + response_list[1].strip() if len(response_list) > 1 else ""
 
-    # 1) Hapus heading yang mengandung kata "resume" / "optimized"
+    # ===== Hapus heading yang mengandung kata resume/optimized =====
+    new_resume = new_resume.replace("# Optimized Resume", "")
+    new_resume = new_resume.replace("## Optimized Resume", "")
+    new_resume = new_resume.replace("Optimized Resume", "")
+    new_resume = new_resume.replace("# Resume", "")
+    new_resume = new_resume.replace("## Resume", "")
+    new_resume = re.sub(r'^#+\s*Resume\s*', '', new_resume, flags=re.MULTILINE)
     new_resume = remove_unwanted_headings(new_resume)
-
-    # 2) Rapikan bullet Work Experience jika AI menaruh ' - ' dalam satu baris
-    new_resume = fix_work_experience_bullets(new_resume)
+    # ===============================================================
 
     # Simpan resume asli (Markdown) jika diperlukan
     original_resume_path = "resumes/original_resume.md"
@@ -158,12 +156,12 @@ def process_resume(resume, jd_string):
 
 def export_resume(new_resume):
     """
-    Meng-export resume hasil optimasi (Markdown) menjadi PDF menggunakan WeasyPrint.
-    Pastikan path 'style.css' sesuai lokasi file style.css Anda.
+    Meng-export resume hasil optimasi (Markdown) menjadi PDF
+    menggunakan WeasyPrint.
     """
     try:
         # Konversi Markdown ke HTML
-        html_content = markdown.markdown(new_resume)
+        html_content = markdown.markdown(new_resume, extensions=['extra', 'nl2br'])
 
         # Path output PDF
         output_pdf_file = "resumes/optimized_resume.pdf"
@@ -171,12 +169,12 @@ def export_resume(new_resume):
         # Gunakan stylesheet (pastikan path style.css benar)
         HTML(string=html_content).write_pdf(
             output_pdf_file,
-            stylesheets=["resumes/style.css"]  # Jika style.css ada di folder yang sama dengan app.py
+            stylesheets=["resumes/style.css"]  # atau "resumes/style.css" jika style.css di folder "resumes"
         )
 
         return output_pdf_file
     except Exception as e:
-        return f"Failed to export resume: {str(e)}"
+        return f"Failed to export resume: {str(e)} 💔"
 
 # Bangun aplikasi Gradio
 with gr.Blocks() as app: