mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-23 21:45:00 +03:00
38 lines
933 B
Python
38 lines
933 B
Python
from dataclasses import dataclass, field
|
|
from typing import TypeAlias, Union, Any
|
|
from const import VoiceChangerType
|
|
|
|
|
|
@dataclass
|
|
class ModelSample:
|
|
id: str = ""
|
|
voiceChangerType: VoiceChangerType | None = None
|
|
|
|
|
|
@dataclass
|
|
class RVCModelSample(ModelSample):
|
|
voiceChangerType: VoiceChangerType = "RVC"
|
|
lang: str = ""
|
|
tag: list[str] = field(default_factory=lambda: [])
|
|
name: str = ""
|
|
modelUrl: str = ""
|
|
indexUrl: str = ""
|
|
termsOfUseUrl: str = ""
|
|
icon: str = ""
|
|
credit: str = ""
|
|
description: str = ""
|
|
|
|
sampleRate: int = 48000
|
|
modelType: str = ""
|
|
f0: bool = True
|
|
|
|
|
|
ModelSamples: TypeAlias = Union[ModelSample, RVCModelSample]
|
|
|
|
|
|
def generateModelSample(params: Any) -> ModelSamples:
|
|
if params["voiceChangerType"] == "RVC":
|
|
return RVCModelSample(**params)
|
|
else:
|
|
return ModelSample(**{k: v for k, v in params.items() if k in ModelSample.__annotations__})
|