Update app.py

app.py

@@ -153,6 +153,19 @@ def display_score(similarity, colors):
     ax.axis('equal')  
     st.pyplot(fig)
 
+
+def save_file(file_name):
+    if hasattr(uploaded_file, 'name'): 
+        file_path = os.path.join("/tmp", uploaded_file.name)
+        with open(file_path, "wb") as f:
+            f.write(uploaded_file.read())
+    else:  
+        file_path = os.path.join("/tmp", "temp_upload")
+        with open(file_path, "w") as f:
+            f.write(uploaded_file)
+    return file_path
+
+
 def save_docx_as_pdf(doc_content, output_path='output.pdf'):
     # Save document content as a .docx file
     temp_doc_path = 'temp.docx'
@@ -167,11 +180,27 @@ 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'):
+    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.")
+
 # if uploaded_resume.type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
 #     save_docx_as_pdf(resume_path, 'uploaded_resume.pdf')
 #     display_doc_as_image('uploaded_resume.pdf')
 
 def display_doc_as_image(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_image1(pdf_path):
     # poppler_path = 'usr/bin'
     # images = convert_from_path(pdf_path, poppler_path=poppler_path)
     # for img in images:
@@ -225,7 +254,7 @@ if uploaded_resume and uploaded_job_description:
                 </script>
             """, unsafe_allow_html=True)
 
-            col1, col2 = st.columns(2)
+            
             # with col1: 
             #     st.markdown("### Uploaded Resume:")
             #     # if resume_path.endswith('.docx'):
@@ -243,8 +272,11 @@ if uploaded_resume and uploaded_job_description:
             #     display_score(similarity_score, pie_colors)
                 
             if generated_resume is not None:
+                         
                 from io import BytesIO
                 
+                gen_resume_path = save_uploaded_file(generated_resume)
+                # uploaded_resume_path = save_uploaded_file(resume)
                 
                 doc = Document()
                 doc.add_paragraph(generated_resume)
@@ -253,14 +285,14 @@ if uploaded_resume and uploaded_job_description:
                 doc.save(resume_bytes)
                 resume_bytes.seek(0)
 
-
+                col1, col2 = st.columns(2)
                 with col1:
-                    save_docx_as_pdf(resume_bytes, 'tailored_resume.pdf')
+                    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 col2:
-                #     st.markdown("### Tailored Resume:")
-                #     save_docx_as_pdf(generated_resume, 'tailored_resume.pdf')
-                #     display_doc_as_image(generated_resume)
     
                     
                 st.download_button(