Update app.py

app.py

@@ -11,6 +11,8 @@ import requests
 import time
 import re
 import textract
+import zipfile
+
 
 from datetime import datetime
 from openai import ChatCompletion
@@ -276,7 +278,31 @@ def process_user_input(user_question):
         create_file(filename, user_question, message.content)
         
         #st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-        
+
+def create_zip_of_all_files():
+    # Get all files in the current directory
+    all_files = glob.glob("*.*")
+    
+    # Name the zip file based on the current date and time
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%Y%m%d_%H%M%S")
+    zip_filename = f"all_files_{safe_date_time}.zip"
+    
+    with zipfile.ZipFile(zip_filename, 'w') as zipf:
+        for file in all_files:
+            zipf.write(file)
+    
+    return zip_filename
+
+def get_zip_download_link(zip_filepath):
+    with open(zip_filepath, 'rb') as f:
+        data = f.read()
+    
+    b64 = base64.b64encode(data).decode()
+    file_name = os.path.basename(zip_filepath)
+    href = f'<a href="data:application/zip;base64,{b64}" target="_blank" download="{file_name}">Download All Files</a>'
+    return href
+
     
 def main():
     # Sidebar and global
@@ -295,6 +321,10 @@ def main():
         st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
         filename=None # since transcription is finished next time just use the saved transcript
 
+    if st.sidebar.button("📦 Download All"):
+        zip_file_path = create_zip_of_all_files()
+        st.sidebar.markdown(get_zip_download_link(zip_file_path), unsafe_allow_html=True)
+
     # prompt interfaces
     user_prompt = st.text_area("Enter prompts, instructions & questions:", '', height=100)