Update app.py

app.py

@@ -6,14 +6,13 @@ 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
+# Pastikan file style.css tersedia sesuai path (misalnya di folder "resumes" atau di direktori yang sama)
 
 # Dapatkan API key dari environment variables
 api_key = os.environ.get("CEREBRAS_API_KEY")
 
-# Inisialisasi MarkItDown
-md_converter = MarkItDown()
+# Inisialisasi MarkItDown dengan semua optional dependencies (pastikan Anda telah menginstal 'markitdown[all]')
+md_converter = MarkItDown(enable_plugins=True)
 
 def create_prompt(resume_string: str, jd_string: str) -> str:
     """
@@ -108,16 +107,12 @@ def process_resume(resume, jd_string):
     Memproses file resume yang di-upload dan job description, lalu
     menghasilkan resume yang telah dioptimasi + saran perbaikan.
     """
-    # 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 menggunakan MarkItDown
-        result = md_converter.convert(resume.name)
+    # Gunakan file-like stream dari upload (tanpa cek ekstensi, dukung semua format yang didukung MarkItDown)
+    try:
+        result = md_converter.convert(resume.file)
         resume_string = result.text_content  # konten Markdown hasil konversi
-    else:
-        return "File format not supported for conversion to Markdown.", "", "", "", ""
+    except Exception as e:
+        return f"Conversion failed: {str(e)}", "", "", "", ""
     
     # Buat prompt untuk AI
     prompt = create_prompt(resume_string, jd_string)
@@ -128,6 +123,7 @@ def process_resume(resume, jd_string):
     # Pisahkan response menjadi "optimized resume" dan "additional suggestions"
     response_list = response_string.split("## Additional Suggestions")
     new_resume = response_list[0].strip()
+    # Ganti tanda asterisk di awal baris menjadi tanda "-" untuk bullet list standar
     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 ""
 
@@ -160,7 +156,7 @@ def export_resume(new_resume):
     menggunakan WeasyPrint.
     """
     try:
-        # Konversi Markdown ke HTML
+        # Konversi Markdown ke HTML dengan ekstensi tambahan agar format terjaga
         html_content = markdown.markdown(new_resume, extensions=['extra', 'nl2br'])
 
         # Path output PDF
@@ -169,7 +165,7 @@ 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"]  # atau "resumes/style.css" jika style.css di folder "resumes"
+            stylesheets=["resumes/style.css"]
         )
 
         return output_pdf_file