Update app.py

app.py

@@ -10,11 +10,31 @@ from docx import Document
 import subprocess
 import shutil
 
+import tempfile
+from PIL import Image
+import io
+from docx2pdf import convert
+import docx
+import numpy as np
+
+# Create temporary directories
+temp_dir = tempfile.mkdtemp()
+
+
 # Checking poppler installation
 st.write("PDFInfo Path:", os.popen("which pdfinfo").read().strip())
 st.write("LibreOffice Path:", os.popen("which libreoffice").read().strip())
 st.write("System PATH:", os.environ['PATH'])
 
+import subprocess
+
+def check_poppler():
+    try:
+        subprocess.check_call(["poppler-utils", "--version"])
+        print("Poppler is installed and ready to use.")
+    except subprocess.CalledProcessError:
+        print("Poppler is not installed. Please install it using 'apt-get install poppler-utils'.")
+
 
 # Helper function to save uploaded files temporarily and return their paths
 def save_uploaded_file(content):
@@ -177,8 +197,16 @@ def save_file(file_name):
             f.write(uploaded_file)
     return file_path
 
+def save_docx_as_pdf(input_path, output_path='output.pdf'):
+    if input_path.lower().endswith('.docx'):
+        from docx2pdf import convert
+        convert(input_path, output_path)
+    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_docx_as_pdf(doc_content, output_path='output.pdf'):
+def save_docx_as_pdf2(doc_content, output_path='output.pdf'):
     # Save document content as a .docx file
     temp_doc_path = 'temp.docx'
     doc = Document()
@@ -192,7 +220,8 @@ def save_docx_as_pdf(doc_content, output_path='output.pdf'):
 
     # subprocess.run(['libreoffice', '--headless', '--convert-to', 'pdf', temp_doc_path, '--outdir', os.path.dirname(output_path)])
 
-def save_docx_as_pdf(input_path, output_path='output.pdf'):
+    
+def save_docx_as_pdf1(input_path, output_path='output.pdf'):
     if input_path.lower().endswith('.docx'):
         from docx2pdf import convert
         convert(input_path, output_path)
@@ -205,25 +234,40 @@ def save_docx_as_pdf(input_path, output_path='output.pdf'):
 #     save_docx_as_pdf(resume_path, 'uploaded_resume.pdf')
 #     display_doc_as_image('uploaded_resume.pdf')
 
-def display_doc_as_image(pdf_path):
+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)}")
 
+def display_doc_as_image(pdf_path):
+    try:
+        images = convert_from_path(pdf_path, size=800)
+        display(Image.fromarray(images[0]))
+    except Exception as e:
+        st.error(f"Failed to display image: {str(e)}")
+
 def display_doc_as_image1(pdf_path):
-    # poppler_path = 'usr/bin'
-    # images = convert_from_path(pdf_path, poppler_path=poppler_path)
-    # for img in images:
-    #     buf = BytesIO()
-    #     img.save(buf, format="PNG")
-    #     st.image(buf)
+    try:
+        img = Image.open(pdf_path)
+        st.image(img)
+    except Exception as e:
+        st.error(f"Failed to display image: {str(e)}")
+
 
-    from IPython.display import display, Image
+# def display_doc_as_image1(pdf_path):
+#     # poppler_path = 'usr/bin'
+#     # images = convert_from_path(pdf_path, poppler_path=poppler_path)
+#     # for img in images:
+#     #     buf = BytesIO()
+#     #     img.save(buf, format="PNG")
+#     #     st.image(buf)
 
-    images = convert_from_bytes(open(pdf_path, 'rb').read(), size=800)
-    display(images[0])
+#     from IPython.display import display, Image
+
+#     images = convert_from_bytes(open(pdf_path, 'rb').read(), size=800)
+#     display(images[0])
 
 # Process if files are uploaded
 if uploaded_resume and uploaded_job_description:
@@ -286,34 +330,70 @@ if uploaded_resume and uploaded_job_description:
             if generated_resume is not None:
                          
                 from io import BytesIO
-                 
-                
-                doc = Document()
-                doc.add_paragraph(generated_resume)
-                
-                resume_bytes = BytesIO()
-                doc.save(resume_bytes)
-                resume_bytes.seek(0)
-
-                gen_resume_path = save_uploaded_file(resume_bytes)
-                # uploaded_resume_path = save_uploaded_file(resume)
-
-                col1, col2 = st.columns(2)
-                with col1:
-                    save_docx_as_pdf(resume_path, 'uploaded_resume.pdf')
-                    display_doc_as_image('uploaded_resume.pdf')
-                with col2:
-                    st.markdown("### Tailored Resume:")
-                    save_docx_as_pdf(gen_resume_path, 'tailored_resume.pdf')
-                    display_doc_as_image('tailored_resume.pdf')
-    
+
+                with tempfile.NamedTemporaryFile(suffix='.docx') as temp_doc:
+                    doc = Document()
+                    doc.add_paragraph(generated_resume)
+                    doc.save(temp_doc.name)
                     
-                st.download_button(
-                    label="Download Resume",
+                    # Convert DOCX to PDF
+                    pdf_path = f"temp_{os.path.basename(temp_doc.name)}.pdf"
+                    convert(temp_doc.name, pdf_path)
+                    
+                    # Display resumes side by side
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        st.write("Uploaded Resume:")
+                        display_doc_as_image(resume_path)
+                    with col2:
+                        st.write("Generated Resume:")
+                        display_doc_as_image(pdf_path)
+                        
+                    # Allow users to download both PDFs
+                    st.download_button(
+                        label="Download Uploaded Resume",
+                        data=resume_bytes,
+                        file_name="uploaded_resume.pdf",
+                        mime="application/pdf"
+                    )
+                    st.download_button(
+                        label="Download Generated Resume",
+                        data=open(pdf_path, 'rb').read(),
+                        file_name="generated_resume.pdf",
+                        mime="application/pdf"
+                    )
+                    st.download_button(
+                    label="Generated Resume (Word)",
                     data=resume_bytes,
                     file_name="tailored_resume.docx",
                     mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                )
+                    )
+                # doc = Document()
+                # doc.add_paragraph(generated_resume)
+                
+                # resume_bytes = BytesIO()
+                # doc.save(resume_bytes)
+                # resume_bytes.seek(0)
+
+                # gen_resume_path = save_uploaded_file(resume_bytes)
+                # # uploaded_resume_path = save_uploaded_file(resume)
+
+                # col1, col2 = st.columns(2)
+                # with col1:
+                #     save_docx_as_pdf(resume_path, 'uploaded_resume.pdf')
+                #     display_doc_as_image('uploaded_resume.pdf')
+                # with col2:
+                #     st.markdown("### Tailored Resume:")
+                #     save_docx_as_pdf(gen_resume_path, 'tailored_resume.pdf')
+                #     display_doc_as_image('tailored_resume.pdf')
+    
+                    
+                # st.download_button(
+                #     label="Download Resume",
+                #     data=resume_bytes,
+                #     file_name="tailored_resume.docx",
+                #     mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+                # )
                 
 else:
     st.warning("Please upload both the resume and job description files.")