Update main.py

main.py

@@ -6,16 +6,16 @@ import firebase_admin
 from controller import ChatController
 from firebase_admin import credentials
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
-from service import MySQLService,LoginService,ChatService
-from request import RequestMySQL,RequestLogin,RequestDefault
+from service import MySQLService, LoginService, ChatService
+from request import RequestMySQL, RequestLogin, RequestDefault
 from auth.authentication import decodeJWT
 from repository import UserRepository
 from auth import authentication
 from datetime import datetime, timedelta
 from fastapi import Depends, HTTPException, Form, File, UploadFile
 from typing import List
-from service import FileService,DefaultService,LoginService
-from request import RequestFile,RequestChat,RequestDefault
+from service import FileService, DefaultService, LoginService
+from request import RequestFile, RequestChat, RequestDefault
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.responses import JSONResponse
 from pydantic.error_wrappers import ErrorWrapper
@@ -42,7 +42,6 @@ app = FastAPI(
 )
 origins = [
     "https://kltn20133118-pychatbot.hf.space",
-    "http://localhost:8501"
 ]
 
 app.add_middleware(
@@ -54,13 +53,16 @@ app.add_middleware(
 )
 ALLOWED_EXTENSIONS = {'csv', 'txt', 'doc', 'docx', 'pdf', 'xlsx', 'pptx', 'json', 'md', 'xlsx'}
 
+
 def allowed_file(filename):
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+
 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)
@@ -79,8 +81,9 @@ class JWTBearer(HTTPBearer):
             raise HTTPException(status_code=401, detail="Invalid authorization code.")
 
     def verify_accesstoken(self, jwtoken: str) -> bool:
-       check  = LoginService.check_token_is_valid(jwtoken)
-       return check
+        check = LoginService.check_token_is_valid(jwtoken)
+        return check
+
     def verify_jwt(self, jwtoken: str) -> bool:
         try:
             payload = decodeJWT(jwtoken)
@@ -91,6 +94,7 @@ class JWTBearer(HTTPBearer):
             print(e)
             return False
 
+
 def get_current_user_email(credentials: str = Depends(JWTBearer())):
     try:
         payload = decodeJWT(credentials)
@@ -101,16 +105,20 @@ def get_current_user_email(credentials: str = Depends(JWTBearer())):
         print(e)
         raise HTTPException(status_code=403, detail="Invalid token or expired token.")
 
