WebSockets

1
import asyncio
2
import websockets
3
import json
4
import base64
5
import pyaudio
6
7
8
API_KEY = "YOUR_API_KEY"
9
WS_URL = "wss://api.murf.ai/v1/speech/stream-input"
10
PARAGRAPH = "With a single WebSocket connection, you can stream text input and receive synthesized audio continuously, without the overhead of repeated HTTP requests. This makes it ideal for use cases where your application sends or receives text in chunks and needs real-time audio to deliver a smooth, conversational experience"
11
12
# Audio format settings (must match your API output)
13
SAMPLE_RATE = 44100
14
CHANNELS = 1
15
FORMAT = pyaudio.paInt16
16
17
async def tts_stream():
18
  async with websockets.connect(
19
      f"{WS_URL}?api-key={API_KEY}&sample_rate=44100&channel_type=MONO&format=WAV"
20
  ) as ws:
21
      # Send voice config first (optional)
22
      voice_config_msg = {
23
          "voice_config": {
24
              "voiceId": "en-US-amara",
25
              "style": "Conversational",
26
              "rate": 0,
27
              "pitch": 0,
28
              "variation": 1
29
          }
30
      }
31
      print(f'Sending payload : {voice_config_msg}')
32
      await ws.send(json.dumps(voice_config_msg))
33
34
      # Send text in one go (or chunk if you want streaming)
35
      text_msg = {
36
          "text": PARAGRAPH,
37
          "end" : True # This will close the context. So you can re-run and concurrency is available.
38
      }
39
      print(f'Sending payload : {text_msg}')
40
      await ws.send(json.dumps(text_msg))
41
42
      # Setup audio stream
43
      pa = pyaudio.PyAudio()
44
      stream = pa.open(format=FORMAT, channels=CHANNELS, rate=SAMPLE_RATE, output=True)
45
46
      first_chunk = True
47
      try:
48
          while True:
49
              response = await ws.recv()
50
              data = json.loads(response)
51
              print(f'Received data:  {data}')
52
              if "audio" in data:
53
                  audio_bytes = base64.b64decode(data["audio"])
54
                  # Skip the first 44 bytes (WAV header) only for the first chunk
55
                  if first_chunk and len(audio_bytes) > 44:
56
                      audio_bytes = audio_bytes[44:]
57
                      first_chunk = False
58
                  stream.write(audio_bytes)
59
              if data.get("isFinalAudio"):
60
                  break
61
      finally:
62
          stream.stop_stream()
63
          stream.close()
64
          pa.terminate()
65
66
if __name__ == "__main__":
67
    asyncio.run(tts_stream())