Update app.py

app.py

@@ -11,6 +11,8 @@ import requests
 import time
 import re
 import textract
+import zipfile  # New import for zipping files
+
 
 from datetime import datetime
 from openai import ChatCompletion
@@ -300,6 +302,28 @@ def divide_prompt(prompt, max_length):
             current_length = len(word)
     chunks.append(' '.join(current_chunk))  # Append the final chunk
     return chunks
+
+def create_zip_of_files(files):
+    """
+    Create a zip file from a list of files.
+    """
+    zip_name = "all_files.zip"
+    with zipfile.ZipFile(zip_name, 'w') as zipf:
+        for file in files:
+            zipf.write(file)
+    return zip_name
+
+
+def get_zip_download_link(zip_file):
+    """
+    Generate a link to download the zip file.
+    """
+    with open(zip_file, 'rb') as f:
+        data = f.read()
+    b64 = base64.b64encode(data).decode()
+    href = f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
+    return href
+
     
 def main():
     openai.api_key = os.getenv('OPENAI_API_KEY')
@@ -382,13 +406,18 @@ def main():
     all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 20]  # exclude files with short names
     all_files.sort(key=lambda x: (os.path.splitext(x)[1], x), reverse=True)  # sort by file type and file name in descending order
 
-    # Sidebar of Files Saving History and surfacing files as context of prompts and responses
     # Added "Delete All" button
     if st.sidebar.button("🗑 Delete All"):
         for file in all_files:
             os.remove(file)
         st.experimental_rerun()
+
+    # Added "Download All" button
+    if st.sidebar.button("⬇️ Download All"):
+        zip_file = create_zip_of_files(all_files)
+        st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
         
+    # Sidebar of Files Saving History and surfacing files as context of prompts and responses
     file_contents=''
     next_action=''
     for file in all_files: