Update app.py

app.py

@@ -18,6 +18,8 @@ from docx2pdf import convert
 import pdfplumber
 import docx
 import numpy as np
+import pypandoc
+
 
 # Create temporary directories
 temp_dir = tempfile.mkdtemp()
@@ -37,7 +39,19 @@ def save_uploaded_file(content):
             f.write(str(content))
     return file_path
 
-
+def save_docx_as_pdf0(input_path, output_path='output.pdf'):
+    if input_path.lower().endswith('.docx'):
+        try:
+            # Convert .docx to .pdf using pypandoc
+            pypandoc.convert_file(input_path, 'pdf', outputfile=output_path)
+            if not os.path.exists(output_path):
+                raise FileNotFoundError("Conversion failed; output PDF not found.")
+        except Exception as e:
+            st.error(f"Failed to convert DOCX to PDF: {str(e)}")
+    elif input_path.lower().endswith('.pdf'):
+        shutil.copy(input_path, output_path)
+    else:
+        raise ValueError("Unsupported file format. Please upload a .docx or .pdf file.")
 
 # def save_uploaded_file(uploaded_file):
 #     file_path = os.path.join("/tmp", uploaded_file.name)
@@ -579,8 +593,8 @@ if uploaded_resume and uploaded_job_description:
                 # Convert the generated .docx to a .pdf
                 # gen_pdf_path = '/tmp/tailored_resume.pdf'
                 gen_pdf_path = save_uploaded_file(gen_docx_path)
-                # save_docx_as_pdf6(gen_docx_path, gen_pdf_path)
-                save_bytes_as_pdf(gen_docx_path, gen_pdf_path)
+                save_docx_as_pdf0(gen_docx_path, gen_pdf_path)
+                # save_bytes_as_pdf(gen_docx_path, gen_pdf_path)
                 # save_docx_as_pdf2(gen_docx_path, gen_pdf_path)
                 
                 # Display uploaded and generated resumes side-by-side