Update app.py

app.py

@@ -9,6 +9,7 @@ from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
 import json
+import io
 
 app = FastAPI()
 model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
@@ -64,14 +65,14 @@ def get_database_length():
 
 @app.post("/admin/database/reset")
 def reset_database():
-    index = faiss.IndexFlatL2(384)
+    index = faiss.reset()
     documents = []
     return {"message": "Database reset"}
 
 @app.get("/admin/documents/download")
 def download_documents():
     # Convert the documents list to a JSON string
-    documents_json = json.dumps([{text: doc} for doc in documents])
+    documents_json = json.dumps([{"text": doc} for doc in documents])
 
     # Create a response with the JSON string as the content
     response = Response(content=documents_json, media_type="application/json")
@@ -83,27 +84,32 @@ def download_documents():
 
 @app.get("/admin/database/download")
 def download_database():
-    # Save the FAISS index to a file
-    faiss.write_index(index, "database.index")
+    # 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 index file as the content
-    response = FileResponse("database.index")
+    # Create a response with the database file as the content
+    response = FileResponse("database.json")
 
     # Set the content disposition header to trigger a download
-    response.headers["Content-Disposition"] = "attachment; filename=database.index"
+    response.headers["Content-Disposition"] = "attachment; filename=database.json"
 
     return response
 
 @app.post("/admin/database/upload")
 def upload_database(file: UploadFile = File(...)):
     # Read the contents of the uploaded file
-    contents = file.file.read()
+    contents = json.load(file.file)
 
     # Load the FAISS index from the file contents
-    index = faiss.read_index_binary(contents.encode())
+    index = faiss.read_index_binary(contents["index"])
 
-    # Clear the existing documents and add the new ones
+    # Load the documents list from the file contents
     documents.clear()
-    documents.extend(index.reconstruct_n(0, index.ntotal))
+    documents.extend(contents["documents"])
 
-    return {"message": "Database uploaded"}
+    return {"message": "Database uploaded", "documents": documents}