added resume download

resume_screener.py

@@ -6,7 +6,9 @@ from uuid import uuid4
 load_dotenv()
 
 def ResumeScreener():
-    
+
+    st.session_state.CURRENT_PAGE = st.session_state.RESUME_SCREENER_PAGE
+
     class EmptyText(Exception):
         """
         A custom exception for when an input component
@@ -104,20 +106,28 @@ def ResumeScreener():
                 </style>
                 """, unsafe_allow_html=True)
                 with st.container(
-                        height=600,
+                        height=700,
                         border=False
                         ):
 
-                    # The LLM chain for summarization
-                    if 'SUMMARIZATION_CHAIN' not in st.session_state:
-                        st.session_state.SUMMARIZATION_CHAIN = summarize_chain()
-
                     # The results component
                     match_resumes(
                         job_description=txt,
                         k=docs_num,
                         vector_store=vector_store,
-                        summarization_chain=st.session_state.SUMMARIZATION_CHAIN
+                        summarization_chain=summarize_chain()
+                    )
+
+                # Download all as ZIP
+                if st.session_state.VECTOR_SCORE:
+                    files = pdfs_dict(st.session_state.VECTOR_SCORE)
+                    zip_buffer = create_zip_with_pdfs(files, create_pdf)
+
+                    st.download_button(
+                        label="Download all as ZIP",
+                        data=zip_buffer,
+                        file_name=f"resumes-{uuid4()}.zip",
+                        mime="application/zip"
                     )
 
                     st.session_state.DISPLAY_RESULTS = False