Update app.py

app.py

@@ -282,9 +282,30 @@ def save_docx_as_pdf1(input_path, output_path='output.pdf'):
 #     display_doc_as_image('uploaded_resume.pdf')
 
 def display_doc_as_image2(pdf_path):
+    # try:
+    #     images = convert_from_path(pdf_path, size=800)
+    #     display(Image(filename=images[0].filename))
+    # except Exception as e:
+    #     st.error(f"Failed to display image: {str(e)}")
     try:
-        images = convert_from_path(pdf_path, size=800)
-        display(Image(filename=images[0].filename))
+        # Convert PDF to images
+        images = []
+        for i in range(1):  # Assuming only one page
+            command = [
+                "pdf2image", "-j", "--png",
+                "-f", str(i+1),
+                "-l", str(i+1),
+                pdf_path
+            ]
+            output = subprocess.check_output(command)
+            
+            # Decode the base64 image data
+            img_data = base64.b64decode(output.split()[0])
+            img = Image.open(BytesIO(img_data))
+            images.append(img)
+        
+        # Display the first page
+        display(images[0])
     except Exception as e:
         st.error(f"Failed to display image: {str(e)}")
 
@@ -464,7 +485,7 @@ if uploaded_resume and uploaded_job_description:
                     display_doc_as_image2('/tmp/uploaded_resume.pdf')
                 with col2:
                     st.markdown("### Tailored Resume:")
-                    display_doc_as_image5(gen_pdf_path)        
+                    display_doc_as_image3(gen_pdf_path)        
                 # from io import BytesIO
                 # doc = Document()