Update app.py

app.py

@@ -8,7 +8,7 @@ from similarity_score_refined import similarity_main
 from pdf2image import convert_from_path, convert_from_bytes
 from docx import Document
 import subprocess
-import shutil
+import shutilhttps://huggingface.co/spaces/pktpaulie/resumeMagic
 
 import tempfile
 from PIL import Image
@@ -23,23 +23,6 @@ temp_dir = tempfile.mkdtemp()
 
 import subprocess
 
-def check_poppler_installation():
-    """Check if poppler is installed and accessible"""
-    try:
-        # Try to execute pdftoppm (part of poppler-utils)
-        result = subprocess.run(['pdftoppm', '-v'], 
-                              capture_output=True, 
-                              text=True)
-        st.success("Poppler is installed and accessible")
-        return True
-    except FileNotFoundError:
-        st.warning("Poppler not found in PATH. Using fallback display methods.")
-        return False
-
-# Call this at startup
-check_poppler_installation()
-
-
 # Helper function to save uploaded files temporarily and return their paths
 def save_uploaded_file(content):
     if hasattr(content, 'name'):  # Check if it's a file-like object
@@ -203,6 +186,28 @@ def save_file(file_name):
             f.write(uploaded_file)
     return file_path
 
+import tempfile
+
+
+def save_bytes_as_pdf(docx_bytes, output_path='output.pdf'):
+    # Create a temporary directory
+    with tempfile.TemporaryDirectory() as tmp_dir:
+        # Write the DOCX bytes to a temporary file
+        temp_file = os.path.join(tmp_dir, 'temp.docx')
+        with open(temp_file, 'wb') as f:
+            f.write(docx_bytes)
+        
+        # Convert the temporary DOCX to PDF
+        pdf_path = os.path.join(tmp_dir, 'output.pdf')
+        convert(temp_file, pdf_path)
+        
+        # Copy the PDF to the desired output location
+        with open(output_path, 'wb') as f:
+            with open(pdf_path, 'rb') as src_f:
+                f.write(src_f.read())
+
+    # Clean up the temporary directory
+    os.remove(output_path)
 
 def save_docx_as_pdf(input_path, output_path='output.pdf'):
     if input_path.lower().endswith('.docx'):
@@ -426,7 +431,7 @@ if uploaded_resume and uploaded_job_description:
                 
                 # Now pass the path to any function that expects a file path
                 pdf_path = '/tmp/generated_resume.pdf'
-                save_docx_as_pdf(temp_doc_path, pdf_path)
+                save_bytes_as_pdf(temp_doc_path, pdf_path)
                 
                 # Display the generated PDF or handle further processing
                 display_doc_as_image(pdf_output_path)