Update app.py

app.py

@@ -6,9 +6,6 @@ from weasyprint import HTML
 from markitdown import MarkItDown
 from cerebras.cloud.sdk import Cerebras
 
-# 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")
 
@@ -91,47 +88,33 @@ def get_resume_response(prompt: str, api_key: str, model: str = "llama-3.3-70b",
     return response_string
 
 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
-    ### Optimized Resume
-    dsb.
-    """
     pattern = r'^#+.*\b(?:[Rr]esume|[Oo]ptimized)\b.*$'
     return re.sub(pattern, '', markdown_text, flags=re.MULTILINE)
 
-def process_resume(resume, jd_string):
+def fix_jobsdesk_bullets(text: str) -> str:
     """
-    Memproses file resume yang di-upload dan job description, lalu
-    menghasilkan resume yang telah dioptimasi + saran perbaikan.
+    Mengubah semua baris yang dimulai dengan tanda '-' (opsional dengan spasi)
+    menjadi format bullet list Markdown standar.
     """
-    # Format yang didukung
+    return re.sub(r'^\s*-\s+', '- ', text, flags=re.MULTILINE)
+
+def process_resume(resume, jd_string):
+    # Cek ekstensi dan konversi dokumen menggunakan MarkItDown
     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 menggunakan MarkItDown
         result = md_converter.convert(resume.name)
-        resume_string = result.text_content  # konten Markdown hasil konversi
+        resume_string = result.text_content
     else:
         return "File format not supported for conversion to Markdown.", "", "", "", ""
     
-    # 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 ""
-
-    # ===== 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", "")
@@ -139,48 +122,35 @@ def process_resume(resume, jd_string):
     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)
-    # ===============================================================
-
-    # Simpan resume asli (Markdown) jika diperlukan
+    new_resume = fix_jobsdesk_bullets(new_resume)  # Ubah tanda '-' menjadi bullet list
+    
     original_resume_path = "resumes/original_resume.md"
     with open(original_resume_path, "w", encoding='utf-8') as f:
         f.write(resume_string)
     
-    # Simpan resume hasil optimasi (Markdown)
     optimized_resume_path = "resumes/optimized_resume.md"
     with open(optimized_resume_path, "w", encoding='utf-8') as f:
         f.write(new_resume)
-
-    # Kembalikan output untuk di-render di Gradio
+    
     return resume_string, new_resume, original_resume_path, optimized_resume_path, suggestions
 
 def export_resume(new_resume):
-    """
-    Meng-export resume hasil optimasi (Markdown) menjadi PDF
-    menggunakan WeasyPrint.
-    """
     try:
-        # Konversi Markdown ke HTML
         html_content = markdown.markdown(new_resume, extensions=['extra', 'nl2br'])
-
-        # Path output PDF
         output_pdf_file = "resumes/optimized_resume.pdf"
-
-        # Gunakan stylesheet (pastikan path style.css benar)
         HTML(string=html_content).write_pdf(
             output_pdf_file,
-            stylesheets=["resumes/style.css"]  # atau "resumes/style.css" jika style.css di folder "resumes"
+            stylesheets=["resumes/style.css"]
         )
-
         return output_pdf_file
     except Exception as e:
         return f"Failed to export resume: {str(e)} 💔"
 
-# Bangun aplikasi Gradio
+# Bagian aplikasi Gradio (sama seperti kode Anda)
 with gr.Blocks() as app:
     gr.Markdown("# Resume Optimizer 📄")
     gr.Markdown("Upload your resume, paste the job description, and get actionable insights!")
-
+    
     with gr.Row():
         resume_input = gr.File(label="Upload Your Resume")    
         jd_input = gr.Textbox(
@@ -189,33 +159,31 @@ with gr.Blocks() as app:
             interactive=True,
             placeholder="Paste job description..."
         )
-
+    
     run_button = gr.Button("Optimize Resume 🤖")
-
+    
     with gr.Row():
         before_md = gr.Markdown(label="Original Resume (Before)")
         after_md = gr.Markdown(label="Optimized Resume (After)")
         output_suggestions = gr.Markdown(label="Suggestions")
-
+    
     with gr.Row():
         download_before = gr.File(label="Download Original Resume")
         download_after = gr.File(label="Download Optimized Resume")
         
     export_button = gr.Button("Export Optimized Resume as PDF 🚀")
     export_result = gr.File(label="Download PDF")
-
-    # Saat tombol Optimize Resume diklik
+    
     run_button.click(
         process_resume,
         inputs=[resume_input, jd_input],
         outputs=[before_md, after_md, download_before, download_after, output_suggestions]
     )
-
-    # Saat tombol Export PDF diklik
+    
     export_button.click(
         export_resume,
         inputs=[after_md],
         outputs=[export_result]
     )
-
-app.launch()
+    
+app.launch()