mirror of
https://github.com/facefusion/facefusion.git
synced 2026-05-31 01:49:29 +02:00
Henry Ruhs
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from starlette.requests import Request
|
|
from starlette.responses import Response
|
|
from starlette.status import HTTP_201_CREATED, HTTP_404_NOT_FOUND
|
|
from starlette.websockets import WebSocket
|
|
|
|
from facefusion import rtc_store, session_context, session_manager
|
|
from facefusion.apis.session_helper import extract_access_token
|
|
from facefusion.apis.stream_helper import handle_image_stream, handle_video_stream
|
|
|
|
|
|
async def websocket_stream(websocket : WebSocket) -> None:
|
|
stream_mode = websocket.query_params.get('mode')
|
|
|
|
if stream_mode == 'image':
|
|
await handle_image_stream(websocket)
|
|
|
|
if stream_mode == 'video':
|
|
await handle_video_stream(websocket)
|
|
|
|
|
|
async def post_stream(request : Request) -> Response:
|
|
access_token = extract_access_token(request.scope)
|
|
session_id = session_manager.find_session_id(access_token)
|
|
session_context.set_session_id(session_id)
|
|
|
|
if session_id:
|
|
sdp_offer = (await request.body()).decode()
|
|
sdp_answer = rtc_store.add_rtc_viewer(session_id, sdp_offer)
|
|
|
|
if sdp_answer:
|
|
return Response(sdp_answer, status_code = HTTP_201_CREATED, media_type = 'application/sdp')
|
|
|
|
return Response(status_code = HTTP_404_NOT_FOUND)
|