Update app.py

app.py

@@ -188,7 +188,12 @@ def save_file(file_name):
 
 import tempfile
 
-
+# Function to save a file from BytesIO to a temporary file
+def save_bytes_to_tempfile(bytes_data, suffix):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
+        temp_file.write(bytes_data)
+        return temp_file.name
+        
 def save_bytes_as_pdf(docx_bytes, output_path='output.pdf'):
     # Create a temporary directory
     with tempfile.TemporaryDirectory() as tmp_dir:
@@ -421,55 +426,83 @@ if uploaded_resume and uploaded_job_description:
             #     display_score(similarity_score, pie_colors)
                 
             if generated_resume is not None:
-                         
-                from io import BytesIO
                 doc = Document()
+                doc.add_paragraph(generated_resume)  # Add the generated content to the document
+                
+                # Save the generated document as a .docx file in memory
+                resume_bytes = BytesIO()
+                doc.save(resume_bytes)
+                resume_bytes.seek(0)
+                
+                # Save the .docx to a temporary file
+                gen_docx_path = save_bytes_to_tempfile(resume_bytes.getvalue(), '.docx')
+                
+                # Convert the generated .docx to a .pdf
+                gen_pdf_path = '/tmp/tailored_resume.pdf'
+                save_docx_as_pdf(gen_docx_path, gen_pdf_path)
+                
+                # Display uploaded and generated resumes side-by-side
+                col1, col2 = st.columns(2)
+                with col1:
+                    st.markdown("### Uploaded Resume:")
+                    save_docx_as_pdf(resume_path, '/tmp/uploaded_resume.pdf')
+                    display_doc_as_image('/tmp/uploaded_resume.pdf')
+                with col2:
+                    st.markdown("### Tailored Resume:")
+                    display_doc_as_image(gen_pdf_path)        
+                # from io import BytesIO
+                # doc = Document()
+                
+                # with tempfile.NamedTemporaryFile(delete=False, suffix=".docx") as temp_doc:
+                #     temp_doc_path = temp_doc.name
+                #     doc.save(temp_doc_path)
+
                 
-                with tempfile.NamedTemporaryFile(delete=False, suffix=".docx") as temp_doc:
-                    temp_doc_path = temp_doc.name
-                    doc.save(temp_doc_path)
+                # Convert the DOCX to PDF and display
+                # pdf_path = '/tmp/generated_resume.pdf'
+                # save_docx_as_pdf(temp_doc_path, pdf_path)
                 
-                # Now pass the path to any function that expects a file path
-                pdf_path = '/tmp/generated_resume.pdf'
-                save_bytes_as_pdf(temp_doc_path, pdf_path)
+                # # Now pass the path to any function that expects a file path
+                # pdf_path = '/tmp/generated_resume.pdf'
+                # save_bytes_as_pdf(temp_doc_path, pdf_path)
                 
-                # Display the generated PDF or handle further processing
-                display_doc_as_image(pdf_path)
+                # # Display the generated PDF or handle further processing
+                # display_doc_as_image(pdf_path)
                 
-                with tempfile.NamedTemporaryFile(suffix='.docx') as temp_doc:
-                    doc = Document()
-                    doc.add_paragraph(generated_resume)
-                    # doc.save(temp_doc.name)
-
-                    # resume_bytes = BytesIO()
-                    # doc.save(resume_bytes)
-                    # resume_bytes.seek(0)
+                # with tempfile.NamedTemporaryFile(suffix='.docx') as temp_doc:
+                #     doc = Document()
+                #     doc.add_paragraph(generated_resume)
+                #     # doc.save(temp_doc.name)
+
+                #     # resume_bytes = BytesIO()
+                #     # doc.save(resume_bytes)
+                #     # resume_bytes.seek(0)
                         
-                    # Convert DOCX to PDF
-                    # pdf_path = f"temp_{os.path.basename(temp_doc.name)}.pdf"
-                    # convert(temp_doc.name, pdf_path)
-                    # save_docx_as_pdf(temp_doc.name, pdf_path)
-                    pdf_path = '/tmp/generated_resume.docx'  
-                    doc.save(pdf_path)
+                #     # Convert DOCX to PDF
+                #     # pdf_path = f"temp_{os.path.basename(temp_doc.name)}.pdf"
+                #     # convert(temp_doc.name, pdf_path)
+                #     # save_docx_as_pdf(temp_doc.name, pdf_path)
+                #     pdf_path = '/tmp/generated_resume.docx'  
+                #     doc.save(pdf_path)
                     
-                    # pdf_path = save_uploaded_file(resume_bytes)
-                    save_docx_as_pdf(generated_resume, pdf_path)
+                #     # pdf_path = save_uploaded_file(resume_bytes)
+                #     save_docx_as_pdf(generated_resume, pdf_path)
                     
-                    # Display resumes side by side
-                    col1, col2 = st.columns(2)
-                    with col1:
-                        st.write("Uploaded Resume:")
-                        if resume_path:
-                            display_doc_as_image(resume_path)
-                        else:
-                            st.warning("No resume file found")
+                #     # Display resumes side by side
+                #     col1, col2 = st.columns(2)
+                #     with col1:
+                #         st.write("Uploaded Resume:")
+                #         if resume_path:
+                #             display_doc_as_image(resume_path)
+                #         else:
+                #             st.warning("No resume file found")
                             
-                    with col2:
-                        st.write("Generated Resume:")
-                        if pdf_path:
-                            display_doc_as_image(pdf_path)
-                        else:
-                            st.warning("No generated resume file found")
+                #     with col2:
+                #         st.write("Generated Resume:")
+                #         if pdf_path:
+                #             display_doc_as_image(pdf_path)
+                #         else:
+                #             st.warning("No generated resume file found")
                         
                     # Allow users to download both PDFs
                     # st.download_button(