-@app.get("/api/mysql/render_chat_history",dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-async def override_render_chat(user_id: Optional[str] = Query(None), current_user_email: str = Depends(get_current_user_email)):
+
+@app.get("/api/mysql/render_chat_history", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+async def override_render_chat(user_id: Optional[str] = Query(None),
+                               current_user_email: str = Depends(get_current_user_email)):
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     request = RequestMySQL.RequestRenderChatHistory(user_id=user_id)
     return MySQLService.render_chat_history(request)
 
+
 @app.put("/api/mysql/edit_chat/", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-async def override_edit_chat(request: RequestMySQL.RequestEditNameChat, current_user_email: str = Depends(get_current_user_email)):
+async def override_edit_chat(request: RequestMySQL.RequestEditNameChat,
+                             current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
@@ -123,20 +131,23 @@ async def override_edit_chat(request: RequestMySQL.RequestEditNameChat, current_
         raise HTTPException(status_code=400, detail="name_old field is required.")
     return MySQLService.edit_chat(request)
 
+
 @app.delete("/api/mysql/delete_chat/", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-async def override_delete_chat(request: RequestMySQL.RequestDeleteChat, current_user_email: str = Depends(get_current_user_email)):
+async def override_delete_chat(request: RequestMySQL.RequestDeleteChat,
+                               current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
-    chat_name= request.chat_name
+    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)
 
 
-@app.get("/api/mysql/load_chat_history",dependencies=[Depends(JWTBearer())] ,tags=["MySQL"])
-async def override_load_chat(chat_id: Optional[str] = Query(None), user_id: Optional[str] = Query(None), current_user_email: str = Depends(get_current_user_email)):
+@app.get("/api/mysql/load_chat_history", dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+async def override_load_chat(chat_id: Optional[str] = Query(None), user_id: Optional[str] = Query(None),
+                             current_user_email: str = Depends(get_current_user_email)):
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
@@ -152,19 +163,22 @@ async def override_load_chat(chat_id: Optional[str] = Query(None), user_id: Opti
     if not support_function.is_positive_integer(chat_id_int):
         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)
+    request = RequestMySQL.RequestLoadChatHistory(chat_id=chat_id, user_id=user_id)
     return MySQLService.load_chat_history(request)
 
-@app.get("/api/default/info_user",dependencies=[Depends(JWTBearer())], tags=["Default"])
-async def override_get_user(user_id: str = Query(None),current_user_email: str = Depends(get_current_user_email)):
+
+@app.get("/api/default/info_user", dependencies=[Depends(JWTBearer())], tags=["Default"])
+async def override_get_user(user_id: str = Query(None), current_user_email: str = Depends(get_current_user_email)):
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     request = RequestDefault.RequestInfoUser(user_id=user_id)
     return DefaultService.info_user(request)
 
+
 @app.post("/api/users/update_user_info", dependencies=[Depends(JWTBearer())], tags=["Login"])
-async def override_update_user_info(request: RequestLogin.RequestUpdateUserInfo, current_user_email: str = Depends(get_current_user_email)):
+async def override_update_user_info(request: RequestLogin.RequestUpdateUserInfo,
+                                    current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check != True:
@@ -186,8 +200,10 @@ async def override_update_user_info(request: RequestLogin.RequestUpdateUserInfo,
                                 data=res.Message(message="photo_url field is required."))
     return LoginService.update_user_info(request)
 
+
 @app.put('/api/users/change_password', dependencies=[Depends(JWTBearer())], tags=["Login"])
-async def override_reset_password_firebase(request: RequestLogin.RequestChangePassword, current_user_email: str = Depends(get_current_user_email)):
+async def override_reset_password_firebase(request: RequestLogin.RequestChangePassword,
+                                           current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check != True:
@@ -201,14 +217,20 @@ async def override_reset_password_firebase(request: RequestLogin.RequestChangePa
         return res.ReponseError(status=400,
                                 data=res.Message(message="current_password field is required."))
     return LoginService.change_password(request)
+
+
 @app.delete("/api/file/delete_all_file/", dependencies=[Depends(JWTBearer())], tags=["File"])
-async def override_delete_folder(request: RequestFile.RequestDeleteAllFile, current_user_email: str = Depends(get_current_user_email)):
+async def override_delete_folder(request: RequestFile.RequestDeleteAllFile,
+                                 current_user_email: str = Depends(get_current_user_email)):
     check = support_function.check_value_user_id(request.user_id, current_user_email)
     if check != True:
         return check
     return FileService.deleteAllFile(request)
+
+
 @app.delete("/api/file/delete_one_file/", dependencies=[Depends(JWTBearer())], tags=["File"])
-async def override_delete_one_file(request: RequestFile.RequestDeleteFile, current_user_email: str = Depends(get_current_user_email)):
+async def override_delete_one_file(request: RequestFile.RequestDeleteFile,
+                                   current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check != True:
@@ -219,27 +241,32 @@ async def override_delete_one_file(request: RequestFile.RequestDeleteFile, curre
                                 data=res.Message(message="name_file is required."))
     return FileService.deleteFile(request)
 
+
 @app.post("/api/file/chatbot/download_folder/", dependencies=[Depends(JWTBearer())], tags=["File"])
-async def override_download_folder_from_dropbox(request: RequestFile.RequestDownLoadFolder, current_user_email: str = Depends(get_current_user_email)):
+async def override_download_folder_from_dropbox(request: RequestFile.RequestDownLoadFolder,
+                                                current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     return FileService.download_folder(request)
 
+
 @app.post("/api/file/chatbot/download_files/", dependencies=[Depends(JWTBearer())], tags=["File"])
-async def override_download_file_by_id(request: RequestFile.RequestDownLoadFile, current_user_email: str = Depends(get_current_user_email)):
+async def override_download_file_by_id(request: RequestFile.RequestDownLoadFile,
+                                       current_user_email: str = Depends(get_current_user_email)):
     user_id = request.user_id
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     return FileService.download_file(request)
 
+
 @app.post("/api/file/upload_files/", dependencies=[Depends(JWTBearer())], tags=["File"])
 async def override_upload_files_dropbox(
-    user_id: str = Form(None),
-    files: List[UploadFile] = File(None),
-    current_user_email: str = Depends(get_current_user_email)
+        user_id: str = Form(None),
+        files: List[UploadFile] = File(None),
+        current_user_email: str = Depends(get_current_user_email)
 ):
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
@@ -247,38 +274,47 @@ async def override_upload_files_dropbox(
     request = RequestFile.RequestUploadFile(files=files, user_id=user_id)
     return FileService.upload_files(request)
 
+
 @app.post("/api/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(...), question: str = Form(None), chat_name: str = Form(None), current_user_email: str = Depends(get_current_user_email)):
+async def override_handle_query2_upgrade_old(request: Request, user_id: str = Form(None), text_all: str = Form(...),
+                                             question: str = Form(None), chat_name: str = Form(None),
+                                             current_user_email: str = Depends(get_current_user_email)):
     check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
-    request = RequestChat.RequestQuery2UpgradeOld(user_id=user_id, text_all=text_all, question=question, chat_name=chat_name)
+    request = RequestChat.RequestQuery2UpgradeOld(user_id=user_id, text_all=text_all, question=question,
+                                                  chat_name=chat_name)
     return ChatService.query2_upgrade_old(request)
 
+
 @app.get("/api/chat/chatbot/extract_file/", dependencies=[Depends(JWTBearer())], tags=["Chatbot"])
 async def override_extract_file(user_id: str, current_user_email: str = Depends(get_current_user_email)):
-    check = support_function.check_value_user_id(user_id,current_user_email)
+    check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     request = RequestChat.RequestExtractFile(user_id=user_id)
     return ChatService.extract_file(request)
 
+
 @app.get("/api/chat/chatbot/generate_question/", dependencies=[Depends(JWTBearer())], tags=["Chatbot"])
-async def override_generate_question(user_id: str , current_user_email: str = Depends(get_current_user_email)):
-    check = support_function.check_value_user_id(user_id,current_user_email)
+async def override_generate_question(user_id: str, current_user_email: str = Depends(get_current_user_email)):
+    check = support_function.check_value_user_id(user_id, current_user_email)
     if check is not True:
         return check
     request = RequestChat.RequestGenerateQuestion(user_id=user_id)
     return ChatService.generate_question(request)
 
+
 @app.post("/api/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)):
-    check = support_function.check_value_user_id(user_id,current_user_email)
+async def override_upload_image(user_id: str = Form(None), file: UploadFile = File(...),
+                                current_user_email: str = Depends(get_current_user_email)):
+    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)
 
+
 app.include_router(MySQLController.router, prefix="/api/mysql")
 app.include_router(LoginController.router, prefix="/api/users")
 app.include_router(FileController.router, prefix="/api/file")
@@ -303,31 +339,47 @@ routes_to_override = {
     "/api/chat/chatbot/generate_question/": {"GET"},
     "/api/default/upload_image/": {"POST"},
     "/api/default/info_user": {"GET"}
-    
+
 }
 
 app.router.routes = [
     route for route in app.router.routes
     if not (
-        route.path in routes_to_override and
-        route.methods.intersection(routes_to_override[route.path])
+            route.path in routes_to_override and
+            route.methods.intersection(routes_to_override[route.path])
     )
 ]
 
-app.add_api_route("/api/mysql/render_chat_history", override_render_chat, methods=["GET"], dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-app.add_api_route("/api/mysql/load_chat_history", override_load_chat, methods=["GET"], dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-app.add_api_route("/api/mysql/edit_chat/", override_edit_chat, methods=["PUT"], dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-app.add_api_route("/api/mysql/delete_chat/", override_delete_chat, methods=["DELETE"], dependencies=[Depends(JWTBearer())], tags=["MySQL"])
-app.add_api_route("/api/users/update_user_info", override_update_user_info, methods=["POST"],dependencies=[Depends(JWTBearer())], tags=["Login"])
-app.add_api_route("/api/users/change_password", override_reset_password_firebase, methods=["PUT"],dependencies=[Depends(JWTBearer())], tags=["Login"])
-app.add_api_route("/api/file/delete_all_file/", override_delete_folder, methods=["DELETE"],dependencies=[Depends(JWTBearer())], tags=["File"])
-app.add_api_route("/api/file/delete_one_file/", override_delete_one_file, methods=["DELETE"],dependencies=[Depends(JWTBearer())], tags=["File"])
-app.add_api_route("/api/file/chatbot/download_folder/", override_download_folder_from_dropbox, methods=["POST"],dependencies=[Depends(JWTBearer())], tags=["File"])
-app.add_api_route("/api/file/chatbot/download_files/", override_download_file_by_id, methods=["POST"],dependencies=[Depends(JWTBearer())], tags=["File"])
-app.add_api_route("/api/file/upload_files/", override_upload_files_dropbox, methods=["POST"],dependencies=[Depends(JWTBearer())], tags=["File"])
-app.add_api_route("/api/chat/chatbot/query/", override_handle_query2_upgrade_old, methods=["POST"],dependencies=[Depends(JWTBearer())], tags=["Chat"])
-app.add_api_route("/api/chat/chatbot/extract_file/", override_extract_file, methods=["GET"], dependencies=[Depends(JWTBearer())], tags=["Chat"])
-app.add_api_route("/api/chat/chatbot/generate_question/", override_generate_question,methods=["GET"], dependencies=[Depends(JWTBearer())], tags=["Chat"])
-app.add_api_route("/api/default/upload_image/", override_upload_image, methods=["POST"], dependencies=[Depends(JWTBearer())], tags=["Default"])
-app.add_api_route("/api/default/info_user",override_get_user, methods=["GET"], dependencies=[Depends(JWTBearer())], tags=["Default"])
-app.include_router(OTPController.router, tags = ["OTP"], prefix="/api/otp")
+app.add_api_route("/api/mysql/render_chat_history", override_render_chat, methods=["GET"],
+                  dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+app.add_api_route("/api/mysql/load_chat_history", override_load_chat, methods=["GET"],
+                  dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+app.add_api_route("/api/mysql/edit_chat/", override_edit_chat, methods=["PUT"], dependencies=[Depends(JWTBearer())],
+                  tags=["MySQL"])
+app.add_api_route("/api/mysql/delete_chat/", override_delete_chat, methods=["DELETE"],
+                  dependencies=[Depends(JWTBearer())], tags=["MySQL"])
+app.add_api_route("/api/users/update_user_info", override_update_user_info, methods=["POST"],
+                  dependencies=[Depends(JWTBearer())], tags=["Login"])
+app.add_api_route("/api/users/change_password", override_reset_password_firebase, methods=["PUT"],
+                  dependencies=[Depends(JWTBearer())], tags=["Login"])
+app.add_api_route("/api/file/delete_all_file/", override_delete_folder, methods=["DELETE"],
+                  dependencies=[Depends(JWTBearer())], tags=["File"])
+app.add_api_route("/api/file/delete_one_file/", override_delete_one_file, methods=["DELETE"],
+                  dependencies=[Depends(JWTBearer())], tags=["File"])
+app.add_api_route("/api/file/chatbot/download_folder/", override_download_folder_from_dropbox, methods=["POST"],
+                  dependencies=[Depends(JWTBearer())], tags=["File"])
+app.add_api_route("/api/file/chatbot/download_files/", override_download_file_by_id, methods=["POST"],
+                  dependencies=[Depends(JWTBearer())], tags=["File"])
+app.add_api_route("/api/file/upload_files/", override_upload_files_dropbox, methods=["POST"],
+                  dependencies=[Depends(JWTBearer())], tags=["File"])
+app.add_api_route("/api/chat/chatbot/query/", override_handle_query2_upgrade_old, methods=["POST"],
+                  dependencies=[Depends(JWTBearer())], tags=["Chat"])
+app.add_api_route("/api/chat/chatbot/extract_file/", override_extract_file, methods=["GET"],
+                  dependencies=[Depends(JWTBearer())], tags=["Chat"])
+app.add_api_route("/api/chat/chatbot/generate_question/", override_generate_question, methods=["GET"],
+                  dependencies=[Depends(JWTBearer())], tags=["Chat"])
+app.add_api_route("/api/default/upload_image/", override_upload_image, methods=["POST"],
+                  dependencies=[Depends(JWTBearer())], tags=["Default"])
+app.add_api_route("/api/default/info_user", override_get_user, methods=["GET"], dependencies=[Depends(JWTBearer())],
+                  tags=["Default"])
+app.include_router(OTPController.router, tags=["OTP"], prefix="/api/otp")