wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

services(playht): make sure we only skip wav header no matter the size

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-11-13 17:01:29 +01:00
parent 2d2ef5a417
commit ef9e86d997
1 changed files with 2 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/pipecat/services/playht.py
View File

@@ -210,17 +210,11 @@ class PlayHTTTSService(TTSService):
async def _receive_task_handler(self): async def _receive_task_handler(self):
try: try:
header_size = 78 # Size of the WAV header + extra bytes we want to skip
header_received = False
async for message in self._get_websocket(): async for message in self._get_websocket():
if isinstance(message, bytes): if isinstance(message, bytes):
chunk_size = len(message) # Skip the WAV header message
if message.startswith(b"RIFF"):
# Skip the WAV header
if not header_received and chunk_size == header_size:
header_received = True
continue continue
await self.stop_ttfb_metrics() await self.stop_ttfb_metrics()
frame = TTSAudioRawFrame(message, self._settings["sample_rate"], 1) frame = TTSAudioRawFrame(message, self._settings["sample_rate"], 1)
await self.push_frame(frame) await self.push_frame(frame)
@@ -230,7 +224,6 @@ class PlayHTTTSService(TTSService):
msg = json.loads(message) msg = json.loads(message)
if "request_id" in msg and msg["request_id"] == self._request_id: if "request_id" in msg and msg["request_id"] == self._request_id:
await self.push_frame(TTSStoppedFrame()) await self.push_frame(TTSStoppedFrame())
header_received = False # Reset for the next audio stream
self._request_id = None self._request_id = None
elif "error" in msg: elif "error" in msg:
logger.error(f"{self} error: {msg}") logger.error(f"{self} error: {msg}")