rtcllm/rtcllm/aav.py
2024-09-11 12:01:41 +08:00

48 lines
1.3 KiB
Python

import random
from aiortc.contrib.media import MediaBlackhole, MediaPlayer, MediaRecorder, MediaRelay
from aiortc import VideoStreamTrack, AudioStreamTrack
class MyMediaPlayer(MediaPlayer):
pass
class MyTrackBasea:
def __init__(self, source=None):
self.source = source
self.track = self.set_track()
self.track.on('ended', self.reload)
def set_track(self):
if self.kind == 'audio':
self.track = self.source.audio
else:
self.track = self.source.video
def self.reload(self):
print(f'reload the source----{self.source._file_path}')
self.set_source(MyMediaPlayer(self.source._file_path))
def set_source(self, source):
self.source = source
self.set_track()
async def recv(self):
if self.source is None:
return None
f = await self.source.audio.recv()
while f is None:
print('MyAudioTrack::recv(): will reload source')
self.set_source(MyMediaPlayer(self.source._file_path))
f = await self.source.audio.recv()
return f
class MyAudioStreamTrack(AudioStreamTrack, MyTrackBasea):
def __init__(self, source=None):
AudioStreamTrack.__init__(self)
MyTrackBase.__init__(self, source=source)
class MyVideoStreamTrack(VideoStreamTrack, MyTrackBasea):
def __init__(self, source=None):
VideoStreamTrack.__init__(self)
MyTrackBase.__init__(self, source=source)