Update app.py

app.py

@@ -16,6 +16,7 @@ from io import BytesIO
 from PyPDF2 import PdfReader
 import hashlib
 import os
+import shutil
 
 # File Imports
 from embedding import get_embeddings, get_image_embeddings, get_embed_chroma , imporve_text # Ensure this file/module is available
@@ -24,10 +25,21 @@ from search import *
 
 
 # Chroma Connections
-client = chromadb.PersistentClient(path="https://huggingface.co/spaces/Prathmesh48/Test_E5/tree/main/embeddings")
+client = chromadb.PersistentClient(path="embeddings")
 collection = client.get_or_create_collection(name="data", metadata={"hnsw:space": "l2"})
 
 
+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/embedding'
+zip_name = 'embedding'
+
+# st.title("Download Embedding Folder")
+
+
 def generate_hash(content):
     return hashlib.sha256(content.encode('utf-8')).hexdigest()
 
@@ -243,6 +255,15 @@ with st.sidebar:
 
     top_similar_count = st.number_input("Top Similarities to be Displayed", value=3, min_value=1, step=1, format="%i")
 
+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"
+        )
 if st.button('Check for Infringement'):
     global log_output  # Placeholder for log output