Update app.py

app.py

@@ -327,27 +327,30 @@ def display_doc_as_image2(pdf_path):
     #     display(Image(filename=images[0].filename))
     # except Exception as e:
     #     st.error(f"Failed to display image: {str(e)}")
-    try:
-        # 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)
+    
+    <iframe src=pdf_path width="100%" height="600px"></iframe>
+    
+    # try:
+    #     # 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)
+    #         # 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)}")
+    #     # Display the first page
+    #     display(images[0])
+    # except Exception as e:
+    #     st.error(f"Failed to display image: {str(e)}")
 
 def display_doc_as_image3(pdf_path):
     try:
@@ -367,12 +370,13 @@ def display_doc_as_image3(pdf_path):
     # except Exception as e:
     #     st.error(f"Failed to display image: {str(e)}")
 
-def display_doc_as_image1(pdf_path):
-    try:
-        img = Image.open(pdf_path)
-        st.image(img)
-    except Exception as e:
-        st.error(f"Failed to display image: {str(e)}")
+def display_doc_as_image1(pdf_url):
+    st.components.v1.iframe(pdf_url, width=800, height=600)
+    # try:
+    #     img = Image.open(pdf_path)
+    #     st.image(img)
+    # except Exception as e:
+    #     st.error(f"Failed to display image: {str(e)}")
 
 
 def display_doc_as_image(file_path):
@@ -533,10 +537,10 @@ if uploaded_resume and uploaded_job_description:
                 with col1:
                     st.markdown("### Uploaded Resume:")
                     save_docx_as_pdf(resume_path, '/tmp/uploaded_resume.pdf')
-                    display_pdf_page1('/tmp/uploaded_resume.pdf')
+                    display_doc_as_image1('/tmp/uploaded_resume.pdf')
                 with col2:
                     st.markdown("### Tailored Resume:")
-                    display_pdf_page(gen_pdf_path)        
+                    display_doc_as_image2(gen_pdf_path)        
                 # from io import BytesIO
                 # doc = Document()