Update app.py

app.py

@@ -24,48 +24,6 @@ import streamlit.components.v1 as components
 # Create temporary directories
 temp_dir = tempfile.mkdtemp()
 
-
-import subprocess
-
-# Helper function to save uploaded files temporarily and return their paths
-def save_uploaded_file(content):
-    if hasattr(content, 'name'):  # Check if it's a file-like object
-        file_path = os.path.join("/tmp", content.name)
-        with open(file_path, "wb") as f:
-            f.write(content.read())
-    else:  # It's a string (the file contents)
-        file_path = os.path.join("/tmp", "temp_upload")
-        with open(file_path, "w") as f:
-            f.write(str(content))
-    return file_path
-
-def save_docx_as_pdf0(input_path, output_path='output.pdf'):
-    if input_path.lower().endswith('.docx'):
-        try:
-            # Convert .docx to .pdf using pypandoc
-            pypandoc.convert_file(input_path, 'pdf', outputfile=output_path)
-            if not os.path.exists(output_path):
-                raise FileNotFoundError("Conversion failed; output PDF not found.")
-        except Exception as e:
-            st.error(f"Failed to convert DOCX to PDF: {str(e)}")
-    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 display_docx_content(file):
-    doc = docx.Document(file)
-    full_text = []
-    for para in doc.paragraphs:
-        full_text.append(para.text)
-    return '\n'.join(full_text)
-    
-# def save_uploaded_file(uploaded_file):
-#     file_path = os.path.join("/tmp", uploaded_file.name)
-#     with open(file_path, "wb") as f:
-#         f.write(uploaded_file.getbuffer())
-#     return file_path
-
 # Custom CSS for styling
 st.markdown("""
     <style>
@@ -162,6 +120,26 @@ st.markdown('</div>', unsafe_allow_html=True)
 st.title("Resume Tailoring with Google Generative AI")
 st.markdown("### Upload your resume and job description to check similarity and generate a tailored resume.")
 
+
+# Helper function to save uploaded files temporarily and return their paths
+def save_uploaded_file(content):
+    if hasattr(content, 'name'):  
+        file_path = os.path.join("/tmp", content.name)
+        with open(file_path, "wb") as f:
+            f.write(content.read())
+    else: 
+        file_path = os.path.join("/tmp", "temp_upload")
+        with open(file_path, "w") as f:
+            f.write(str(content))
+    return file_path
+    
+# def save_uploaded_file(uploaded_file):
+#     file_path = os.path.join("/tmp", uploaded_file.name)
+#     with open(file_path, "wb") as f:
+#         f.write(uploaded_file.getbuffer())
+#     return file_path
+
+
 # Two columns for file uploaders
 col1, col2 = st.columns(2)
 with col1:
@@ -199,19 +177,14 @@ 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
-
-import tempfile
+    
+def display_docx_content(file):
+    doc = docx.Document(file)
+    full_text = []
+    for para in doc.paragraphs:
+        full_text.append(para.text)
+    return '\n'.join(full_text)
+    
 
 # Function to save a file from BytesIO to a temporary file
 def save_bytes_to_tempfile(bytes_data, suffix):
@@ -240,7 +213,7 @@ def save_bytes_as_pdf(docx_bytes, output_path='output.pdf'):
     os.remove(output_path)
 
 
-def display_content_with_page_numbers(content, words_per_page=500):
+def display_content_with_page_numbers(content, words_per_page=290):
     # Split content into words
     words = content.split()
     total_pages = (len(words) // words_per_page) + (1 if len(words) % words_per_page != 0 else 0)
@@ -251,9 +224,10 @@ def display_content_with_page_numbers(content, words_per_page=500):
         end_index = start_index + words_per_page
         page_content = ' '.join(words[start_index:end_index])
         
-        st.markdown(f"#### Page {i + 1}")
-        st.write(page_content)
-        
+        # st.markdown(f"#### Page {i + 1}")
+        # st.write(page_content)
+    st.markdown(f"#### Page {total_pages}")
+    
 def save_docx_as_pdf(input_path, output_path='output.pdf'):
     if input_path.lower().endswith('.docx'):
         try:
@@ -269,28 +243,7 @@ def save_docx_as_pdf(input_path, output_path='output.pdf'):
         raise ValueError("Unsupported file format. Please upload a .docx or .pdf file.")
 
 
-
-# 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_pdf1(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 display_pdf_page1(pdf_path):
+def display_pdf_page(pdf_path):
     try:
         # Open PDF file
         with open(pdf_path, 'rb') as file:
@@ -326,22 +279,6 @@ def display_pdf_pages_as_images(pdf_path):
     except Exception as e:
         st.error(f"Failed to display PDF as image: {str(e)}")
 
-def display_pdf_page(pdf_path, page_number=1):
-    with open(pdf_path, 'rb') as file:
-        reader = PyPDF2.PdfReader(file)
-        page = reader.pages[page_number]
-        x_object = page.extract_text()
-    
-    # Convert text to image (using PIL)
-    img = Image.new('RGB', (800, 1000))
-    draw = ImageDraw.Draw(img)
-    font = ImageFont.truetype("arial.ttf", 20)
-    
-    # Draw text on the image
-    draw.text((10, 10), x_object[:500], fill=(255, 255, 255), font=font)
-    
-    # Display the image
-    display(img)
     
 def display_doc_as_image2(pdf_path):   
     iframe_code = f"""
@@ -365,27 +302,19 @@ if uploaded_resume and uploaded_job_description:
         with st.spinner("Computing Match"):
             similarity_score, pie_colors = get_score(resume_path, job_description_path)
             display_score(similarity_score, pie_colors)
-
-        
+ 
         #Autoscroll
-        st.markdown("""
-            <script>
-                window.scrollTo(0, document.body.scrollHeight);
-            </script>
-        """, unsafe_allow_html=True)
+        components.html("""
+                <script>
+                    window.onload = function() {
+                        window.scrollTo(0, document.body.scrollHeight);
+                    };
+                </script>
+            """)
 
     # Generate Tailored Resume Section
     st.markdown("---")
     # st.subheader("Tailor Resume")
-
-
-    # iframe_code = f"""
-    # <iframe src="{resume_path}" width="100%" height="600px"></iframe>
-    # """
-    # st.markdown(iframe_code, unsafe_allow_html=True)
-
-    # st.components.v1.iframe(resume_path, width=800, height=600)
-
     
     if st.button("Tailor Resume"):
         with st.spinner("Generating resume..."):
@@ -420,18 +349,11 @@ if uploaded_resume and uploaded_job_description:
                 # 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'
                 gen_pdf_path = save_uploaded_file(gen_docx_path)
                 # st.write(display_docx_content(gen_pdf_path))
                 
-                # save_docx_as_pdf0(gen_docx_path, gen_pdf_path)
-                # save_bytes_as_pdf(gen_docx_path, gen_pdf_path)
-                # save_docx_as_pdf2(gen_docx_path, gen_pdf_path)
-                
                 # Display uploaded and generated resumes side-by-side
-                # if st.button("View Uploaded Resume"):
                 col1, col2 = st.columns(2)
                 with col1:
                     st.markdown("### Uploaded Resume:")
@@ -440,36 +362,26 @@ if uploaded_resume and uploaded_job_description:
                 with col2:
                     st.markdown("### Tailored Resume:")
                     # display_pdf_pages_as_images(gen_pdf_path)
+                    display_content_with_page_numbers(generated_resume, 290)
                     st.write(generated_resume)
                     # display_content_with_page_numbers(generated_resume, 290)
+                    
                     st.success(f"Download tailored resume")
-                    st.download_button(
-                        label="Generated Resume (PDF)",
-                        data=open(gen_pdf_path, 'rb').read(),
-                        file_name="tailored_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"
-                #     )
-                            
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        st.download_button(
+                            label="Generated Resume (PDF)",
+                            data=open(gen_pdf_path, 'rb').read(),
+                            file_name="tailored_resume.pdf",
+                            mime="application/pdf"
+                            )
+                    with col2:
+                        st.download_button(
+                            label="Generated Resume (Word)",
+                            data=resume_bytes,
+                            file_name="tailored_resume.docx",
+                            mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+                        )                            
                 
                 
 else: