|
from pydantic import BaseModel |
|
from typing import Any, Union |
|
|
|
import torch |
|
|
|
from modules.speaker import Speaker, speaker_mgr |
|
|
|
|
|
from modules.data import styles_mgr |
|
|
|
from pydub import AudioSegment |
|
|
|
from modules.ssml import merge_prompt |
|
|
|
|
|
from enum import Enum |
|
|
|
|
|
class ParamsTypeError(Exception): |
|
pass |
|
|
|
|
|
class AudioFormat(str, Enum): |
|
mp3 = "mp3" |
|
wav = "wav" |
|
|
|
|
|
class BaseResponse(BaseModel): |
|
message: str |
|
data: Any |
|
|
|
class Config: |
|
json_encoders = { |
|
torch.Tensor: lambda v: v.tolist(), |
|
Speaker: lambda v: v.to_json(), |
|
} |
|
|
|
|
|
def success_response(data: Any, message: str = "Success") -> BaseResponse: |
|
return BaseResponse(message=message, data=data) |
|
|
|
|
|
def wav_to_mp3(wav_data, bitrate="48k"): |
|
audio = AudioSegment.from_wav( |
|
wav_data, |
|
) |
|
return audio.export(format="mp3", bitrate=bitrate) |
|
|
|
|
|
def to_number(value, t, default=0): |
|
try: |
|
number = t(value) |
|
return number |
|
except (ValueError, TypeError) as e: |
|
return default |
|
|
|
|
|
def calc_spk_style(spk: Union[str, int], style: Union[str, int]): |
|
voice_attrs = { |
|
"spk": None, |
|
"seed": None, |
|
"prompt1": None, |
|
"prompt2": None, |
|
"prefix": None, |
|
"temperature": None, |
|
} |
|
params = {} |
|
|
|
if type(spk) == int: |
|
voice_attrs["spk"] = spk |
|
elif type(spk) == str: |
|
if spk.isdigit(): |
|
voice_attrs["spk"] = int(spk) |
|
else: |
|
spker = speaker_mgr.get_speaker(spk) |
|
if spker: |
|
voice_attrs["spk"] = spker |
|
|
|
if type(style) == int or type(style) == float: |
|
raise ParamsTypeError("The style parameter cannot be a number.") |
|
elif type(style) == str and style != "": |
|
if style.isdigit(): |
|
raise ParamsTypeError("The style parameter cannot be a number.") |
|
else: |
|
style_params = styles_mgr.find_params_by_name(style) |
|
for k, v in style_params.items(): |
|
params[k] = v |
|
|
|
voice_attrs = {k: v for k, v in voice_attrs.items() if v is not None} |
|
|
|
merge_prompt(voice_attrs, params) |
|
|
|
voice_attrs["spk"] = params.get("spk", voice_attrs.get("spk", None)) |
|
voice_attrs["seed"] = params.get("seed", voice_attrs.get("seed", None)) |
|
voice_attrs["temperature"] = params.get( |
|
"temp", voice_attrs.get("temperature", None) |
|
) |
|
voice_attrs["prefix"] = params.get("prefix", voice_attrs.get("prefix", None)) |
|
voice_attrs["prompt1"] = params.get("prompt1", voice_attrs.get("prompt1", None)) |
|
voice_attrs["prompt2"] = params.get("prompt2", voice_attrs.get("prompt2", None)) |
|
|
|
if voice_attrs.get("temperature", "") == "min": |
|
|
|
voice_attrs["temperature"] = 0.000000000001 |
|
if voice_attrs.get("temperature", "") == "max": |
|
voice_attrs["temperature"] = 1 |
|
|
|
voice_attrs = {k: v for k, v in voice_attrs.items() if v is not None} |
|
|
|
|
|
return voice_attrs |
|
|
