feat: add upload file

main.py

@@ -149,7 +149,6 @@ def delete_source_route():
 
     return jsonify({"message": f"Folder '{folder_name}' successfully deleted and recreated."})
 
-
 @api_app.post('/predict')
 async def predict(data: Predict):
     global QA
@@ -176,26 +175,45 @@ async def predict(data: Predict):
         raise HTTPException(status_code=400, detail="Prompt Incorrect")
 
 @api_app.post("/save_document/")
-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)
+async def create_upload_file(file: UploadFile):
+    # Get the file size (in bytes)
+    file.file.seek(0, 2)
+    file_size = file.file.tell()
+
+    # move the cursor back to the beginning
+    await file.seek(0)
+
+    if file_size > 2 * 1024 * 1024:
+        # more than 2 MB
+        raise HTTPException(status_code=400, detail="File too large")
+
+    # check the content type (MIME type)
+    content_type = file.content_type
+    if content_type not in ["image/jpeg", "image/png", "image/gif"]:
+        raise HTTPException(status_code=400, detail="Invalid file type")
+
+    # do something with the valid file
+    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):