SeaVoice RESTful APIs
This is the tutorial about how to use SeaVoice RESTful APIs to try Seasalt Speech-To-Text (STT) and Text-To-Speech (TTS) services.
Please contact info@seasalt.ai if you have any questions.
STT protocols
Log into SeaAuth:
https://seaauth.seasalt.ai/api/v1/users/login
{
"username": "test",
"password": "test",
"scope": "seavoice"
}
Obtain access token in the response
{
"account": "test",
"access_token": "eyJ0eXAiOi*****",
"token_type": "Bearer",
"refresh_token": "71bbffd5368*****"
}
Refresh access token if it is expired
3.1 check token lifetime
import jwt from datetime import datetime token = "eyJ0eXAiOiJKV1....." data = jwt.decode(token, options={"verify_signature": False}) print(datetime.fromtimestamp(data["exp"])) # 2022-12-06 12:42:543.2 refresh token with
https://seaauth.seasalt.ai/api/v1/users/rotate_tokeninput payload
{ "access_token": "eyJ0eXAiOi*****", "refresh_token": "71bbffd5368*****", "token_type": "Bearer", "service": "seavoice", }Output
{ "access_token": "eyJ0esr3hifi*****", "refresh_token": "fd27g1bfdg53g*****", "token_type": "Bearer", "service": "seavoice", }
Connect to SeaVoice STT websocket server:
wss://seavoice.seasalt.ai/api/v1/stt/ws
If successfully connected, Client sends json packages to stt server, for example:
authentication command
{
"command": "authentication",
"payload": {
"token": "<ACCESS_TOKEN>",
"settings": {
"language": "zh-TW",
"sample_rate": 16000,
"itn": false,
"contexts": {
"Seasalt": {
"rewrite": ["sea salt", "c salt"]
},
"SeaVoice": {
"rewrite": ["c voice"]
}
}
},
}
}
Note
"language": Currently supported languages: zh-TW, en-US"sample_rate": sample rate of the audio, e.g. 16000, 44100. Make sure this matches your audio."itn": whether to run inverse text normalisation and punctuation to recognition result, e.g. “mister” becomes “mr.”"contexts": a json dict to boost certain hotwords and/or phrases for recognition, and optionally rewrite certain spoken forms to a specific written form. Each key is a word/phrase for context biasing; each corresponding value is an optional dict containing a key ‘rewrite’ which maps to a list of possible spoken forms that will be rewritten to the written form (the key). In the above example, the word “seasalt” will be boosted and all occurences of “sea salt” and “c salt” will be rewritten to the capitalised “Seasalt”. Also, if a certain sentence is expected, you can also boost the whole sentence, e.g. “Seasalt is an AI company”
start recognition command: sending audio data for recognition
{
"command": "audio_data",
"payload": "<BASE64_ENCODED_AUDIO_DATA>"
}
stop recognition command
{
"command": "stop"
}
STT server receives audio data, performs recognition, and sends recognizing/recognized events to Client
info event (begin)
{
"event": "info",
"payload": {
"status": "begin"
}
}
info event (error)
{
"event": "info",
"payload": {
"status": "error",
"error": {
"message": "<ERROR_MESSAGE>",
"code": "<ERROR_CODE>"
}
}
}
recognizing event: intermediate streaming ASR results
{
"event": "recognizing"
"payload": {
"segment_id": "<SEG_ID>",
"text": "<PARTIAL_RESULTS>",
"voice_start_time": 0.1
}
}
recognized event: final ASR results
{
"event": "recognized"
"payload": {
"segment_id": "<SEG_ID>",
"text": "<FINAL_RESULTS>",
"voice_start_time": 0.1,
"duration": 2.5
}
}
Note
"voice_start_time": timestamp in seconds of that segment relative to the start of the audio."duration": duration of that segment.
Sample Client Script (STT)
Setup
# Python venv setup (recommends using Python 3.8.10)
python3 -m venv venv/seavoice
source venv/seavoice/bin/activate
pip install --upgrade pip
pip install websockets==10.3
pip install aiohttp==3.8.1
pip install PyJWT==2.5.0
Run client script
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2022 Seasalt AI, Inc
"""Client script for stt endpoint
Prerequisite:
python 3.8
python package:
- aiohttp==3.8.1
- websockets==10.3
- PyJWT==2.5.0
Usage:
python stt_client.py \
--account test \
--password test \
--lang zh-TW \
--enable-itn false \
--contexts "{}" \
--context_score 0 \
--audio-path test_audio.wav \
--sample-rate 8000
Get help:
python stt_client.py --help
"""
import argparse
import asyncio
import base64
import json
import logging
import time
from enum import Enum
from pathlib import Path
from urllib.parse import urljoin
import aiohttp
import jwt
import websockets
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(message)s",
handlers=[logging.StreamHandler()],
)
SEAAUTH_SCOPE_NAME: str = "seavoice"
TOKEN_TYPE: str = "Bearer"
AUDIO_CHUNK_INTERVAL: float = 0.1
ACCESS_TOKEN_LIFE_TIME_MINIMUM_IN_SECOND: int = 300
class Language(str, Enum):
EN_US = "en-US"
ZH_TW = "zh-TW"
async def main(args: argparse.Namespace):
if args.enable_log_file:
file_handler = logging.FileHandler("stt_client.log")
file_handler.setFormatter(logging.Formatter("%(asctime)s [%(levelname)s] %(message)s"))
logging.getLogger().addHandler(file_handler)
logging.info("Start to get access token.")
access_token = await _get_access_token(args)
await _do_stt(args, access_token)
async def _get_access_token(args: argparse.Namespace) -> str:
if args.service_token:
logging.info("Using service token...")
return args.service_token
credential = _get_credential_from_file(args.seaauth_credential_path)
if credential and credential["account"] == args.account:
access_token, refresh_token = credential["access_token"], credential["refresh_token"]
if _is_access_token_expired(credential["access_token"]):
credential = await _refresh_access_token(access_token, refresh_token)
_save_credential(
args.account, credential["access_token"], credential["refresh_token"], args.seaauth_credential_path
)
else:
logging.info(f"Got access token from {args.seaauth_credential_path}.")
else:
credential = await _login_seaauth(args.account, args.password)
_save_credential(args.account, credential["access_token"], credential["refresh_token"], args.seaauth_credential_path)
return credential["access_token"]
async def _login_seaauth(account: str, password: str) -> dict:
"""Login with SeaAuth.
Example of response:
{
"account": "test",
"access_token": "eyJ0eXAiOi*****",
"token_type": "Bearer",
"refresh_token": "71bbffd5368*****"
}
"""
logging.info("logging in to SeaAuth...")
payload = {"username": account, "password": password, "scope": SEAAUTH_SCOPE_NAME}
data = aiohttp.FormData()
data.add_fields(*payload.items())
async with aiohttp.ClientSession() as session:
async with session.post(urljoin(args.seaauth_url, "/api/v1/users/login"), data=data) as response:
if response.status >= 400:
raise Exception(await response.text())
data = await response.json()
logging.info("logged in to SeaAuth.")
return data
async def _refresh_access_token(access_token: str, refresh_token: str) -> dict:
logging.info("refreshing token...")
payload = {
"access_token": access_token,
"refresh_token": refresh_token,
"token_type": TOKEN_TYPE,
"scope": SEAAUTH_SCOPE_NAME,
}
async with aiohttp.ClientSession() as session:
async with session.post(urljoin(args.seaauth_url, "/api/v1/users/rotate_token"), json=payload) as response:
if response.status >= 400:
raise Exception(await response.text())
data = await response.json()
logging.info("Token is refreshed")
return data
async def _do_stt(args: argparse.Namespace, access_token: str):
start_time = time.time()
stt_endpoint_url = urljoin(args.seavoice_ws_url, "/api/v1/stt/ws")
logging.info("establishing ws connection...")
async with websockets.connect(stt_endpoint_url, open_timeout=20) as websocket:
logging.info("established ws connection")
stt_start_time = time.time()
is_begin, is_end = asyncio.Event(), asyncio.Event()
await asyncio.gather(
_receive_events(websocket, is_begin, is_end),
_send_commands(args, access_token, websocket, is_begin, is_end),
)
# wait for audio synthesized
logging.info("stt finished")
logging.info(
f"_do_stt finished, time elapsed: total: {time.time()-start_time: .3f}s, stt:{time.time()-stt_start_time: .3f}s"
)
async def _send_commands(
args: argparse.Namespace,
access_token: str,
websocket,
is_begin: asyncio.Event,
is_end: asyncio.Event,
):
logging.info("sending authentication command...")
await _send_authentication_command(args, websocket, access_token)
chunk_size = int(int(args.sample_rate) * 2 * AUDIO_CHUNK_INTERVAL)
# wait until received the begin event from server
await is_begin.wait()
logging.info("sending audio_data commands...")
await _send_audio_data_chunkily(websocket, args.audio_path, chunk_size)
logging.info("sending stop commands...")
await _send_stop_command(websocket)
logging.info("waiting for end event...")
await is_end.wait()
async def _receive_events(websocket, is_begin: asyncio.Event, is_end: asyncio.Event):
async for message in websocket:
event = json.loads(message)
event_name = event.get("event", "")
event_payload = event.get("payload", {})
if event_name == "info":
if event_payload.get("status") == "begin":
logging.info(f"received an info begin event: {event_payload}")
is_begin.set()
elif event_payload.get("status") == "error":
logging.info(f"received an info error event: {event_payload}")
raise Exception(f"received an info error event: {event_payload}")
elif event_payload.get("status") == "end":
logging.info("received an info end event")
is_end.set()
else:
logging.info(f"received an unknown info event: {event}")
elif event_name == "recognizing" or event_name == "recognized":
logging.info(f"received an {event_name} event: {event_payload}")
else:
logging.info(f"received an unknown event: {event}")
async def _send_stop_command(websocket):
command_str = json.dumps({"command": "stop"})
await websocket.send(command_str)
async def _send_authentication_command(args: argparse.Namespace, websocket, access_token: str):
try:
with open(args.contexts_file, "r") as f:
contexts_json = json.load(f)
logging.info(f"Loaded contexts json file from {args.contexts_file}")
except Exception as e:
logging.warning(f"Not loading contexts json file from: {args.contexts_file} due to exception {e}")
contexts_json = {}
authentication_command = {
"command": "authentication",
"payload": {
"token": access_token,
"settings": {
"language": args.lang,
"sample_rate": args.sample_rate,
"itn": args.enable_itn,
"contexts": json.dumps(contexts_json),
"context_score": args.context_score,
},
},
}
command_str = json.dumps(authentication_command)
await websocket.send(command_str)
async def _send_audio_data_chunkily(websocket, audio_path: str, chunk_size: int):
with open(audio_path, "rb") as f:
while True:
audio = f.read(chunk_size)
if audio == b"":
break
await _send_one_audio_data_command(websocket, audio)
await asyncio.sleep(AUDIO_CHUNK_INTERVAL)
async def _send_one_audio_data_command(websocket, audio: bytes):
audio_data_command = {"command": "audio_data", "payload": base64.b64encode(audio).decode()}
await websocket.send(json.dumps(audio_data_command))
def _check_file_path_exists(audio_path: str):
if not Path(audio_path).exists():
raise Exception(f"No audio file exists at {audio_path}.")
def _convert_argument_str_to_bool(args: argparse.Namespace) -> argparse.Namespace:
args.enable_itn = args.enable_itn.lower() == "true"
args.enable_log_file = args.enable_log_file.lower() == "true"
return args
def _is_access_token_expired(access_token: str) -> bool:
life_time = _get_token_lifetime(access_token)
return life_time < ACCESS_TOKEN_LIFE_TIME_MINIMUM_IN_SECOND
def _get_token_lifetime(access_token: str) -> int:
try:
data = jwt.decode(access_token, options={"verify_signature": False})
return data["exp"] - int(time.time())
except Exception as error:
logging.info(f"Invalid access_token format error:{error}")
def _save_credential(
account: str,
access_token: str,
refresh_token: str,
seaauth_credential_path: str,
):
Path(seaauth_credential_path).touch(exist_ok=True)
with open(seaauth_credential_path, "w") as f:
json.dump({"account": account, "access_token": access_token, "refresh_token": refresh_token}, f)
logging.info(f"The credential is saved to {seaauth_credential_path}.")
def _get_credential_from_file(seaauth_credential_path: str) -> dict:
if not Path(seaauth_credential_path).exists():
logging.info(f"No credential file exists at {seaauth_credential_path}.")
return {}
try:
with open(seaauth_credential_path, "r") as f:
credential = json.load(f)
except Exception as error:
logging.error(f"Cannot parse {seaauth_credential_path} into json due to {error}")
raise error
if "access_token" not in credential or "refresh_token" not in credential:
raise Exception(f"{credential} not includes both access_token and refresh_token.")
return credential
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--account", type=str, required=False, help="account of a SeaAuth account.")
parser.add_argument("--password", type=str, required=False, help="password of a SeaAuth account.")
parser.add_argument(
"--service-token",
dest="service_token",
type=str,
required=False,
help="If you provide the service token, we could use this token direclty without login by password",
)
parser.add_argument(
"--lang",
type=str,
required=True,
choices=[lang for lang in Language],
help='Language of TTS server, must in ["zh-TW", "en-US"]',
)
parser.add_argument(
"--sample-rate",
dest="sample_rate",
type=int,
required=True,
help="Set the sample rate of speech.",
)
parser.add_argument(
"--audio-path",
dest="audio_path",
type=str,
required=True,
help="The path of wav file for speech to text.",
)
parser.add_argument(
"--seaauth-url",
dest="seaauth_url",
type=str,
required=False,
default="https://seaauth.seasalt.ai",
help="Url of SeaAuth.",
)
parser.add_argument(
"--seaauth-credential-path",
dest="seaauth_credential_path",
type=str,
required=False,
default="seavoice_credential.json",
help="Credential storage of access token and refresh token.",
)
parser.add_argument(
"--seavoice-ws-url",
dest="seavoice_ws_url",
type=str,
required=False,
default="wss://seavoice.seasalt.ai",
help="Url of SeaVoice.",
)
parser.add_argument(
"--enable-itn",
dest="enable_itn",
type=str,
required=False,
default="true",
help="Enable the ITN feature(true or false), default is true.",
)
parser.add_argument(
"--contexts-file",
dest="contexts_file",
type=str,
required=False,
default="",
help="Path to a json file containing contexts for boosting and rewrite rules.",
)
parser.add_argument(
"--context-score",
dest="context_score",
type=float,
required=False,
default=2.0,
help="Set the default context score of stt server.",
)
parser.add_argument(
"--enable-log-file",
dest="enable_log_file",
type=str,
required=False,
default="false",
help="Save the log to the file(true or false), default is false.",
)
args = parser.parse_args()
_check_file_path_exists(args.audio_path)
args = _convert_argument_str_to_bool(args)
asyncio.run(main(args))
TTS protocols
Log into SeaAuth:
https://seaauth.seasalt.ai/api/v1/users/login
{
"username": "test",
"password": "test",
"scope": "seavoice"
}
Obtain access token in the response
{
"account": "test",
"access_token": "eyJ0eXAiOi*****",
"token_type": "Bearer",
"refresh_token": "71bbffd5368*****"
}
Refresh access token if it is expired
3.1 check token lifetime
import jwt from datetime import datetime token = "eyJ0eXAiOiJKV1....." data = jwt.decode(token, options={"verify_signature": False}) print(datetime.fromtimestamp(data["exp"])) # 2022-12-06 12:42:543.2 refresh token with
https://seaauth.seasalt.ai/api/v1/users/rotate_tokeninput payload
{ "access_token": "eyJ0eXAiOi*****", "refresh_token": "71bbffd5368*****", "token_type": "Bearer", "service": "seavoice", }Output
{ "access_token": "eyJ0esr3hifi*****", "refresh_token": "fd27g1bfdg53g*****", "token_type": "Bearer", "service": "seavoice", }
Connect to SeaVoice TTS websocket server:
wss://seavoice.seasalt.ai/api/v1/tts/ws
If successfully connected, Client sends json packages to TTS server, for example (settings and data are shown with default values),
authentication command
{
"command": "authentication",
"payload": {
"token": "{access_token}",
"settings": {
"language": "en-US",
"voice": "Mike",
},
}
}
synthesis command
{
"command": "synthesis",
"payload": {
"settings": {
"pitch": 0,
"speed": 0,
"volume": 50,
"rules": "",
"sample_rate": 8000,
},
"data": {
"text": "test",
"ssml": true
}
}
}
Note
- <language> / <voice>: Choose from the following options
- zh-TW
Tongtong
Vivian
- en-US
Mike
Moxie
Lissa
- <pitch>
default: 0.0
range: [-5.0, 5.0]
description: adjust the pitch of the synthesized voice, where positive values raise the pitch and negative values lower the pitch.
- <speed>
default = 1.0
range: [0.0, 3.0]
description: adjust the speed of the synthesized voice, where values > 1.0 speed up the speech and values < 1.0 slows down the speech.
- <volume>
default: 50.0
range: [0.0, 100.0]
description: adjust the volume of the synthesized voice, where values > 50.0 increases the volume and values < 50.0 decreases the volume.
- <sample_rate>
default: 22050
range: [8000, 48000]
description: set the output audio sample rate
- <rules>
default: (empty string)
description: pronunciation rules as a string in the following format
<WORD1> | <PRONUNCIATION1>\n<WORD2> | <PRONUNCIATION2>for “zh-TW”, pronunciation can be specified in zhuyin, pinyin, or Chinese characters, e.g.
TSMC | 台積電\n你好 | ㄋㄧˇ ㄏㄠˇ\n為了 | wei4 le5for “en-US”, pronunciation can be specified with English words, e.g. “XÆA12 | ex ash ay twelvenSideræl|psydeereal”
- <ssml>
default: false
description: should be True if
<text>is an SSML string, i.e. using SSML tags. See Supported SSML Tags for more info.
After sending the package, Client calls ws.recv() to wait for TTS server to send the streaming audio data.
TTS server performs synthesis and keeps sending streaming audio data to Client. The audio package format is as follows:
{
"status": "SEQ_STATUS",
"message": "MESSAGE",
"sid": "SEQ_ID",
"data": {
"audio": "AUDIO_DATA",
"status": "STATUS"
}
}
Note
<SEQ_STATUS>: Either “ok” or an error message
<MESSAGE>: Additional information based on the status
<SEQ_ID>: audio sequence id
<STATUS>: if status is 1 it means streaming synthesis is still in progress; if status is 2, it means synthesis is complete.
Client receives audio data frames.
After finishing processing all TEXT or SSML string, TTS server closes the websocket connection.
Sample Client Script (TTS)
Setup
# Python venv setup (recommends using Python 3.8.10)
python3 -m venv venv/seavoice
source venv/seavoice/bin/activate
pip install --upgrade pip
pip install websockets==10.3
pip install aiohttp==3.8.1
pip install PyJWT==2.5.0
Run client script
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2022 Seasalt AI, Inc
"""Client script for tts endpoint
prerequisite:
python 3.8
python package:
- aiohttp==3.8.1
- websockets==10.3
- PyJWT==2.5.0
Usage:
python tts_client.py \
--account test \
--password test \
--lang zh-TW \
--voice Tongtong \
--text "你好這裡是Seasalt，今天的日期是<say-as interpret-as='date' format='m/d/Y'>10/11/2022</say-as>" \
--rules "Seasalt | 海研科技\n"
`--lang`: supports `zh-tw`, `en-us`, `en-gb`
`--text`: input text to synthesize, supports SSML format
`--ssml`: set this to 'true' if the text is in SSML format
`--rules`: optional, globally applied pronunciation rules in the format of `<word> | <pronunciation>\n`
`--pitch`: optional, adjust pitch of synthesized speech, must be > 0.01 or < -0.01
`--speed`: optional, adjust speed of synthesized speech, must be > 1.01 or < 0.99
`--sample-rate`: optional, set the sample rate of synthesized speech
"""
import argparse
import asyncio
import base64
import json
import logging
import wave
from enum import Enum
from pathlib import Path
from urllib.parse import urljoin
import time
import aiohttp
import jwt
import websockets
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s [%(levelname)s] %(message)s",
handlers=[logging.StreamHandler()],
)
SEAAUTH_SCOPE_NAME: str = "seavoice"
TOKEN_TYPE: str = "Bearer"
VOICE_CHANNELS: int = 1
VOICE_SAMPLE_WIDTH: int = 2
ACCESS_TOKEN_LIFE_TIME_MINIMUM_IN_SECOND: int = 60
class Voices(str, Enum):
TONGTONG = "Tongtong"
VIVIAN = "Vivian"
MIKE = "Mike"
MOXIE = "Moxie"
LISSA = "Lissa"
class Language(str, Enum):
EN_US = "en-US"
ZH_TW = "zh-TW"
VOICES_LANGUAGES_MAPPING = {
Voices.TONGTONG: [Language.ZH_TW],
Voices.VIVIAN: [Language.ZH_TW],
Voices.MIKE: [Language.EN_US],
Voices.MOXIE: [Language.EN_US],
Voices.LISSA: [Language.EN_US],
}
async def main(args: argparse.Namespace):
logging.info("Start to get access token.")
access_token = await _get_access_token(args)
await _do_tts(args, access_token)
async def _get_access_token(args: argparse.Namespace) -> str:
credential = _get_credential_from_file(args.seaauth_credential_path)
if credential and credential["account"] == args.account:
access_token, refresh_token = credential["access_token"], credential["refresh_token"]
if _is_access_token_expired(credential["access_token"]):
credential = await _refresh_access_token(access_token, refresh_token)
_save_credential(
args.account, credential["access_token"], credential["refresh_token"], args.seaauth_credential_path
)
else:
logging.info(f"Got access token from {args.seaauth_credential_path}.")
else:
credential = await _login_seaauth(args.account, args.password, args.seaauth_url)
_save_credential(args.account, credential["access_token"], credential["refresh_token"], args.seaauth_credential_path)
return credential["access_token"]
async def _refresh_access_token(access_token: str, refresh_token: str) -> dict:
payload = {
"access_token": access_token,
"refresh_token": refresh_token,
"token_type": TOKEN_TYPE,
"service": SEAAUTH_SCOPE_NAME,
}
logging.info("refresh token...")
async with aiohttp.ClientSession() as session:
async with session.post(urljoin(args.seaauth_url, "/api/v1/users/rotate_token"), json=payload) as response:
if response.status >= 400:
raise Exception(await response.text())
data = await response.json()
logging.info(f"Token is refreshed, auth_result: {data}")
return data
async def _login_seaauth(account: str, password: str, seaauth_url: str) -> dict:
"""Login with SeaAuth.
Example of response:
{
"account": "test",
"access_token": "eyJ0eXAiOi*****",
"token_type": "Bearer",
"refresh_token": "71bbffd5368*****"
}
"""
payload = {"username": account, "password": password, "scope": SEAAUTH_SCOPE_NAME}
data = aiohttp.FormData()
data.add_fields(*payload.items())
async with aiohttp.ClientSession() as session:
async with session.post(urljoin(seaauth_url, "/api/v1/users/login"), data=data) as response:
if response.status >= 400:
raise Exception(await response.text())
data = await response.json()
return data
async def _do_tts(args: argparse.Namespace, access_token: str):
tts_endpoint_url = urljoin(args.seavoice_ws_url, "/api/v1/tts/ws")
logging.info("establishing ws connection...")
async with websockets.connect(tts_endpoint_url) as websocket:
logging.info("established ws connection")
is_begin = asyncio.Event()
is_synthesized = asyncio.Event()
await asyncio.gather(
_receive_events(websocket, is_begin, is_synthesized, args),
_send_commands(websocket, access_token, is_begin, is_synthesized, args),
)
logging.info("tts finished")
async def _send_commands(
websocket,
access_token: str,
is_begin: asyncio.Event,
is_synthesized: asyncio.Event,
args: argparse.Namespace,
):
logging.info("sending authentication command...")
await _send_authentication_command(websocket, access_token, args)
# wait until received the begin event from server
await is_begin.wait()
logging.info("sending synthesis commands...")
await _send_synthesis_commands(websocket, args)
# wait for audio synthsized
logging.info("waiting is_synthesized event...")
await is_synthesized.wait()
await websocket.close()
async def _receive_events(
websocket,
is_begin: asyncio.Event,
is_synthesized: asyncio.Event,
args: argparse.Namespace
):
with wave.open(args.output, "w") as f:
f.setnchannels(VOICE_CHANNELS)
f.setsampwidth(VOICE_SAMPLE_WIDTH)
f.setframerate(args.sample_rate)
async for message in websocket:
event = json.loads(message)
event_name = event.get("event", "")
event_payload = event.get("payload", {})
if event_name == "info":
if event_payload.get("status") == "begin":
logging.info(f"received an info event: {event_payload}")
is_begin.set()
elif event_payload.get("status") == "error":
logging.error(f"received an error event: {event_payload}")
raise Exception(f"received an error event: {event_payload}")
elif event_name == "audio_data":
synthesis_status = event_payload["status"]
logging.info(f"received an audio_data event, staus:{synthesis_status}")
# warning: it's a IO blocking operation.
f.writeframes(base64.b64decode(event_payload["audio"]))
if synthesis_status == "synthesized":
is_synthesized.set()
else:
logging.info(f"received an unknown event: {event}")
async def _send_authentication_command(
websocket,
access_token: str,
args: argparse.Namespace
):
authentication_command = {
"command": "authentication",
"payload": {
"token": access_token,
"settings": {
"language": args.lang,
"voice": args.voice,
},
},
}
command_str = json.dumps(authentication_command)
await websocket.send(command_str)
async def _send_synthesis_commands(websocket, args: argparse.Namespace):
synthesis_command = {
"command": "synthesis",
"payload": {
"settings": {
"pitch": args.pitch,
"speed": args.speed,
"volume": args.volume,
"rules": args.rules,
"sample_rate": args.sample_rate,
},
"data": {"text": args.text, "ssml": args.ssml},
},
}
command_str = json.dumps(synthesis_command)
await websocket.send(command_str)
def _check_voice(args: argparse.Namespace):
if args.lang not in VOICES_LANGUAGES_MAPPING[args.voice]:
raise Exception(
f"{args.voice} only support {','.join(VOICES_LANGUAGES_MAPPING[args.voice])}, the input is {args.lang}."
)
def _convert_argument_str_to_bool(args: argparse.Namespace) -> argparse.Namespace:
args.ssml = args.ssml.lower() == "true"
return args
def _is_access_token_expired(access_token: str) -> bool:
life_time = _get_token_lifetime(access_token)
return life_time < ACCESS_TOKEN_LIFE_TIME_MINIMUM_IN_SECOND
def _get_token_lifetime(access_token: str) -> int:
try:
data = jwt.decode(access_token, options={"verify_signature": False})
return data["exp"] - int(time.time())
except Exception as error:
logging.info(f"Invalid access_token format error:{error}")
def _save_credential(
account: str,
access_token: str,
refresh_token: str,
seaauth_credential_path: str,
):
Path(seaauth_credential_path).touch(exist_ok=True)
with open(seaauth_credential_path, "w") as f:
json.dump({"account": account, "access_token": access_token, "refresh_token": refresh_token}, f)
logging.info(f"The credential is saved to {seaauth_credential_path}.")
def _get_credential_from_file(seaauth_credential_path: str) -> dict:
if not Path(seaauth_credential_path).exists():
logging.info(f"No credential file exists at {seaauth_credential_path}.")
return {}
try:
with open(seaauth_credential_path, "r") as f:
credential = json.load(f)
except Exception as error:
logging.error(f"Cannot parse {seaauth_credential_path} into json due to {error}")
raise error
if "access_token" not in credential or "refresh_token" not in credential:
raise Exception(f"{credential} not includes both access_token and refresh_token.")
return credential
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--account", type=str, required=True, help="account of a SeaAuth account.")
parser.add_argument("--password", type=str, required=True, help="password of a SeaAuth account.")
parser.add_argument(
"--lang",
type=str,
required=True,
choices=[lang for lang in Language],
help='Language of TTS server, must in ["zh-TW", "en-US"]',
)
parser.add_argument(
"--voice",
type=str,
required=True,
choices=[voice for voice in Voices],
help="Voice of the synthesized.",
)
parser.add_argument(
"--text",
type=str,
required=True,
help="Text to synthesize. Supports SSML text.",
)
parser.add_argument(
"--ssml",
type=str,
required=False,
default="false",
help="Set this to true if text is in SSML format.",
)
parser.add_argument(
"--seaauth-url",
type=str,
dest="seaauth_url",
required=False,
default="https://seaauth.seasalt.ai",
help="Url of SeaAuth.",
)
parser.add_argument(
"--seaauth-credential-path",
dest="seaauth_credential_path",
type=str,
required=False,
default="seavoice_credential.json",
help="Credential storage of access token and refresh token.",
)
parser.add_argument(
"--seavoice-ws-url",
type=str,
dest="seavoice_ws_url",
required=False,
default="wss://seavoice.seasalt.ai",
help="Url of SeaVoice.",
)
parser.add_argument(
"--rules",
type=str,
required=False,
default="",
help="Global pronunciation rules.",
)
parser.add_argument(
"--output",
type=str,
default="test_audio.wav",
help="Path to output audio file.",
)
parser.add_argument(
"--sample-rate",
dest="sample_rate",
type=int,
default=22050,
help="Optional, set the sample rate of synthesized speech, default 22050.",
)
parser.add_argument(
"--pitch",
type=float,
default=0.0,
help="Optional, adjust pitch of synthesized speech, [-5, 5] default is 0.",
)
parser.add_argument(
"--speed",
type=float,
default=1.0,
help="Optional, adjust speed of synthesized speech, [0, 2] default is 1.",
)
parser.add_argument(
"--volume",
type=float,
default=50.0,
help="Optional, adjust volume of synthesize speech, [0, 100] default is 50.",
)
args = parser.parse_args()
_check_voice(args)
args = _convert_argument_str_to_bool(args)
asyncio.run(main(args))