Update app.py

app.py

@@ -6,13 +6,14 @@ from weasyprint import HTML
 from markitdown import MarkItDown
 from cerebras.cloud.sdk import Cerebras
 
-# Pastikan file style.css tersedia sesuai path (misalnya di folder "resumes" atau di direktori yang sama)
+# 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 dengan semua optional dependencies (pastikan Anda telah menginstal 'markitdown[all]')
-md_converter = MarkItDown(enable_plugins=True)
+# Inisialisasi MarkItDown
+md_converter = MarkItDown()
 
 def create_prompt(resume_string: str, jd_string: str) -> str:
     """
@@ -107,12 +108,16 @@ def process_resume(resume, jd_string):
     Memproses file resume yang di-upload dan job description, lalu
     menghasilkan resume yang telah dioptimasi + saran perbaikan.
     """
-    # Gunakan file-like stream dari upload (tanpa cek ekstensi, dukung semua format yang didukung MarkItDown)
-    try:
-        result = md_converter.convert(resume.file)
+    # 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)
         resume_string = result.text_content  # konten Markdown hasil konversi
-    except Exception as e:
-        return f"Conversion failed: {str(e)}", "", "", "", ""
+    else:
+        return "File format not supported for conversion to Markdown.", "", "", "", ""
     
     # Buat prompt untuk AI
     prompt = create_prompt(resume_string, jd_string)
@@ -123,7 +128,6 @@ 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 ""
 
@@ -156,7 +160,7 @@ def export_resume(new_resume):
     menggunakan WeasyPrint.
     """
     try:
-        # Konversi Markdown ke HTML dengan ekstensi tambahan agar format terjaga
+        # Konversi Markdown ke HTML
         html_content = markdown.markdown(new_resume, extensions=['extra', 'nl2br'])
 
         # Path output PDF
@@ -165,7 +169,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"]
+            stylesheets=["resumes/style.css"]  # atau "resumes/style.css" jika style.css di folder "resumes"
         )
 
         return output_pdf_file
@@ -214,4 +218,4 @@ with gr.Blocks() as app:
         outputs=[export_result]
     )
 
-app.launch()
+app.launch()