Update app.py

app.py

@@ -1,21 +1,24 @@
 import os
 import re
-from markitdown import MarkItDown
-from cerebras.cloud.sdk import Cerebras
-from weasyprint import HTML
 import markdown
 import gradio as gr
+from weasyprint import HTML
+from markitdown import MarkItDown
+from cerebras.cloud.sdk import Cerebras
 
-# Ensure you get the API key from environment variables
+# 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")
 
-# Initialize MarkItDown instance
+# Inisialisasi MarkItDown
 md_converter = MarkItDown()
 
-# Functions for resume optimization
 def create_prompt(resume_string: str, jd_string: str) -> str:
     """
-    Creates a detailed prompt for AI-powered resume optimization based on a job description.
+    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.
@@ -66,6 +69,9 @@ You are a professional resume optimization expert specializing in tailoring resu
 """
 
 def get_resume_response(prompt: str, api_key: str, model: str = "llama-3.3-70b", temperature: float = 0.7) -> str:
+    """
+    Mengirim prompt ke model Cerebras (LLM) dan mengembalikan hasil streaming response.
+    """
     client = Cerebras(api_key=api_key)
     stream = client.chat.completions.create(
         messages=[
@@ -84,77 +90,98 @@ def get_resume_response(prompt: str, api_key: str, model: str = "llama-3.3-70b",
         response_string += chunk.choices[0].delta.content or ""
     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):
     """
-    Process the uploaded resume and job description, optimize it, and return the result.
+    Memproses file resume yang di-upload dan job description, lalu
+    menghasilkan resume yang telah dioptimasi + saran perbaikan.
     """
-    # Supported file extensions
+    # Format yang didukung
     supported_extensions = ('.pptx', '.docx', '.pdf', '.jpg', '.jpeg', '.png', '.xlsx')
     
-    # Check if the uploaded resume has a supported extension
+    # Cek apakah file resume memiliki ekstensi yang didukung
     if resume.name.lower().endswith(supported_extensions):
-        # Convert file to Markdown using MarkItDown
+        # Konversi file ke Markdown menggunakan MarkItDown
         result = md_converter.convert(resume.name)
-        resume_string = result.text_content  # Get the converted Markdown content
+        resume_string = result.text_content  # konten Markdown hasil konversi
     else:
-        return "File format not supported for conversion to Markdown.", "", "", "", ""  # Ensure all outputs are returned
+        return "File format not supported for conversion to Markdown.", "", "", "", ""
     
-    # Create optimization prompt
+    # Buat prompt untuk AI
     prompt = create_prompt(resume_string, jd_string)
 
-    # Generate response from AI
+    # Dapatkan response dari AI
     response_string = get_resume_response(prompt, api_key)
 
-    # Split response into optimized resume and suggestions
+    # Pisahkan response menjadi "optimized resume" dan "additional suggestions"
     response_list = response_string.split("## Additional Suggestions")
     new_resume = response_list[0].strip()
     suggestions = "## Additional Suggestions\n\n" + response_list[1].strip() if len(response_list) > 1 else ""
-    
-    new_resume = new_resume.replace("# Optimized Resume", "")
-    new_resume = new_resume.replace("## Optimized Resume", "")
-    new_resume = new_resume.replace("Optimized Resume", "")
-    new_resume = re.sub(r'^\* ', '- ', new_resume, flags=re.MULTILINE)
 
-    # You can use the original resume content to create a file link if needed
+    # ===== Hapus heading yang mengandung kata resume/optimized =====
+    new_resume = remove_unwanted_headings(new_resume)
+    # ===============================================================
+
+    # Simpan resume asli (Markdown) jika diperlukan
     original_resume_path = "resumes/original_resume.md"
-    with open(original_resume_path, "w") as f:
+    with open(original_resume_path, "w", encoding='utf-8') as f:
         f.write(resume_string)
     
-    # Create a download path for the optimized resume
+    # Simpan resume hasil optimasi (Markdown)
     optimized_resume_path = "resumes/optimized_resume.md"
-    with open(optimized_resume_path, "w") as f:
+    with open(optimized_resume_path, "w", encoding='utf-8') as f:
         f.write(new_resume)
 
-    # Return the required outputs (5 values)
+    # Kembalikan output untuk di-render di Gradio
     return resume_string, new_resume, original_resume_path, optimized_resume_path, suggestions
 
 def export_resume(new_resume):
     """
-    Export the optimized resume (in Markdown format) as a PDF file.
+    Meng-export resume hasil optimasi (Markdown) menjadi PDF
+    menggunakan WeasyPrint.
     """
     try:
-        # Convert Markdown to HTML
-        html_content = markdown.markdown(new_resume, extensions=['extra', 'nl2br'])
+        # Konversi Markdown ke HTML
+        html_content = markdown.markdown(new_resume)
 
-        # Convert HTML to PDF and save
+        # Path output PDF
         output_pdf_file = "resumes/optimized_resume.pdf"
-        stylesheets = ['resumes/style.css']  # Ensure this path is correct and accessible
 
-        # Write HTML to PDF with optional styling
-        HTML(string=html_content).write_pdf(output_pdf_file, stylesheets=['resumes/style.css'])
+        # Gunakan stylesheet (pastikan path style.css benar)
+        HTML(string=html_content).write_pdf(
+            output_pdf_file,
+            stylesheets=["style.css"]  # atau "resumes/style.css" jika style.css di folder "resumes"
+        )
 
-        return output_pdf_file  # Return the file path for download
+        return output_pdf_file
     except Exception as e:
         return f"Failed to export resume: {str(e)} 💔"
 
-# Gradio App
+# Bangun aplikasi Gradio
 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(label="Paste the Job Description Here", lines=9, interactive=True, placeholder="Paste job description...")
+        jd_input = gr.Textbox(
+            label="Paste the Job Description Here",
+            lines=9,
+            interactive=True,
+            placeholder="Paste job description..."
+        )
 
     run_button = gr.Button("Optimize Resume 🤖")
 
@@ -170,12 +197,18 @@ with gr.Blocks() as app:
     export_button = gr.Button("Export Optimized Resume as PDF 🚀")
     export_result = gr.File(label="Download PDF")
 
-    # Bindings
+    # 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]
     )
-    export_button.click(export_resume, inputs=[after_md], outputs=[export_result])
+
+    # Saat tombol Export PDF diklik
+    export_button.click(
+        export_resume,
+        inputs=[after_md],
+        outputs=[export_result]
+    )
 
 app.launch()