diff --git a/facefusion/apis/endpoints/stream.py b/facefusion/apis/endpoints/stream.py index b777d64c..0eef623a 100644 --- a/facefusion/apis/endpoints/stream.py +++ b/facefusion/apis/endpoints/stream.py @@ -5,7 +5,7 @@ from starlette.websockets import WebSocket, WebSocketState from facefusion import rtc_manager from facefusion.apis import websocket_manager -from facefusion.apis.stream_manager import destroy_stream, negotiate_video, process_image +from facefusion.apis.stream_manager import destroy_video, negotiate_video, process_image from facefusion.apis.websocket_helper import get_sec_websocket_protocol @@ -44,7 +44,7 @@ async def post_stream(request : Request) -> Response: async def delete_stream(request : Request) -> Response: - if destroy_stream(): + if destroy_video(): return Response(status_code = HTTP_200_OK) return Response(status_code = HTTP_404_NOT_FOUND) diff --git a/facefusion/apis/stream_manager.py b/facefusion/apis/stream_manager.py index 9369a856..7332cde3 100644 --- a/facefusion/apis/stream_manager.py +++ b/facefusion/apis/stream_manager.py @@ -161,7 +161,7 @@ def process_video(rtc_peer : RtcPeer) -> None: process_manager.end() -def destroy_stream() -> bool: +def destroy_video() -> bool: if rtc_manager.has_peer(): rtc_manager.delete_peer() return not rtc_manager.has_peer() diff --git a/tests/apis/test_stream_manager.py b/tests/apis/test_stream_manager.py index 601138ab..5d4cb8b6 100644 --- a/tests/apis/test_stream_manager.py +++ b/tests/apis/test_stream_manager.py @@ -7,7 +7,7 @@ from unittest.mock import AsyncMock, patch import pytest from facefusion import rtc, rtc_manager, state_manager, store_creator -from facefusion.apis.stream_manager import destroy_stream, negotiate_video, process_image, process_video, receive_vision_frames +from facefusion.apis.stream_manager import destroy_video, negotiate_video, process_image, process_video, receive_vision_frames from facefusion.common_helper import is_linux, is_windows from facefusion.download import conditional_download from facefusion.hash_helper import create_hash @@ -193,7 +193,7 @@ def test_destroy_stream() -> None: set_session_id('test-destroy-stream') rtc_manager.set_peer(rtc_peer) - assert destroy_stream() is True + assert destroy_video() is True assert rtc_manager.get_peer() is None - assert destroy_stream() is False + assert destroy_video() is False