CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-02 10:51:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

resolve todos in stream endpoint (#1114)

Browse Source
This commit is contained in:
Henry Ruhs
2026-05-15 12:40:50 +02:00
committed by GitHub
parent 98adce8a2b
commit ad3b582c49
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/apis/endpoints/stream.py
+6 -5
View File
@@ -8,24 +8,25 @@ from facefusion.apis.session_helper import extract_access_token
from facefusion.apis.stream_helper import add_rtc_viewer, handle_image_stream, handle_video_stream
# TODO: reject websocket with invalid or missing stream mode
async def websocket_stream(websocket : WebSocket) -> None:
stream_mode = websocket.query_params.get('mode')
if stream_mode == 'image':
await handle_image_stream(websocket)
return await handle_image_stream(websocket)
if stream_mode == 'video':
await handle_video_stream(websocket)
return await handle_video_stream(websocket)
return await websocket.close(1008)
# TODO: validate content type is application/sdp, sanitize sdp input
async def post_stream(request : Request) -> Response:
content_type = request.headers.get('content-type')
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:
if content_type == 'application/sdp' and session_id:
sdp_offer = await request.body()
sdp_answer = add_rtc_viewer(session_id, sdp_offer.decode())