Add default argument

pull/1271/head
John Toniutti 3 years ago
parent 05f20b60b8
commit cb775cba9c
No known key found for this signature in database
GPG Key ID: 3C5434E3920EE1E1

@ -78,7 +78,6 @@ vocals = (
class TikTok: class TikTok:
"""TikTok Text-to-Speech Wrapper""" """TikTok Text-to-Speech Wrapper"""
max_chars: Final = 300
BASE_URL: Final = "https://api16-normal-c-useast1a.tiktokv.com/media/api/text/speech/invoke/" BASE_URL: Final = "https://api16-normal-c-useast1a.tiktokv.com/media/api/text/speech/invoke/"
def __init__(self): def __init__(self):
@ -95,8 +94,8 @@ class TikTok:
if random_voice: if random_voice:
voice = self.random_voice() voice = self.random_voice()
else: else:
# if tiktok_voice is not set in the config file, then it will automatically choose one # if tiktok_voice is not set in the config file, then use a random voice
voice = settings.config["settings"]["tts"].get("tiktok_voice") voice = settings.config["settings"]["tts"].get("tiktok_voice", None)
# get the audio from the TikTok API # get the audio from the TikTok API
data = self.get_voices(voice=voice, text=text) data = self.get_voices(voice=voice, text=text)
@ -107,7 +106,7 @@ class TikTok:
raise TikTokTTSException(status_code, data["message"]) raise TikTokTTSException(status_code, data["message"])
# decode data from base64 to binary # decode data from base64 to binary
raw_voices = [data["data"]["v_str"]][0] raw_voices = data["data"]["v_str"]
decoded_voices = base64.b64decode(raw_voices) decoded_voices = base64.b64decode(raw_voices)
# write voices to specified filepath # write voices to specified filepath
@ -117,7 +116,7 @@ class TikTok:
def get_voices(self, text: str, voice: Optional[str] = None) -> dict: def get_voices(self, text: str, voice: Optional[str] = None) -> dict:
"""If voice is not passed, the API will try to use the most fitting voice""" """If voice is not passed, the API will try to use the most fitting voice"""
# sanitize text # sanitize text
text = text.replace("+", "plus").replace(" ", "+").replace("&", "and").replace("r/", "") text = text.replace("+", "plus").replace("&", "and").replace("r/", "")
# prepare url request # prepare url request
params = { params = {

Loading…
Cancel
Save