Update app.py

app.py

@@ -13,7 +13,8 @@ import io
 
 app = FastAPI()
 model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
-index = faiss.IndexFlatL2(384)  # 384 is the dimensionality of the MiniLM model
+embedding_dimension = 384 # 384 is the dimensionality of the MiniLM model
+index = faiss.IndexFlatL2(embedding_dimension)  
 documents = []
 
 templates = Jinja2Templates(directory=".")
@@ -65,7 +66,8 @@ def get_database_length():
 
 @app.post("/admin/database/reset")
 def reset_database():
-    index = faiss.reset()
+    global index
+    index = faiss.IndexFlatL2(embedding_dimension)
     documents = []
     return {"message": "Database reset"}
 
@@ -82,6 +84,14 @@ def download_documents():
 
     return response
 
+##### TESTING ######
+@app.get("/admin/documents/upload")
+def upload_documents(request: EmbedRequest):
+    documents.extend(request.texts)
+    return {
+        "message": f"{len(new_documents)} new documents uploaded"
+    }
+
 @app.get("/admin/database/download")
 def download_database():
     # Save the FAISS index and documents list to a single file