import sys
import os
# By using XTTS you agree to CPML license https://coqui.ai/cpml
os.environ["COQUI_TOS_AGREED"] = "1"
import langid
import gradio as gr
from TTS.api import TTS
HF_TOKEN = os.environ.get("HF_TOKEN")
from huggingface_hub import HfApi
# will use api to restart space on a unrecoverable error
api = HfApi(token=HF_TOKEN)
repo_id = "coqui/xtts"
tts = TTS("tts_models/multilingual/multi-dataset/xtts_v1")
tts.to("cuda")
DEVICE_ASSERT_DETECTED=0
DEVICE_ASSERT_PROMPT=None
DEVICE_ASSERT_LANG=None
def predict(prompt, language, audio_file_pth, mic_file_path, use_mic, agree):
if agree == True:
supported_languages=["en","es","fr","de","it","pt","pl","tr","ru","nl","cs","ar","zh-cn"]
if language not in supported_languages:
gr.Warning("Language you put in is not in is not in our Supported Languages, please choose from dropdown")
return (
None,
None,
)
language_predicted=langid.classify(prompt)[0].strip() # strip need as there is space at end!
if language_predicted == "zh":
#we use zh-cn
language_predicted = "zh-cn"
print(f"Detected language:{language_predicted}, Chosen language:{language}")
if len(prompt)>10:
#allow any language for short text as some may be common
if language_predicted != language:
#Please duplicate and remove this check if you really want this
#Or auto-detector fails to identify language (which it can on pretty short text or mixed text)
gr.Warning(f"Auto-Predicted Language in prompt (detected: {language_predicted}) does not match language you chose (chosen: {language}) , please choose correct language id. If you think this is incorrect please duplicate this space and modify code.")
return (
None,
None,
)
if use_mic == True:
if mic_file_path is not None:
speaker_wav=mic_file_path
else:
gr.Warning("Please record your voice with Microphone, or uncheck Use Microphone to use reference audios")
return (
None,
None,
)
else:
speaker_wav=audio_file_pth
if len(prompt)<2:
gr.Warning("Please give a longer prompt text")
return (
None,
None,
)
if len(prompt)>200:
gr.Warning("Text length limited to 200 characters for this demo, please try shorter text")
return (
None,
None,
)
global DEVICE_ASSERT_DETECTED
if DEVICE_ASSERT_DETECTED:
global DEVICE_ASSERT_PROMPT
global DEVICE_ASSERT_LANG
#we want to know whos caused this error
print(f"Unrecoverable exception caused by language:{DEVICE_ASSERT_LANG} prompt:{DEVICE_ASSERT_PROMPT}")
try:
tts.tts_to_file(
text=prompt,
file_path="output.wav",
speaker_wav=speaker_wav,
language=language,
)
except RuntimeError as e :
if "device-side assert" in str(e):
# cannot do anything on cuda device side error, need tor estart
print(f"Exit due to: Unrecoverable exception caused by language:{language} prompt:{prompt}", flush=True)
gr.Warning("Unhandled Exception encounter, please retry in a minute")
print("Cuda device-assert Runtime encountered need restart")
if not DEVICE_ASSERT_DETECTED:
DEVICE_ASSERT_DETECTED=1
DEVICE_ASSERT_PROMPT=prompt
DEVICE_ASSERT_LANG=language
# HF Space specific.. This error is unrecoverable need to restart space
api.restart_space(repo_id=repo_id)
else:
print("RuntimeError: non device-side assert error:", str(e))
raise e
return (
gr.make_waveform(
audio="output.wav",
),
"output.wav",
)
else:
gr.Warning("Please accept the Terms & Condition!")
return (
None,
None,
)
title = "Coqui🐸 XTTS"
description = """
XTTS is a Voice generation model that lets you clone voices into different languages by using just a quick 3-second audio clip.
XTTS is built on previous research, like Tortoise, with additional architectural innovations and training to make cross-language voice cloning and multilingual speech generation possible.
This is the same model that powers our creator application Coqui Studio as well as the Coqui API. In production we apply modifications to make low-latency streaming possible.
Leave a star on the Github 🐸TTS, where our open-source inference and training code lives.
For faster inference without waiting in the queue, you should duplicate this space and upgrade to GPU via the settings.
Language Selectors:
Arabic: ar, Brazilian Portuguese: pt , Chinese: zh-cn, Czech: cs,
Dutch: nl, English: en, French: fr, Italian: it, Polish: pl,
Russian: ru, Spanish: es, Turkish: tr
By using this demo you agree to the terms of the Coqui Public Model License at https://coqui.ai/cpml