You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
115 lines
3.5 KiB
115 lines
3.5 KiB
from utils import settings
|
|
import requests
|
|
from requests.adapters import HTTPAdapter, Retry
|
|
|
|
from attr import attrs, attrib
|
|
from attr.validators import instance_of
|
|
|
|
from TTS.common import BaseApiTTS, get_random_voice
|
|
|
|
# TTS examples: https://twitter.com/scanlime/status/1512598559769702406
|
|
|
|
voices = dict()
|
|
|
|
voices["nonhuman"] = [ # DISNEY VOICES
|
|
"en_us_ghostface", # Ghost Face
|
|
"en_us_chewbacca", # Chewbacca
|
|
"en_us_c3po", # C3PO
|
|
"en_us_stitch", # Stitch
|
|
"en_us_stormtrooper", # Stormtrooper
|
|
"en_us_rocket", # Rocket
|
|
# ENGLISH VOICES
|
|
]
|
|
voices["human"] = [
|
|
"en_au_001", # English AU - Female
|
|
"en_au_002", # English AU - Male
|
|
"en_uk_001", # English UK - Male 1
|
|
"en_uk_003", # English UK - Male 2
|
|
"en_us_001", # English US - Female (Int. 1)
|
|
"en_us_002", # English US - Female (Int. 2)
|
|
"en_us_006", # English US - Male 1
|
|
"en_us_007", # English US - Male 2
|
|
"en_us_009", # English US - Male 3
|
|
"en_us_010",
|
|
]
|
|
|
|
voices["non_eng"] = [
|
|
"fr_001", # French - Male 1
|
|
"fr_002", # French - Male 2
|
|
"de_001", # German - Female
|
|
"de_002", # German - Male
|
|
"es_002", # Spanish - Male
|
|
# AMERICA VOICES
|
|
"es_mx_002", # Spanish MX - Male
|
|
"br_001", # Portuguese BR - Female 1
|
|
"br_003", # Portuguese BR - Female 2
|
|
"br_004", # Portuguese BR - Female 3
|
|
"br_005", # Portuguese BR - Male
|
|
# ASIA VOICES
|
|
"id_001", # Indonesian - Female
|
|
"jp_001", # Japanese - Female 1
|
|
"jp_003", # Japanese - Female 2
|
|
"jp_005", # Japanese - Female 3
|
|
"jp_006", # Japanese - Male
|
|
"kr_002", # Korean - Male 1
|
|
"kr_003", # Korean - Female
|
|
"kr_004", # Korean - Male 2
|
|
]
|
|
|
|
|
|
# good_voices: 'en_us_002', 'en_us_006'
|
|
# ok: 'en_au_002', 'en_uk_001'
|
|
# less: en_us_stormtrooper
|
|
# more or less: en_us_rocket, en_us_ghostface
|
|
|
|
|
|
@attrs
|
|
class TikTok(BaseApiTTS): # TikTok Text-to-Speech Wrapper
|
|
random_voice: bool = attrib(
|
|
validator=instance_of(bool),
|
|
default=False
|
|
)
|
|
uri_base: str = "https://api16-normal-useast5.us.tiktokv.com/media/api/text/speech/invoke/"
|
|
max_chars: int = 300
|
|
decode_base64: bool = True
|
|
|
|
def make_request(
|
|
self,
|
|
text: str,
|
|
):
|
|
"""
|
|
Makes a requests to remote TTS service
|
|
|
|
Args:
|
|
text: text to be voice over
|
|
|
|
Returns:
|
|
Request's response
|
|
"""
|
|
voice = (
|
|
get_random_voice(voices, "human")
|
|
if self.random_voice
|
|
else str(settings.config["settings"]["tts"]["tiktok_voice"]).lower()
|
|
if str(settings.config["settings"]["tts"]["tiktok_voice"]).lower() in [
|
|
voice.lower() for dict_title in voices for voice in voices[dict_title]]
|
|
else get_random_voice(voices, "human")
|
|
)
|
|
try:
|
|
r = requests.post(
|
|
self.uri_base,
|
|
params={
|
|
"text_speaker": voice,
|
|
"req_text": text,
|
|
"speaker_map_type": 0,
|
|
})
|
|
except requests.exceptions.SSLError:
|
|
# https://stackoverflow.com/a/47475019/18516611
|
|
session = requests.Session()
|
|
retry = Retry(connect=3, backoff_factor=0.5)
|
|
adapter = HTTPAdapter(max_retries=retry)
|
|
session.mount("http://", adapter)
|
|
session.mount("https://", adapter)
|
|
r = session.post(f"{self.uri_base}{voice}&req_text={text}&speaker_map_type=0")
|
|
# print(r.text)
|
|
return r.json()["data"]["v_str"]
|