Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, UploadFile, File
+from fastapi import FastAPI, UploadFile, File, FileResponse
 from fastapi.middleware.cors import CORSMiddleware
 # Loading
 import os
@@ -132,6 +132,19 @@ def search(prompt: str):
     for hit in hits:
       print(hit.payload, "score:", hit.score)
     return hits
+
+@app.get("/download-database/")
+async def download_database():
+    # Path to the database directory
+    database_dir = join(os.getcwd(), 'database')
+    # Path for the zip file
+    zip_path = join(os.getcwd(), 'database.zip')
+    
+    # Create a zip file of the database directory
+    shutil.make_archive(zip_path.replace('.zip', ''), 'zip', database_dir)
+    
+    # Return the zip file as a response for download
+    return FileResponse(zip_path, media_type='application/zip', filename='database.zip')
     
 @app.get("/")
 def api_home():