add capabilities endpoint

facefusion/apis/core.py

@@ -4,6 +4,7 @@ from starlette.middleware.cors import CORSMiddleware
 from starlette.routing import Route, WebSocketRoute
 
 from facefusion.apis.endpoints.assets import delete_assets, get_asset, get_assets, upload_asset
+from facefusion.apis.endpoints.capabilities import get_capabilities
 from facefusion.apis.endpoints.metrics import get_metrics, websocket_metrics
 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
@@ -24,6 +25,7 @@ def create_api() -> Starlette:
 			Route('/assets', upload_asset, methods = [ 'POST' ], middleware = [ session_guard ]),
 			Route('/assets/{asset_id}', get_asset, methods = [ 'GET' ], middleware = [ session_guard ]),
 			Route('/assets', delete_assets, methods = [ 'DELETE' ], middleware = [ session_guard ]),
+			Route('/capabilities', get_capabilities, methods = [ 'GET' ]),
 			Route('/metrics', get_metrics, methods = [ 'GET' ], middleware = [ session_guard ]),
 			WebSocketRoute('/metrics', websocket_metrics, middleware = [ session_guard ]),
 			WebSocketRoute('/ping', websocket_ping, middleware = [ session_guard ])

facefusion/apis/endpoints/capabilities.py

@@ -0,0 +1,18 @@
+from starlette.requests import Request
+from starlette.responses import JSONResponse
+from starlette.status import HTTP_200_OK
+
+import facefusion.choices
+
+
+async def get_capabilities(request : Request) -> JSONResponse:
+	capabilities =\
+	{
+		'formats':
+		{
+			'audio': facefusion.choices.audio_formats,
+			'image': facefusion.choices.image_formats,
+			'video': facefusion.choices.video_formats
+		}
+	}
+	return JSONResponse(capabilities, status_code = HTTP_200_OK)