feat: add dele folder

main.py

@@ -4,6 +4,7 @@ from fastapi.staticfiles import StaticFiles
 
 from pydantic import BaseModel
 import os
+import glob
 import shutil
 import subprocess
 
@@ -74,14 +75,16 @@ QA = RetrievalQA.from_chain_type(
 class Predict(BaseModel):
     prompt: str
 
+class Delete(BaseModel):
+    filename: str
+
 app = FastAPI(title="homepage-app")
 api_app = FastAPI(title="api app")
 
 app.mount("/api", api_app, name="api")
 app.mount("/", StaticFiles(directory="static",html = True), name="static")
 
-
-@api_app.get("/run_ingest")
+@api_app.get("/training")
 def run_ingest_route():
     global DB
     global RETRIEVER
@@ -133,16 +136,29 @@ def run_ingest_route():
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error occurred: {str(e)}")
 
-@api_app.get("/api/delete_source")
+@api_app.get("/api/delete")
 def delete_source_route():
-    folder_name = "SOURCE_DOCUMENTS"
+    upload_dir = os.path.join(os.getcwd(), PATH_NAME_SOURCE_DIRECTORY)
+    files = glob.glob(os.path.join(upload_dir, '*'))
 
-    if os.path.exists(folder_name):
-        shutil.rmtree(folder_name)
+    return {"directory": upload_dir, "files": files}
 
-    os.makedirs(folder_name)
+@api_app.delete("/api/delete")
+def delete_source_route(data: Delete):
+    filename = data.filename
+    path_source_documents = os.path.join(os.getcwd(), PATH_NAME_SOURCE_DIRECTORY)
+    file_to_delete = f"{path_source_documents}/${filename}"
 
-    return {"message": f"Folder '{folder_name}' successfully deleted and recreated."}
+    if os.path.exists(file_to_delete):
+        try:
+            os.remove(file_to_delete)
+            print(f"{file_to_delete} has been deleted.")
+
+            return {"message": f"{file_to_delete} has been deleted."}
+        except OSError as e:
+            raise HTTPException(status_code=400, detail=print(f"error: {e}."))
+    else:
+         raise HTTPException(status_code=400, detail=print(f"The file {file_to_delete} does not exist."))
 
 @api_app.post('/predict')
 async def predict(data: Predict):
@@ -208,26 +224,6 @@ async def create_upload_file(file: UploadFile):
         shutil.copyfileobj(file.file, buffer)
 
     return {"filename": file.filename}
-# async def create_upload_file(file: Union[UploadFile, None] = None):
-#     try:
-#         if not file:
-#             raise HTTPException(status_code=400, detail="No upload file sent")
-#         else:
-#             if file.filename == "":
-#                 raise HTTPException(status_code=400, detail="No selected file")
-#             if file:
-#                 filename = file.filename
-#                 folder_path = "SOURCE_DOCUMENTS"
-
-#                 if not os.path.exists(folder_path):
-#                     os.makedirs(folder_path)
-
-#                 file_path = os.path.join(folder_path, filename)
-#                 file.save(file_path)
-
-#                 return {"response": "File saved successfully"}
-#     except Exception as e:
-#         raise HTTPException(status_code=400, detail=e)
 
 @api_app.websocket("/ws")
 async def websocket_endpoint(websocket: WebSocket):