Update app.py

app.py

@@ -608,30 +608,47 @@ if uploaded_resume and uploaded_job_description:
                 # save_docx_as_pdf2(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_image1('/tmp/uploaded_resume.pdf')
-                    display_pdf_pages_as_images(resume_path)
-                with col2:
-                    st.markdown("### Tailored Resume:")
-                    # display_doc_as_image2(gen_pdf_path) 
-                    # display_pdf_pages_as_images(gen_pdf_path)
-                    st.write(generated_resume)
-
-                st.download_button(
-                    label="Generated Resume (PDF)",
-                    data=open(gen_pdf_path, 'rb').read(),
-                    file_name="generated_resume.pdf",
-                    mime="application/pdf"
+                if st.button("View Uploaded Resume"):
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        st.markdown("### Uploaded Resume:")
+                        save_docx_as_pdf(resume_path, '/tmp/uploaded_resume.pdf')
+                        # display_doc_as_image1('/tmp/uploaded_resume.pdf')
+                        display_pdf_pages_as_images(resume_path)
+                    with col2:
+                        st.markdown("### Tailored Resume:")
+                        # display_doc_as_image2(gen_pdf_path) 
+                        # display_pdf_pages_as_images(gen_pdf_path)
+                        st.write(generated_resume)
+    
+                        st.success(f"Download tailored resume")
+                        st.download_button(
+                            label="Generated Resume (PDF)",
+                            data=open(gen_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"
+                        )
+                else:
+                    st.success(f"Download tailored resume")
+                    st.download_button(
+                        label="Generated Resume (PDF)",
+                        data=open(gen_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"
                     )
-                st.download_button(
-                    label="Generated Resume (Word)",
-                    data=resume_bytes,
-                    file_name="tailored_resume.docx",
-                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-                )
+                    
                 # from io import BytesIO
                 # doc = Document()