| from typing import Any, Union |
|
|
| from pydantic import BaseModel |
| from pydub import AudioSegment |
|
|
| from modules.data import styles_mgr |
| from modules.speaker import speaker_mgr |
| from modules.ssml import merge_prompt |
|
|
|
|
| class ParamsTypeError(Exception): |
| pass |
|
|
|
|
| class BaseResponse(BaseModel): |
| message: str |
| data: Any |
|
|
|
|
| def success_response(data: Any, message: str = "ok") -> BaseResponse: |
| return BaseResponse(message=message, data=data) |
|
|
|
|
| def wav_to_mp3(wav_data, bitrate="48k"): |
| audio: AudioSegment = 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, |
| "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["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 |
|
|
