Update app.py

app.py

@@ -97,6 +97,22 @@ def upload_documents(file: UploadFile = File(...)):
     # Get the list of documents from the JSON data
     new_documents = data["texts"]
 
+    # Add the new documents to the documents list
+    documents.extend(new_documents)
+
+    return {"message": f"{len(new_documents)} new documents uploaded"}
+
+@app.post("/admin/documents/embed")
+def embed_documents(file: UploadFile = File(...)):
+    # Read the contents of the uploaded file
+    contents = file.file.read()
+
+    # Load the JSON data from the file contents
+    data = json.loads(contents)
+
+    # Get the list of documents from the JSON data
+    new_documents = data["texts"]
+
     # Encode the new documents and add them to the FAISS database
     new_embeddings = model.encode(new_documents)
     index.add(np.array(new_embeddings))
@@ -141,13 +157,14 @@ def download_database_0():
 @app.post("/admin/database/upload")
 def upload_database(file: UploadFile = File(...)):
     # Read the contents of the uploaded file
-    contents = json.load(file.file)
+    contents = file.file.read()
 
     # Load the FAISS index from the file contents
-    index = faiss.read_index_binary(contents["index"])
+    global index
+    index = faiss.read_index_binary(contents)
 
-    # Load the documents list from the file contents
-    documents.clear()
-    documents.extend(contents["documents"])
+    # Clear the existing documents list and add the new documents
+    #global documents
+    #documents = index.reconstruct_n(0, index.ntotal).tolist()
 
-    return {"message": "Database uploaded", "documents": documents}
+    return {"message": f"Database uploaded with {len(documents)} documents"}