+2
902ff00cb0
902ff00cb0
Co-authored-by: Jason Cameron <git@jasoncameron.dev> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: github-actions <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: RequieMa <57754488+RequieMa@users.noreply.github.com> Co-authored-by: prafful <praffulsharma1230@gmail.com> Co-authored-by: M. Sanaullah <thebolly@gmail.com> Co-authored-by: embee <emilien.bev.com@gmail.com> Co-authored-by: Rodrigo <55567123+rodrigodasilv@users.noreply.github.com> Co-authored-by: Emilien Bevierre <44171454+emilienbev@users.noreply.github.com> Co-authored-by: tkhmielnitzky <tkhmielnitzky@gmail.com> Co-authored-by: bnfone <89687390+bnfone@users.noreply.github.com> Co-authored-by: Cyteon <129582290+Cyteon@users.noreply.github.com>
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from typing import Tuple
|
|
|
|
from rich.console import Console
|
|
|
|
from TTS.aws_polly import AWSPolly
|
|
from TTS.elevenlabs import elevenlabs
|
|
from TTS.engine_wrapper import TTSEngine
|
|
from TTS.GTTS import GTTS
|
|
from TTS.openai_tts import OpenAITTS
|
|
from TTS.pyttsx import pyttsx
|
|
from TTS.streamlabs_polly import StreamlabsPolly
|
|
from TTS.TikTok import TikTok
|
|
from utils import settings
|
|
from utils.console import print_step, print_table
|
|
|
|
console = Console()
|
|
|
|
TTSProviders = {
|
|
"GoogleTranslate": GTTS,
|
|
"AWSPolly": AWSPolly,
|
|
"StreamlabsPolly": StreamlabsPolly,
|
|
"TikTok": TikTok,
|
|
"pyttsx": pyttsx,
|
|
"ElevenLabs": elevenlabs,
|
|
"OpenAI": OpenAITTS,
|
|
}
|
|
|
|
|
|
def save_text_to_mp3(reddit_obj) -> Tuple[int, int]:
|
|
"""Saves text to MP3 files.
|
|
|
|
Args:
|
|
reddit_obj (): Reddit object received from reddit API in reddit/subreddit.py
|
|
|
|
Returns:
|
|
tuple[int,int]: (total length of the audio, the number of comments audio was generated for)
|
|
"""
|
|
|
|
voice = settings.config["settings"]["tts"]["voice_choice"]
|
|
if str(voice).casefold() in map(lambda _: _.casefold(), TTSProviders):
|
|
text_to_mp3 = TTSEngine(get_case_insensitive_key_value(TTSProviders, voice), reddit_obj)
|
|
else:
|
|
while True:
|
|
print_step("Please choose one of the following TTS providers: ")
|
|
print_table(TTSProviders)
|
|
choice = input("\n")
|
|
if choice.casefold() in map(lambda _: _.casefold(), TTSProviders):
|
|
break
|
|
print("Unknown Choice")
|
|
text_to_mp3 = TTSEngine(get_case_insensitive_key_value(TTSProviders, choice), reddit_obj)
|
|
return text_to_mp3.run()
|
|
|
|
|
|
def get_case_insensitive_key_value(input_dict, key):
|
|
return next(
|
|
(value for dict_key, value in input_dict.items() if dict_key.lower() == key.lower()),
|
|
None,
|
|
)
|