Update main.py

main.py

@@ -24,7 +24,8 @@ from typing import Optional
 import json
 from function import support_function
 from response import ResponseDefault as res
-
+import os
+os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
 app = FastAPI(
     title="ChatBot HCMUTE",
     description="Python ChatBot is intended for use in the topic Customizing chatbots. With the construction of 2 students Vo Nhu Y - 20133118 and Nguyen Quang Phuc 20133080",
@@ -41,7 +42,8 @@ app = FastAPI(
     }
 )
 origins = [
-    "http://localhost:8000",
+    "https://kltn20133118-pychatbot.hf.space",
+    "https://pychatbot20133.streamlit.app"
 ]
 
 app.add_middleware(
@@ -51,8 +53,7 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
-ALLOWED_EXTENSIONS = {'csv', 'txt', 'doc', 'docx', 'pdf', 'xlsx', 'pptx', 'json', 'md', 'xlsx'}
-
+ALLOWED_EXTENSIONS = {'csv', 'txt', 'doc', 'docx', 'pdf', 'xlsx', 'pptx', 'json', 'md'}
 
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
@@ -62,7 +63,6 @@ if not firebase_admin._apps:
     cred = credentials.Certificate("firebase_certificate.json")
     fred = firebase_admin.initialize_app(cred)
 
-
 class JWTBearer(HTTPBearer):
     def __init__(self, auto_error: bool = True):
         super(JWTBearer, self).__init__(auto_error=auto_error)
@@ -112,7 +112,28 @@ async def override_render_chat(user_id: str,
     if check is not True:
         return check
     request = RequestMySQL.RequestRenderChatHistory(user_id=user_id)
-    return MySQLService.render_chat_history(request)
+    return await MySQLService.render_chat_history(request)
+
+@app.delete("/api/v1/mysql/chat_history/delete_last_chat_record",dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+async def override_delete_last_chat_record(request: RequestMySQL.RequestStopChat, current_user_email: str = Depends(get_current_user_email)):
+    user_id = request.user_id
+    check = support_function.check_value_user_id_controller(user_id)
+    if check is not True:
+        return check
+    chat_name = request.chat_name
+    if chat_name is None or chat_name.strip() == "":
+        raise HTTPException(status_code=400, detail="chat_name field is required.")
+    return await MySQLService.delete_last_chat_detail_by_chat_name(request)
+
+@app.post("/api/v1/mysql/chat_history/create",dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+async def override_create_chat_history(request:RequestMySQL.RequestCreateChatHistory,current_user_email: str = Depends(get_current_user_email)):
+    check = support_function.check_value_user_id(request.user_id, current_user_email)
+    if check is not True:
+        return check
+    chat_name = request.chat_name
+    if chat_name is None or chat_name.strip() == "":
+        raise HTTPException(status_code=400, detail="chat_name field is required.")
+    return await MySQLService.create_chat_history(request)
 
 @app.put("/api/v1/mysql/edit_chat", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
 async def override_edit_chat(request: RequestMySQL.RequestEditNameChat,
@@ -127,7 +148,7 @@ async def override_edit_chat(request: RequestMySQL.RequestEditNameChat,
     name_old = request.name_old
     if name_old is None or name_old.strip() == "":
         raise HTTPException(status_code=400, detail="name_old field is required.")
-    return MySQLService.edit_chat(request)
+    return await MySQLService.edit_chat(request)
 
 @app.delete("/api/v1/mysql/chat_history/delete", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
 async def override_delete_chat(request: RequestMySQL.RequestDeleteChat,
@@ -139,7 +160,7 @@ async def override_delete_chat(request: RequestMySQL.RequestDeleteChat,
     chat_name = request.chat_name
     if chat_name is None or chat_name.strip() == "":
         raise HTTPException(status_code=400, detail="chat_name field is required.")
-    return MySQLService.delete_chat(request)
+    return await MySQLService.delete_chat(request)
 
 @app.delete("/api/v1/mysql/detail_chat/delete", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
 async def override_delete_detail_chat_detail(request: RequestMySQL.RequestDeleteDetailChat,
@@ -151,7 +172,7 @@ async def override_delete_detail_chat_detail(request: RequestMySQL.RequestDelete
     id_chat_detail = request.id_chat_detail
     if id_chat_detail is None or id_chat_detail.strip() == "":
         raise HTTPException(status_code=400, detail="id_chat_detail field is required.")
-    return MySQLService.delete_chat_detail_by_id(request)
+    return await MySQLService.delete_chat_detail_by_id(request)
 
 @app.get("/api/v1/mysql/detail_chat/{user_id}/{chat_id}", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
 async def override_load_chat(chat_id: str, user_id: str,
@@ -172,7 +193,7 @@ async def override_load_chat(chat_id: str, user_id: str,
         return res.ReponseError(status=400,
                                 data=res.Message(message="chat_id must be greater than 0"))
     request = RequestMySQL.RequestLoadChatHistory(chat_id=chat_id, user_id=user_id)
-    return MySQLService.load_chat_history(request)
+    return await MySQLService.load_chat_history(request)
 
 
 @app.get("/api/v1/default/info_user/{user_id}", dependencies=[Depends(JWTBearer())], tags=["Default"])
@@ -181,10 +202,10 @@ async def override_get_user(user_id: str , current_user_email: str = Depends(get
     if check is not True:
         return check
     request = RequestDefault.RequestInfoUser(user_id=user_id)
-    return DefaultService.info_user(request)
+    return await DefaultService.info_user(request)
 
 
-@app.post("/api/v1/users/update_user_info", dependencies=[Depends(JWTBearer())], tags=["User"])
+@app.put("/api/v1/users/update_user_info", dependencies=[Depends(JWTBearer())], tags=["User"])
 async def override_update_user_info(request: RequestUser.RequestUpdateUserInfo,
                                     current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
@@ -206,8 +227,7 @@ async def override_update_user_info(request: RequestUser.RequestUpdateUserInfo,
     if photo_url is None or photo_url.strip() == "":
         return res.ReponseError(status=400,
                                 data=res.Message(message="photo_url field is required."))
-    return UserService.update_user_info(request)
-
+    return await UserService.update_user_info(request)
 
 @app.put('/api/v1/users/change_password', dependencies=[Depends(JWTBearer())], tags=["User"])
 async def override_reset_password_firebase(request: RequestUser.RequestChangePassword,
@@ -230,7 +250,7 @@ async def override_reset_password_firebase(request: RequestUser.RequestChangePas
         return res.ReponseError(status=400,
                                 data=res.Message(message="confirm_new_password field is required."))
 
-    return UserService.change_password(request)
+    return await UserService.change_password(request)
 
 @app.delete("/api/v1/file/delete", dependencies=[Depends(JWTBearer())], tags=["File"])
 async def override_delete_folder(request: RequestFile.RequestDeleteAllFile,
@@ -238,7 +258,7 @@ async def override_delete_folder(request: RequestFile.RequestDeleteAllFile,
     check = support_function.check_value_user_id(request.user_id, current_user_email)
     if check != True:
         return check
-    return FileService.deleteAllFile(request)
+    return await FileService.deleteAllFile(request)
 
 @app.delete("/api/v1/file/delete_file", dependencies=[Depends(JWTBearer())], tags=["File"])
 async def override_delete_one_file(request: RequestFile.RequestDeleteFile,
@@ -251,8 +271,7 @@ async def override_delete_one_file(request: RequestFile.RequestDeleteFile,
     if name_file is None or name_file.strip() == "":
         return res.ReponseError(status=400,
                                 data=res.Message(message="name_file is required."))
-    return FileService.deleteFile(request)
-
+    return await FileService.deleteFile(request)
 
 @app.post("/api/v1/file/chatbot/download_folder", dependencies=[Depends(JWTBearer())], tags=["File"])
 async def override_download_folder_from_dropbox(request: RequestFile.RequestDownLoadFolder,
@@ -261,8 +280,7 @@ async def override_download_folder_from_dropbox(request: RequestFile.RequestDown
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
-    return FileService.download_folder(request)
-
+    return await FileService.download_folder(request)
 
 @app.post("/api/v1/file/chatbot/download_files", dependencies=[Depends(JWTBearer())], tags=["File"])
 async def override_download_file_by_id(request: RequestFile.RequestDownLoadFile,
@@ -271,8 +289,7 @@ async def override_download_file_by_id(request: RequestFile.RequestDownLoadFile,
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
-    return FileService.download_file(request)
-
+    return await  FileService.download_file(request)
 
 @app.post("/api/v1/file/upload_files", dependencies=[Depends(JWTBearer())], tags=["File"])
 async def override_upload_files_dropbox(
@@ -283,9 +300,12 @@ async def override_upload_files_dropbox(
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
+    for file in files:
+        if file.size > 15 * 1024 * 1024:
+            raise HTTPException(status_code=413,
+                                detail=f"File {file.filename} too large. Maximum size allowed is 15MB.")
     request = RequestFile.RequestUploadFile(files=files, user_id=user_id)
-    return FileService.upload_files(request)
-
+    return await FileService.upload_files(request)
 
 @app.post("/api/v1/chat/chatbot/query", dependencies=[Depends(JWTBearer())], tags=["Chatbot"])
 async def override_handle_query2_upgrade_old(request: Request, user_id: str = Form(None), text_all: str = Form(...),
@@ -296,8 +316,7 @@ async def override_handle_query2_upgrade_old(request: Request, user_id: str = Fo
         return check
     request = RequestChat.RequestQuery2UpgradeOld(user_id=user_id, text_all=text_all, question=question,
                                                   chat_name=chat_name)
-    return ChatService.query2_upgrade_old(request)
-
+    return await ChatService.query2_upgrade_old(request)
 
 @app.get("/api/v1/chat/chatbot/extract_file/{user_id}", dependencies=[Depends(JWTBearer())], tags=["Chatbot"])
 async def override_extract_file(user_id: str, current_user_email: str = Depends(get_current_user_email)):
@@ -305,8 +324,7 @@ async def override_extract_file(user_id: str, current_user_email: str = Depends(
     if check is not True:
         return check
     request = RequestChat.RequestExtractFile(user_id=user_id)
-    return ChatService.extract_file(request)
-
+    return await ChatService.extract_file(request)
 
 @app.get("/api/v1/chat/chatbot/generate_question/{user_id}", dependencies=[Depends(JWTBearer())], tags=["Chatbot"])
 async def override_generate_question(user_id: str, current_user_email: str = Depends(get_current_user_email)):
@@ -314,17 +332,18 @@ async def override_generate_question(user_id: str, current_user_email: str = Dep
     if check is not True:
         return check
     request = RequestChat.RequestGenerateQuestion(user_id=user_id)
-    return ChatService.generate_question(request)
+    return await ChatService.generate_question(request)
 
-
-@app.post("/api/v1/default/upload_images", dependencies=[Depends(JWTBearer())], tags=["Default"])
+@app.post("/api/v1/default/upload_image", dependencies=[Depends(JWTBearer())], tags=["Default"])
 async def override_upload_image(user_id: str = Form(None), file: UploadFile = File(...),
                                 current_user_email: str = Depends(get_current_user_email)):
+    if file.file.size > 10 * 1024 * 1024:
+        raise HTTPException(status_code=413, detail="File too large")
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     request = RequestDefault.RequestUpLoadImage(user_id=user_id, files=file)
-    return DefaultService.upload_image_service(request)
+    return await DefaultService.upload_image_service(request)
 
 app.include_router(AuthenticationController.router, tags=["Authentication"], prefix="/api/v1/auth")
 app.include_router(MySQLController.router, prefix="/api/v1/mysql")
@@ -338,7 +357,7 @@ routes_to_override = {
     "/api/v1/mysql/detail_chat/{user_id}/{chat_id}": {"GET"},
     "/api/v1/mysql/edit_chat": {"PUT"},
     "/api/v1/mysql/chat_history/delete": {"DELETE"},
-    "/api/v1/users/update_user_info": {"POST"},
+    "/api/v1/users/update_user_info": {"PUT"},
     "/api/v1/users/change_password": {"PUT"},
     "/api/v1/file/delete": {"DELETE"},
     "/api/v1/file/delete_file": {"DELETE"},
@@ -348,9 +367,11 @@ routes_to_override = {
     "/api/v1/chat/chatbot/query": {"POST"},
     "/api/v1/chat/chatbot/extract_file/{user_id}": {"GET"},
     "/api/v1/chat/chatbot/generate_question/{user_id}": {"GET"},
-    "/api/v1/default/upload_images": {"POST"},
+    "/api/v1/default/upload_image": {"POST"},
     "/api/v1/default/info_user/{user_id}": {"GET"},
-    "/api/v1/mysql/detail_chat/delete":{"DELETE"}
+    "/api/v1/mysql/detail_chat/delete":{"DELETE"},
+    "/api/v1/mysql/chat_history/create": {"POST"},
+    "/api/v1/mysql/chat_history/delete_last_chat_record": {"DELETE"}
 }
 
 app.router.routes = [
@@ -367,11 +388,15 @@ app.add_api_route("/api/v1/mysql/detail_chat/{user_id}/{chat_id}", override_load
                   dependencies=[Depends(JWTBearer())], tags=["MySQL"])
 app.add_api_route("/api/v1/mysql/edit_chat", override_edit_chat, methods=["PUT"], dependencies=[Depends(JWTBearer())],
                   tags=["MySQL"])
+app.add_api_route("/api/v1/mysql/chat_history/create", override_create_chat_history, methods=["POST"], dependencies=[Depends(JWTBearer())],
+                  tags=["MySQL"])
 app.add_api_route("/api/v1/mysql/chat_history/delete", override_delete_chat, methods=["DELETE"],
                   dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+app.add_api_route("/api/v1/mysql/chat_history/delete_last_chat_record", override_delete_last_chat_record, methods=["DELETE"],
+                  dependencies=[Depends(JWTBearer())], tags=["MySQL"])
 app.add_api_route("/api/v1/mysql/detail_chat/delete", override_delete_detail_chat_detail, methods=["DELETE"],
                   dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-app.add_api_route("/api/v1/users/update_user_info", override_update_user_info, methods=["POST"],
+app.add_api_route("/api/v1/users/update_user_info", override_update_user_info, methods=["PUT"],
                   dependencies=[Depends(JWTBearer())], tags=["User"])
 app.add_api_route("/api/v1/users/change_password", override_reset_password_firebase, methods=["PUT"],
                   dependencies=[Depends(JWTBearer())], tags=["User"])
@@ -391,7 +416,7 @@ app.add_api_route("/api/v1/chat/chatbot/extract_file/{user_id}", override_extrac
                   dependencies=[Depends(JWTBearer())], tags=["Chat"])
 app.add_api_route("/api/v1/chat/chatbot/generate_question/{user_id}", override_generate_question, methods=["GET"],
                   dependencies=[Depends(JWTBearer())], tags=["Chat"])
-app.add_api_route("/api/v1/default/upload_images", override_upload_image, methods=["POST"],
+app.add_api_route("/api/v1/default/upload_image", override_upload_image, methods=["POST"],
                   dependencies=[Depends(JWTBearer())], tags=["Default"])
 app.add_api_route("/api/v1/default/info_user/{user_id}", override_get_user, methods=["GET"], dependencies=[Depends(JWTBearer())],
                   tags=["Default"])