Update function to return tailored resume

resume_generation_gemini_pro.py

@@ -135,8 +135,8 @@ Please generate a resume that:
     else:
         return "Failed to generate resume text."
 
-# Main function for Streamlit app
-def Gemini_pro_main(current_resume,job_description):
+#Entry function for the model
+def generate_gemini(current_resume,job_description):
     st.header('Resume Tailoring')
 
     # Load the resume and job description from Google Drive
@@ -155,6 +155,28 @@ def Gemini_pro_main(current_resume,job_description):
         save_resume_to_docx(tailored_resume, file_path)
         st.success(f"Tailored resume saved to {file_path}")
 
+    return tailored_resume
+
+# Main function for Streamlit app
+# def Gemini_pro_main(current_resume,job_description):
+#     st.header('Resume Tailoring')
+
+#     # Load the resume and job description from Google Drive
+#     resume_text = extract_text(current_resume)
+#     job_description = extract_text(job_description)
+
+#     # Tailor resume based on job description
+#     tailored_resume = tailor_resume(resume_text, job_description)
+#     st.write("**Tailored Resume:**")
+#     st.write(tailored_resume)
+#     print(tailored_resume)
+
+#         # Save the tailored resume to a .docx file
+#     if tailored_resume:
+#         file_path = f"Tailored_Resume_{datetime.now().strftime('%Y%m%d_%H%M%S')}.docx"
+#         save_resume_to_docx(tailored_resume, file_path)
+#         st.success(f"Tailored resume saved to {file_path}")
+
 # if __name__ == '__main__':
 #     main()