Update backup.py

backup.py

@@ -7,6 +7,8 @@ from bs4 import BeautifulSoup
 import hashlib
 import json
 import uuid 
+import glob
+import zipfile
 
 EXCLUDED_FILES = ['app.py', 'requirements.txt', 'pre-requirements.txt', 'packages.txt', 'README.md','.gitattributes', "backup.py","Dockerfile"]
 URLS = {
@@ -27,6 +29,25 @@ if not os.path.exists("history.json"):
     with open("history.json", "w") as f:
         json.dump({}, f)
 
+  
+@st.cache_resource
+def create_zip_of_files(files):
+    zip_name = "all_files.zip"
+    with zipfile.ZipFile(zip_name, 'w') as zipf:
+        for file in files:
+            zipf.write(file)
+    return zip_name
+    
+@st.cache_resource
+def get_zip_download_link(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 download_file(url, local_filename):
     if url.startswith('http://') or url.startswith('https://'):
         try:
@@ -157,8 +178,11 @@ def main():
                 json.dump({}, f)
 
         with open("history.json", "r") as f:
-            history = json.load(f)
-
+            try:
+                history = json.load(f)
+            except:
+                print('error')
+                
         # Handling URL submission
         if url:
             subdir = hashlib.md5(url.encode()).hexdigest()
@@ -179,14 +203,54 @@ def main():
             for subdir in history.values():
                 show_download_links(subdir)
 
+
+        if st.sidebar.button("🗑 Delete All"):
+            # Compose all_files
+            all_files = glob.glob("*.*")
+            all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]  # 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
+                
+            for file in all_files:
+                os.remove(file)
+            st.experimental_rerun()
+            
+        if st.sidebar.button("⬇️ Download All"):
+            # Compose all_files
+            #all_files = glob.glob("*.*")
+            #all_files = [file for file in all_files if len(os.path.splitext(file)[0]) >= 10]  # 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
+            for subdir in history.values(): # go through each download directory
+                #show_download_links(subdir)
+                global file_sequence_numbers
+                for file in list_files(subdir): # go through each file that was downloaded
+                    file_path = os.path.join(subdir, file)
+                    if file_path not in file_sequence_numbers:
+                        file_sequence_numbers[file_path] = 1
+                    else:
+                        file_sequence_numbers[file_path] += 1
+                    sequence_number = file_sequence_numbers[file_path]
+                    if os.path.isfile(file_path):
+                        st.markdown(get_download_link(file_path), unsafe_allow_html=True)
+                        show_file_operations(file_path, sequence_number)
+                    else:
+                        st.write(f"File not found: {file}")
+                zip_file = create_zip_of_files(list_files(subdir))
+                #with open(file, "rb") as f:
+                #    bytes = f.read()
+                #    b64 = base64.b64encode(bytes).decode()
+                #    href = f'<a href="data:file/octet-stream;base64,{b64}" download=\'{os.path.basename(file)}\'>Download: {os.path.basename(file)}</a>'
+                st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
+    
         # Expander for showing URL history and download links
         with st.expander("URL History and Downloaded Files"):
-            for url, subdir in history.items():
-                st.markdown(f"#### {url}")
-                show_download_links(subdir)
-
+            try:
+                for url, subdir in history.items():
+                    st.markdown(f"#### {url}")
+                    show_download_links(subdir)
+            except:
+                print('url history is empty')
         # Update each time to show files we have
         for subdir in history.values():
             show_download_links(subdir)
 if __name__ == "__main__":
-    main()
+    main()