Update build.py

build.py

@@ -16,7 +16,7 @@ logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
 pinecone_key = os.environ['PINECONE_KEY']
 mongo_key=os.environ["MONGO_KEY"]
 
-def add_file(username, filename):
+def add_file(collection,username, filename):
     document = {
         "brain": username,
         "filename": filename,
@@ -25,7 +25,7 @@ def add_file(username, filename):
     collection.insert_one(document)
 
 
-def delete_file(username, filename):
+def delete_file(collection,username, filename):
     query = {
         "brain": username,
         "filename": filename
@@ -81,7 +81,7 @@ def updateBrain(brainName, files):
 
     for prevfile in newfiles:
         index.delete_ref_doc(prevfile, delete_from_docstore=True)
-        delete_file(brainName,prevfile)
+        delete_file(collection,brainName,prevfile)
 
     uploadedFiles=[]
     j = 1
@@ -90,7 +90,7 @@ def updateBrain(brainName, files):
         try:
             index.insert(doc)
             if doc.doc_id not in uploadedFiles:
-                add_file(brainName,doc.doc_id)
+                add_file(collection,brainName,doc.doc_id)
                 uploadedFiles.append(doc.doc_id)
         except Exception as e:
             if doc.doc_id not in corrupt: