diff --git a/facefusion/apis/core.py b/facefusion/apis/core.py index 10541ba9..45299786 100644 --- a/facefusion/apis/core.py +++ b/facefusion/apis/core.py @@ -3,9 +3,9 @@ from starlette.middleware import Middleware from starlette.middleware.cors import CORSMiddleware from starlette.routing import Route, WebSocketRoute -from facefusion.apis.ping import websocket_ping -from facefusion.apis.session import create_session, create_session_guard, destroy_session, get_session, refresh_session -from facefusion.apis.state import get_state, set_state +from facefusion.apis.endpoints.ping import websocket_ping +from facefusion.apis.endpoints.session import create_session, create_session_guard, destroy_session, get_session, refresh_session +from facefusion.apis.endpoints.state import get_state, set_state def create_api() -> Starlette: diff --git a/facefusion/apis/ping.py b/facefusion/apis/endpoints/ping.py similarity index 100% rename from facefusion/apis/ping.py rename to facefusion/apis/endpoints/ping.py diff --git a/facefusion/apis/session.py b/facefusion/apis/endpoints/session.py similarity index 88% rename from facefusion/apis/session.py rename to facefusion/apis/endpoints/session.py index 59b24b38..7265b80d 100644 --- a/facefusion/apis/session.py +++ b/facefusion/apis/endpoints/session.py @@ -30,7 +30,7 @@ async def create_session(request : Request) -> JSONResponse: return JSONResponse( { - 'message': translator.get('something_went_wrong', __package__) + 'message': translator.get('something_went_wrong', 'facefusion.apis') }, status_code = HTTP_401_UNAUTHORIZED) @@ -53,7 +53,7 @@ async def get_session(request : Request) -> JSONResponse: return JSONResponse( { - 'message': translator.get('something_went_wrong', __package__) + 'message': translator.get('something_went_wrong', 'facefusion.apis') }, status_code = HTTP_401_UNAUTHORIZED) @@ -73,7 +73,7 @@ async def refresh_session(request : Request) -> JSONResponse: return JSONResponse( { - 'message': translator.get('something_went_wrong', __package__) + 'message': translator.get('something_went_wrong', 'facefusion.apis') }, status_code = HTTP_401_UNAUTHORIZED) @@ -88,12 +88,12 @@ async def destroy_session(request : Request) -> JSONResponse: return JSONResponse( { - 'message': translator.get('ok', __package__) + 'message': translator.get('ok', 'facefusion.apis') }, status_code = HTTP_200_OK) return JSONResponse( { - 'message': translator.get('something_went_wrong', __package__) + 'message': translator.get('something_went_wrong', 'facefusion.apis') }, status_code = HTTP_401_UNAUTHORIZED) @@ -110,14 +110,14 @@ def create_session_guard(app : ASGIApp) -> ASGIApp: response = JSONResponse( { - 'message': translator.get('invalid_access_token', __package__) + 'message': translator.get('invalid_access_token', 'facefusion.apis') }, status_code = HTTP_426_UPGRADE_REQUIRED) return await response(scope, receive, send) response = JSONResponse( { - 'message': translator.get('invalid_access_token', __package__) + 'message': translator.get('invalid_access_token', 'facefusion.apis') }, status_code = HTTP_401_UNAUTHORIZED) return await response(scope, receive, send) diff --git a/facefusion/apis/state.py b/facefusion/apis/endpoints/state.py similarity index 100% rename from facefusion/apis/state.py rename to facefusion/apis/endpoints/state.py