diff --git a/rtcllm/aav.py b/rtcllm/aav.py index 8119179..7470aff 100644 --- a/rtcllm/aav.py +++ b/rtcllm/aav.py @@ -8,7 +8,7 @@ class MyMediaPlayer(MediaPlayer): pass class MyTrackBase(MediaStreamTrack): - def __init__(self, source=None): + def __init__(self, source): super().__init__() self.dumb = None self.source = source @@ -47,13 +47,13 @@ class MyTrackBase(MediaStreamTrack): class MyAudioStreamTrack(MyTrackBase): kind = 'audio' def __init__(self, source): - super().__init__(source=source) + super().__init__(source) self.dumb = AudioStreamTrack() class MyVideoStreamTrack(MyTrackBase): kind = 'video' def __init__(self, source): - super().__init__(source=source) + super().__init__(source) self.dumb = AudioStreamTrack() """ diff --git a/rtcllm/rtc.py b/rtcllm/rtc.py index f94f05e..b96ccd3 100644 --- a/rtcllm/rtc.py +++ b/rtcllm/rtc.py @@ -51,7 +51,6 @@ RTCPeerConnection.get_local_candidates = pc_get_local_candidates class RTCLLM: def __init__(self, ws_url, iceServers): - # self.stt_model = WhisperModel('large-v3', device="cuda", compute_type="float16") self.ws_url = ws_url self.omni_infer = OmniInference(ckpt_dir='/d/models/mini-omni') self.iceServers = iceServers @@ -132,6 +131,7 @@ class RTCLLM: # pc.addTrack(LoopingVideoTrack('./1.mp4')) # pc.addTrack(player.video) await self.ws_send(json.dumps({'type':'callAccepted', 'to':data['from']})) + print('auto_accept_call() end') async def pc_track(self, peerid, track): peer = self.peers[peerid]