forked from killua/TakwayDisplayPlatform
464 lines
16 KiB
Python
464 lines
16 KiB
Python
|
import gc
|
|||
|
|
import os
|
|||
|
|
import numpy as np
|
|||
|
|
import torch
|
|||
|
|
from torch import LongTensor
|
|||
|
|
from typing import Optional
|
|||
|
|
import soundfile as sf
|
|||
|
|
import logging
|
|||
|
|
import gradio as gr
|
|||
|
|
import librosa
|
|||
|
|
# bert_vits2
|
|||
|
|
from .bert_vits2 import utils
|
|||
|
|
from .bert_vits2.infer import get_net_g, latest_version, infer_multilang, infer
|
|||
|
|
from .bert_vits2.config import config
|
|||
|
|
from .bert_vits2 import re_matching
|
|||
|
|
from .bert_vits2.tools.sentence import split_by_language
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
class TextToSpeech:
|
|||
|
|
def __init__(self,
|
|||
|
|
device='cuda',
|
|||
|
|
):
|
|||
|
|
self.device = device = torch.device(device)
|
|||
|
|
|
|||
|
|
if config.webui_config.debug:
|
|||
|
|
logger.info("Enable DEBUG")
|
|||
|
|
|
|||
|
|
hps = utils.get_hparams_from_file(config.webui_config.config_path)
|
|||
|
|
self.hps = hps
|
|||
|
|
# 若config.json中未指定版本则默认为最新版本
|
|||
|
|
version = hps.version if hasattr(hps, "version") else latest_version
|
|||
|
|
self.version = version
|
|||
|
|
|
|||
|
|
net_g = get_net_g(
|
|||
|
|
model_path=config.webui_config.model, version=version, device=device, hps=hps
|
|||
|
|
)
|
|||
|
|
self.net_g = net_g
|
|||
|
|
|
|||
|
|
self.speaker_ids = speaker_ids = hps.data.spk2id
|
|||
|
|
self.speakers = speakers = list(speaker_ids.keys())
|
|||
|
|
self.speaker = speakers[0]
|
|||
|
|
|
|||
|
|
self.languages = languages = ["ZH", "JP", "EN", "mix", "auto"]
|
|||
|
|
|
|||
|
|
def free_up_memory(self):
|
|||
|
|
# Prior inference run might have large variables not cleaned up due to exception during the run.
|
|||
|
|
# Free up as much memory as possible to allow this run to be successful.
|
|||
|
|
gc.collect()
|
|||
|
|
if torch.cuda.is_available():
|
|||
|
|
torch.cuda.empty_cache()
|
|||
|
|
|
|||
|
|
def process_mix(self, slice):
|
|||
|
|
_speaker = slice.pop()
|
|||
|
|
_text, _lang = [], []
|
|||
|
|
for lang, content in slice:
|
|||
|
|
content = content.split("|")
|
|||
|
|
content = [part for part in content if part != ""]
|
|||
|
|
if len(content) == 0:
|
|||
|
|
continue
|
|||
|
|
if len(_text) == 0:
|
|||
|
|
_text = [[part] for part in content]
|
|||
|
|
_lang = [[lang] for part in content]
|
|||
|
|
else:
|
|||
|
|
_text[-1].append(content[0])
|
|||
|
|
_lang[-1].append(lang)
|
|||
|
|
if len(content) > 1:
|
|||
|
|
_text += [[part] for part in content[1:]]
|
|||
|
|
_lang += [[lang] for part in content[1:]]
|
|||
|
|
return _text, _lang, _speaker
|
|||
|
|
|
|||
|
|
def process_auto(self, text):
|
|||
|
|
_text, _lang = [], []
|
|||
|
|
for slice in text.split("|"):
|
|||
|
|
if slice == "":
|
|||
|
|
continue
|
|||
|
|
temp_text, temp_lang = [], []
|
|||
|
|
sentences_list = split_by_language(slice, target_languages=["zh", "ja", "en"])
|
|||
|
|
for sentence, lang in sentences_list:
|
|||
|
|
if sentence == "":
|
|||
|
|
continue
|
|||
|
|
temp_text.append(sentence)
|
|||
|
|
if lang == "ja":
|
|||
|
|
lang = "jp"
|
|||
|
|
temp_lang.append(lang.upper())
|
|||
|
|
_text.append(temp_text)
|
|||
|
|
_lang.append(temp_lang)
|
|||
|
|
return _text, _lang
|
|||
|
|
|
|||
|
|
def generate_audio(
|
|||
|
|
self,
|
|||
|
|
slices,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
speaker,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
skip_start=False,
|
|||
|
|
skip_end=False,
|
|||
|
|
):
|
|||
|
|
audio_list = []
|
|||
|
|
# silence = np.zeros(hps.data.sampling_rate // 2, dtype=np.int16)
|
|||
|
|
|
|||
|
|
self.free_up_memory()
|
|||
|
|
|
|||
|
|
with torch.no_grad():
|
|||
|
|
for idx, piece in enumerate(slices):
|
|||
|
|
skip_start = idx != 0
|
|||
|
|
skip_end = idx != len(slices) - 1
|
|||
|
|
audio = infer(
|
|||
|
|
piece,
|
|||
|
|
reference_audio=reference_audio,
|
|||
|
|
emotion=emotion,
|
|||
|
|
sdp_ratio=sdp_ratio,
|
|||
|
|
noise_scale=noise_scale,
|
|||
|
|
noise_scale_w=noise_scale_w,
|
|||
|
|
length_scale=length_scale,
|
|||
|
|
sid=speaker,
|
|||
|
|
language=language,
|
|||
|
|
hps=self.hps,
|
|||
|
|
net_g=self.net_g,
|
|||
|
|
device=self.device,
|
|||
|
|
skip_start=skip_start,
|
|||
|
|
skip_end=skip_end,
|
|||
|
|
style_text=style_text,
|
|||
|
|
style_weight=style_weight,
|
|||
|
|
)
|
|||
|
|
audio16bit = gr.processing_utils.convert_to_16_bit_wav(audio)
|
|||
|
|
audio_list.append(audio16bit)
|
|||
|
|
return audio_list
|
|||
|
|
|
|||
|
|
|
|||
|
|
def generate_audio_multilang(
|
|||
|
|
self,
|
|||
|
|
slices,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
speaker,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
skip_start=False,
|
|||
|
|
skip_end=False,
|
|||
|
|
en_ratio=1.0
|
|||
|
|
):
|
|||
|
|
audio_list = []
|
|||
|
|
# silence = np.zeros(hps.data.sampling_rate // 2, dtype=np.int16)
|
|||
|
|
|
|||
|
|
self.free_up_memory()
|
|||
|
|
|
|||
|
|
with torch.no_grad():
|
|||
|
|
for idx, piece in enumerate(slices):
|
|||
|
|
skip_start = idx != 0
|
|||
|
|
skip_end = idx != len(slices) - 1
|
|||
|
|
audio = infer_multilang(
|
|||
|
|
piece,
|
|||
|
|
reference_audio=reference_audio,
|
|||
|
|
emotion=emotion,
|
|||
|
|
sdp_ratio=sdp_ratio,
|
|||
|
|
noise_scale=noise_scale,
|
|||
|
|
noise_scale_w=noise_scale_w,
|
|||
|
|
length_scale=length_scale,
|
|||
|
|
sid=speaker,
|
|||
|
|
language=language[idx],
|
|||
|
|
hps=self.hps,
|
|||
|
|
net_g=self.net_g,
|
|||
|
|
device=self.device,
|
|||
|
|
skip_start=skip_start,
|
|||
|
|
skip_end=skip_end,
|
|||
|
|
en_ratio=en_ratio
|
|||
|
|
)
|
|||
|
|
audio16bit = gr.processing_utils.convert_to_16_bit_wav(audio)
|
|||
|
|
audio_list.append(audio16bit)
|
|||
|
|
return audio_list
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def process_text(self,
|
|||
|
|
text: str,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text=None,
|
|||
|
|
style_weight=0,
|
|||
|
|
en_ratio=1.0
|
|||
|
|
):
|
|||
|
|
hps = self.hps
|
|||
|
|
audio_list = []
|
|||
|
|
if language == "mix":
|
|||
|
|
bool_valid, str_valid = re_matching.validate_text(text)
|
|||
|
|
if not bool_valid:
|
|||
|
|
return str_valid, (
|
|||
|
|
hps.data.sampling_rate,
|
|||
|
|
np.concatenate([np.zeros(hps.data.sampling_rate // 2)]),
|
|||
|
|
)
|
|||
|
|
for slice in re_matching.text_matching(text):
|
|||
|
|
_text, _lang, _speaker = self.process_mix(slice)
|
|||
|
|
if _speaker is None:
|
|||
|
|
continue
|
|||
|
|
print(f"Text: {_text}\nLang: {_lang}")
|
|||
|
|
audio_list.extend(
|
|||
|
|
self.generate_audio_multilang(
|
|||
|
|
_text,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
_speaker,
|
|||
|
|
_lang,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
en_ratio=en_ratio
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
elif language.lower() == "auto":
|
|||
|
|
_text, _lang = self.process_auto(text)
|
|||
|
|
print(f"Text: {_text}\nLang: {_lang}")
|
|||
|
|
audio_list.extend(
|
|||
|
|
self.generate_audio_multilang(
|
|||
|
|
_text,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
speaker,
|
|||
|
|
_lang,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
en_ratio=en_ratio
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
audio_list.extend(
|
|||
|
|
self.generate_audio(
|
|||
|
|
text.split("|"),
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
speaker,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
return audio_list
|
|||
|
|
|
|||
|
|
def tts_split(
|
|||
|
|
self,
|
|||
|
|
text: str,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
cut_by_sent,
|
|||
|
|
interval_between_para,
|
|||
|
|
interval_between_sent,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
en_ratio
|
|||
|
|
):
|
|||
|
|
while text.find("\n\n") != -1:
|
|||
|
|
text = text.replace("\n\n", "\n")
|
|||
|
|
text = text.replace("|", "")
|
|||
|
|
para_list = re_matching.cut_para(text)
|
|||
|
|
para_list = [p for p in para_list if p != ""]
|
|||
|
|
audio_list = []
|
|||
|
|
for p in para_list:
|
|||
|
|
if not cut_by_sent:
|
|||
|
|
audio_list += self.process_text(
|
|||
|
|
p,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
en_ratio
|
|||
|
|
)
|
|||
|
|
silence = np.zeros((int)(44100 * interval_between_para), dtype=np.int16)
|
|||
|
|
audio_list.append(silence)
|
|||
|
|
else:
|
|||
|
|
audio_list_sent = []
|
|||
|
|
sent_list = re_matching.cut_sent(p)
|
|||
|
|
sent_list = [s for s in sent_list if s != ""]
|
|||
|
|
for s in sent_list:
|
|||
|
|
audio_list_sent += self.process_text(
|
|||
|
|
s,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
en_ratio
|
|||
|
|
)
|
|||
|
|
silence = np.zeros((int)(44100 * interval_between_sent))
|
|||
|
|
audio_list_sent.append(silence)
|
|||
|
|
if (interval_between_para - interval_between_sent) > 0:
|
|||
|
|
silence = np.zeros(
|
|||
|
|
(int)(44100 * (interval_between_para - interval_between_sent))
|
|||
|
|
)
|
|||
|
|
audio_list_sent.append(silence)
|
|||
|
|
audio16bit = gr.processing_utils.convert_to_16_bit_wav(
|
|||
|
|
np.concatenate(audio_list_sent)
|
|||
|
|
) # 对完整句子做音量归一
|
|||
|
|
audio_list.append(audio16bit)
|
|||
|
|
audio_concat = np.concatenate(audio_list)
|
|||
|
|
return ("Success", (self.hps.data.sampling_rate, audio_concat))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def tts_fn(
|
|||
|
|
self,
|
|||
|
|
text: str,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
prompt_mode,
|
|||
|
|
style_text=None,
|
|||
|
|
style_weight=0,
|
|||
|
|
):
|
|||
|
|
if style_text == "":
|
|||
|
|
style_text = None
|
|||
|
|
if prompt_mode == "Audio prompt":
|
|||
|
|
if reference_audio == None:
|
|||
|
|
return ("Invalid audio prompt", None)
|
|||
|
|
else:
|
|||
|
|
reference_audio = self.load_audio(reference_audio)[1]
|
|||
|
|
else:
|
|||
|
|
reference_audio = None
|
|||
|
|
|
|||
|
|
audio_list = self.process_text(
|
|||
|
|
text,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
reference_audio,
|
|||
|
|
emotion,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
audio_concat = np.concatenate(audio_list)
|
|||
|
|
return "Success", (self.hps.data.sampling_rate, audio_concat)
|
|||
|
|
|
|||
|
|
def load_audio(self, path):
|
|||
|
|
audio, sr = librosa.load(path, 48000)
|
|||
|
|
# audio = librosa.resample(audio, 44100, 48000)
|
|||
|
|
return sr, audio
|
|||
|
|
|
|||
|
|
def format_utils(self, text, speaker):
|
|||
|
|
_text, _lang = self.process_auto(text)
|
|||
|
|
res = f"[{speaker}]"
|
|||
|
|
for lang_s, content_s in zip(_lang, _text):
|
|||
|
|
for lang, content in zip(lang_s, content_s):
|
|||
|
|
# res += f"<{lang.lower()}>{content}"
|
|||
|
|
# 部分中文会被识别成日文，强转成中文
|
|||
|
|
lang = lang.lower().replace("jp", "zh")
|
|||
|
|
res += f"<{lang}>{content}"
|
|||
|
|
res += "|"
|
|||
|
|
return "mix", res[:-1]
|
|||
|
|
|
|||
|
|
def synthesize(self,
|
|||
|
|
text,
|
|||
|
|
speaker_idx=0, # self.speakers 的 index，指定说话
|
|||
|
|
sdp_ratio=0.5,
|
|||
|
|
noise_scale=0.6,
|
|||
|
|
noise_scale_w=0.9,
|
|||
|
|
length_scale=1.0, # 越大语速越慢
|
|||
|
|
language="mix", # ["ZH", "EN", "mix"] 三选一
|
|||
|
|
opt_cut_by_send=False, # 按句切分 在按段落切分的基础上再按句子切分文本
|
|||
|
|
interval_between_para=1.0, # 段间停顿(秒)，需要大于句间停顿才有效
|
|||
|
|
interval_between_sent=0.2, # 句间停顿(秒)，勾选按句切分才生效
|
|||
|
|
audio_prompt=None,
|
|||
|
|
text_prompt="",
|
|||
|
|
prompt_mode="Text prompts",
|
|||
|
|
style_text="", # "使用辅助文本的语意来辅助生成对话（语言保持与主文本相同）\n\n"
|
|||
|
|
# "**注意**：不要使用**指令式文本**（如：开心），要使用**带有强烈情感的文本**（如：我好快乐！！！）\n\n"
|
|||
|
|
# "效果较不明确，留空即为不使用该功能"
|
|||
|
|
style_weight=0.7, # "主文本和辅助文本的bert混合比率，0表示仅主文本，1表示仅辅助文本
|
|||
|
|
en_ratio=1.0 # 中英混合时，英文速度控制，越大英文速度越慢
|
|||
|
|
):
|
|||
|
|
"""
|
|||
|
|
return: audio, sample_rate
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
speaker = self.speakers[speaker_idx]
|
|||
|
|
|
|||
|
|
if language == "mix":
|
|||
|
|
language, text = self.format_utils(text, speaker)
|
|||
|
|
|
|||
|
|
text_output, audio_output = self.tts_split(
|
|||
|
|
text,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
opt_cut_by_send,
|
|||
|
|
interval_between_para,
|
|||
|
|
interval_between_sent,
|
|||
|
|
audio_prompt,
|
|||
|
|
text_prompt,
|
|||
|
|
style_text,
|
|||
|
|
style_weight,
|
|||
|
|
en_ratio
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
text_output, audio_output = self.tts_fn(
|
|||
|
|
text,
|
|||
|
|
speaker,
|
|||
|
|
sdp_ratio,
|
|||
|
|
noise_scale,
|
|||
|
|
noise_scale_w,
|
|||
|
|
length_scale,
|
|||
|
|
language,
|
|||
|
|
audio_prompt,
|
|||
|
|
text_prompt,
|
|||
|
|
prompt_mode,
|
|||
|
|
style_text,
|
|||
|
|
style_weight
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# return text_output, audio_output
|
|||
|
|
return audio_output[1], audio_output[0]
|
|||
|
|
|
|||
|
|
def print_speakers_info(self):
|
|||
|
|
for i, speaker in enumerate(self.speakers):
|
|||
|
|
print(f"id: {i}, speaker: {speaker}")
|