48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import asyncio
|
|
import random
|
|
import sys
|
|
|
|
import edge_tts
|
|
from edge_tts import VoicesManager
|
|
from langdetect import DetectorFactory, detect
|
|
|
|
DetectorFactory.seed = 0
|
|
|
|
TEXT = sys.argv[1]
|
|
LANG = detect(TEXT) if sys.argv[2] == "Auto" else sys.argv[2]
|
|
RATE = sys.argv[3]
|
|
VOLUME = sys.argv[4]
|
|
GENDER = sys.argv[5] if len(sys.argv) == 6 else None
|
|
OUTPUT_FILE = "tts.wav"
|
|
|
|
print("Running TTS...")
|
|
print(f"Text: {TEXT}, Language: {LANG}, Gender: {GENDER}, Rate: {RATE}, Volume: {VOLUME}")
|
|
|
|
async def _main() -> None:
|
|
voices = await VoicesManager.create()
|
|
if GENDER is not None:
|
|
# From "zh-cn" to "zh-CN" etc.
|
|
if LANG == "zh-cn" or LANG == "zh-tw":
|
|
LOCALE = LANG[:-2] + LANG[-2:].upper()
|
|
voice = voices.find(Gender=GENDER, Locale=LOCALE)
|
|
else:
|
|
voice = voices.find(Gender=GENDER, Language=LANG)
|
|
VOICE = random.choice(voice)["Name"]
|
|
print(f"Using random {LANG} voice: {VOICE}")
|
|
else:
|
|
VOICE = LANG
|
|
|
|
communicate = edge_tts.Communicate(text = TEXT, voice = VOICE, rate = RATE, volume = VOLUME)
|
|
await communicate.save(OUTPUT_FILE)
|
|
|
|
if __name__ == "__main__":
|
|
if sys.platform.startswith("win"):
|
|
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
|
asyncio.run(_main())
|
|
else:
|
|
loop = asyncio.get_event_loop_policy().get_event_loop()
|
|
try:
|
|
loop.run_until_complete(_main())
|
|
finally:
|
|
loop.close()
|