fmt

Author: typester

examples/play_audio_stream.py

@@ -10,6 +10,7 @@
 BLOCKSIZE = 480  # 10ms chunks at 48kHz
 CHANNELS = 1
 
+
 class AudioBuffer:
     def __init__(self, blocksize=BLOCKSIZE):
         self.blocksize = blocksize
@@ -20,8 +21,8 @@ def add_frame(self, frame_data):
 
     def get_chunk(self):
         if len(self.buffer) >= self.blocksize:
-            chunk = self.buffer[:self.blocksize]
-            self.buffer = self.buffer[self.blocksize:]
+            chunk = self.buffer[: self.blocksize]
+            self.buffer = self.buffer[self.blocksize :]
             return chunk
         return None
 
@@ -34,6 +35,7 @@ def get_padded_chunk(self):
             return chunk
         return np.zeros(self.blocksize, dtype=np.int16)
 
+
 async def audio_player(queue: asyncio.Queue):
     """Pull from the queue and stream audio using sounddevice."""
     buffer = AudioBuffer(BLOCKSIZE)
@@ -62,14 +64,15 @@ def callback(outdata, frames, time, status):
         samplerate=SAMPLERATE,
         channels=CHANNELS,
         blocksize=BLOCKSIZE,
-        dtype='int16',
+        dtype="int16",
         callback=callback,
-        latency='low'
+        latency="low",
     )
     with stream:
         while True:
             await asyncio.sleep(0.1)  # keep the loop alive
 
+
 async def rtc_session(room, queue: asyncio.Queue):
     track: rtc.RemoteAudioTrack | None = None
     while not track:
@@ -88,7 +91,7 @@ async def rtc_session(room, queue: asyncio.Queue):
         track=track,
         sample_rate=SAMPLERATE,
         num_channels=1,
-        noise_cancellation=noise_cancellation.BVC(), # or NC()
+        noise_cancellation=noise_cancellation.BVC(),  # or NC()
     )
 
     print("playing stream")
@@ -110,6 +113,7 @@ async def rtc_session(room, queue: asyncio.Queue):
         # Clean up the stream when done
         await stream.aclose()
 
+
 async def main():
     queue = asyncio.Queue(maxsize=50)
     player_task = asyncio.create_task(audio_player(queue))
@@ -150,4 +154,5 @@ async def main():
         except asyncio.CancelledError:
             pass
 
+
 asyncio.run(main())