大功告成

webrtc/tracks.py

@@ -6,10 +6,7 @@ from collections import deque
 
 import numpy as np
 from aiortc import MediaStreamTrack
-from av import AudioFrame, VideoFrame
-
-from core.state_machine import Arbitrator, SessionState
-from services.avatar import AvatarService
+from av import AudioFrame
 
 
 class AudioBus:
@@ -55,41 +52,6 @@ class AudioBus:
             self._level = 0.0
 
 
-class AvatarVideoTrack(MediaStreamTrack):
-    kind = "video"
-
-    def __init__(
-        self,
-        avatar_service: AvatarService,
-        arbitrator: Arbitrator,
-        audio_bus: AudioBus,
-        fps: int = 25,
-    ) -> None:
-        super().__init__()
-        self.avatar_service = avatar_service
-        self.arbitrator = arbitrator
-        self.audio_bus = audio_bus
-        self.fps = fps
-        self._pts = 0
-        self._time_base = Fraction(1, 90000)
-
-    async def recv(self) -> VideoFrame:
-        await asyncio.sleep(1 / self.fps)
-        generated = await self.avatar_service.pop_generated_frame()
-        if generated is not None:
-            arr = generated
-        else:
-            level = await self.audio_bus.level()
-            speaking = self.arbitrator.state == SessionState.AVATAR_SPEAKING or level > 0.02
-            arr = await self.avatar_service.render_frame(mouth_open=level, speaking=speaking)
-
-        frame = VideoFrame.from_ndarray(arr, format="bgr24")
-        frame.pts = self._pts
-        frame.time_base = self._time_base
-        self._pts += int(90000 / self.fps)
-        return frame
-
-
 class AvatarAudioTrack(MediaStreamTrack):
     kind = "audio"