Update app.py

app.py

@@ -85,7 +85,6 @@ def download_documents():
 
     return response
 
-##### TESTING ######
 @app.post("/admin/documents/upload")
 def upload_documents(file: UploadFile = File(...)):
     # Read the contents of the uploaded file
@@ -137,38 +136,27 @@ def download_database():
     return response
 
 
-def download_database_1():
-    # Save the FAISS index to a file
-    faiss.write_index(index, "database.index")
-
-    # Create a response with the index file as the content
-    response = FileResponse("database.index")
-
-    # Set the content disposition header to trigger a download
-    response.headers["Content-Disposition"] = "attachment; filename=database.index"
-
-    return response
+@app.post("/admin/database/upload")
+def upload_database(file: UploadFile = File(...)):
+    # Open the uploaded file as a binary file object
+    with open(file.filename, "wb") as f:
+        f.write(file.file.read())
 
+    # Open the file as a binary file object
+    with open(file.filename, "rb") as f:
+        # Load the FAISS index from the file object
+        global index
+        index = faiss.read_index_binary(f)
 
-def download_database_0():
-    # Save the FAISS index and documents list to a single file
-    data = {
-        "index": faiss.write_index_binary(index),
-        "documents": documents
-    }
-    with open("database.json", "w") as f:
-        json.dump(data, f)
-
-    # Create a response with the database file as the content
-    response = FileResponse("database.json")
+    # Clear the existing documents list and add the new documents
+    global documents
+    documents = index.reconstruct_n(0, index.ntotal).tolist()
 
-    # Set the content disposition header to trigger a download
-    response.headers["Content-Disposition"] = "attachment; filename=database.json"
+    return {"message": f"Database uploaded with {len(documents)} documents"}
 
-    return response
 
 @app.post("/admin/database/upload")
-def upload_database(file: UploadFile = File(...)):
+def upload_database_0(file: UploadFile = File(...)):
     # Read the contents of the uploaded file
     contents = file.file.read()