Update download button section

app.py

@@ -5,7 +5,7 @@ import pandas as pd
 import matplotlib.pyplot as plt
 from resume_generation_gemini_pro import Gemini_pro_main
 from similarity_score_refined import similarity_main
-from io import BytesIO
+
 
 # Helper function to save uploaded files temporarily and return their paths
 def save_uploaded_file(uploaded_file):
@@ -132,21 +132,22 @@ if uploaded_resume and uploaded_job_description:
 
     if st.button("Tailor Resume"):
         with st.spinner("Generating resume..."):
-            generated_resume = Gemini_pro_main(resume_path, job_description_path)
+            generated_resume = generate_gemini(resume_path, job_description_path)
             st.subheader("Generated Tailored Resume:")
             st.write(generated_resume)
 
-            
-            resume_bytes = BytesIO()
-            generated_resume.save(resume_bytes)
-            resume_bytes.seek(0)
-            
-            st.download_button(
-            label="Download Resume",
-            data=resume_bytes,
-            file_name="tailored_resume.docx",
-            mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
-            )
+            if generated_resume:
+                from io import BytesIO
+                resume_bytes = BytesIO()
+                generated_resume.save(resume_bytes)
+                resume_bytes.seek(0)
+                
+                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.")