refac

backend/apps/rag/main.py

@@ -1136,6 +1136,7 @@ def store_doc(
 
 class ProcessDocForm(BaseModel):
     file_id: str
+    collection_name: Optional[str] = None
 
 
 @app.post("/process/doc")
@@ -1148,6 +1149,8 @@ def process_doc(
         file_path = file.meta.get("path", f"{UPLOAD_DIR}/{file.filename}")
 
         f = open(file_path, "rb")
+
+        collection_name = form_data.collection_name
         if collection_name == None:
             collection_name = calculate_sha256(f)[:63]
         f.close()

backend/apps/webui/models/files.py

@@ -99,5 +99,14 @@ class FilesTable:
         except:
             return False
 
+    def delete_all_files(self) -> bool:
+        try:
+            query = File.delete()
+            query.execute()  # Remove the rows, return number of rows removed.
+
+            return True
+        except:
+            return False
+
 
 Files = FilesTable(DB)

backend/apps/webui/routers/files.py

@@ -53,6 +53,7 @@ def upload_file(
 
         # replace filename with uuid
         id = str(uuid.uuid4())
+        filename = f"{id}_{filename}"
         file_path = f"{UPLOAD_DIR}/{filename}"
 
         contents = file.file.read()
@@ -143,3 +144,20 @@ async def delete_file_by_id(id: str, user=Depends(get_verified_user)):
             status_code=status.HTTP_401_UNAUTHORIZED,
             detail=ERROR_MESSAGES.NOT_FOUND,
         )
+
+
+############################
+# Delete All Files
+############################
+
+
+@router.delete("/all")
+async def delete_all_files(user=Depends(get_admin_user)):
+    result = Files.delete_all_files()
+    if result:
+        return {"message": "All files deleted successfully"}
+    else:
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=ERROR_MESSAGES.DEFAULT("Error deleting files"),
+        )