Dwnld test

app.py

@@ -19,6 +19,7 @@ import os
 from plotly.subplots import make_subplots
 import plotly.graph_objects as go
 from PIL import Image
+import shutil
 
 
 # File Imports
@@ -261,9 +262,9 @@ if st.button('Check for Infringement'):
     with tab1:
         with st.spinner('Processing...'):
 
-            for path in os.listdir('/home/user/app'):
+            for path in os.listdir('/home/user/app/embeddings'):
                 print(path)
-            if not os.path.exists('/home/user/app/embeddings'):
+            if os.path.exists('/home/user/app/embeddings'):
                 download_db()
                 print("\u2713 Downloaded Database\n\n")
 
@@ -352,4 +353,22 @@ if st.button('Check for Infringement'):
                 # Call the fragment
                 image_viewer()
 
+def zip_folder(folder_path, zip_name):
+    # Create a zip file from the folder
+    shutil.make_archive(zip_name, 'zip', folder_path)
+    return zip_name + '.zip'
+
+folder_path = '/home/user/app/embeddings'
+zip_name = 'embedding'
+
+if st.button("Download"):
+    zip_file = zip_folder(folder_path, zip_name)
+    with open(zip_file, "rb") as f:
+        st.download_button(
+            label="Download ZIP",
+            data=f,
+            file_name=zip_file,
+            mime="application/zip"
+        )
+