mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-23 21:45:00 +03:00
57 lines
2.5 KiB
Python
57 lines
2.5 KiB
Python
import os
|
|
import sys
|
|
import tempfile
|
|
from typing import Literal, TypeAlias
|
|
|
|
|
|
ModelType: TypeAlias = Literal['MMVCv15', 'MMVCv13', 'so-vits-svc-40v2', 'so-vits-svc-40', 'so-vits-svc-40_c', 'DDSP-SVC', 'RVC']
|
|
|
|
ERROR_NO_ONNX_SESSION = "ERROR_NO_ONNX_SESSION"
|
|
|
|
|
|
tmpdir = tempfile.TemporaryDirectory()
|
|
# print("generate tmpdir:::",tmpdir)
|
|
SSL_KEY_DIR = os.path.join(tmpdir.name, "keys") if hasattr(sys, "_MEIPASS") else "keys"
|
|
MODEL_DIR = os.path.join(tmpdir.name, "logs") if hasattr(sys, "_MEIPASS") else "logs"
|
|
UPLOAD_DIR = os.path.join(tmpdir.name, "upload_dir") if hasattr(sys, "_MEIPASS") else "upload_dir"
|
|
NATIVE_CLIENT_FILE_WIN = os.path.join(sys._MEIPASS, "voice-changer-native-client.exe") if hasattr(sys, "_MEIPASS") else "voice-changer-native-client"
|
|
NATIVE_CLIENT_FILE_MAC = os.path.join(sys._MEIPASS, "voice-changer-native-client.app", "Contents", "MacOS",
|
|
"voice-changer-native-client") if hasattr(sys, "_MEIPASS") else "voice-changer-native-client"
|
|
|
|
HUBERT_ONNX_MODEL_PATH = os.path.join(sys._MEIPASS, "model_hubert/hubert_simple.onnx") if hasattr(sys,
|
|
"_MEIPASS") else "model_hubert/hubert_simple.onnx"
|
|
|
|
|
|
TMP_DIR = os.path.join(tmpdir.name, "tmp_dir") if hasattr(sys, "_MEIPASS") else "tmp_dir"
|
|
os.makedirs(TMP_DIR, exist_ok=True)
|
|
|
|
|
|
modelType: ModelType = "MMVCv15"
|
|
|
|
|
|
def getModelType() -> ModelType:
|
|
return modelType
|
|
|
|
|
|
def setModelType(_modelType: ModelType):
|
|
global modelType
|
|
modelType = _modelType
|
|
|
|
|
|
def getFrontendPath():
|
|
frontend_path = os.path.join(sys._MEIPASS, "dist") if hasattr(sys, "_MEIPASS") else "../client/demo/dist"
|
|
# if modelType == "MMVCv15":
|
|
|
|
# elif modelType == "MMVCv13":
|
|
# frontend_path = os.path.join(sys._MEIPASS, "dist_v13") if hasattr(sys, "_MEIPASS") else "../client/demo_v13/dist"
|
|
# elif modelType == "so-vits-svc-40":
|
|
# frontend_path = os.path.join(sys._MEIPASS, "dist_v13") if hasattr(sys, "_MEIPASS") else "../client/demo_v13/dist"
|
|
# elif modelType == "so-vits-svc-40v2":
|
|
# frontend_path = os.path.join(sys._MEIPASS, "dist_v13") if hasattr(sys, "_MEIPASS") else "../client/demo_v13/dist"
|
|
# elif modelType == "DDSP-SVC":
|
|
# frontend_path = os.path.join(sys._MEIPASS, "dist_v13") if hasattr(sys, "_MEIPASS") else "../client/demo_v13/dist"
|
|
# elif modelType == "so-vits-svc-40_c":
|
|
# frontend_path = os.path.join(sys._MEIPASS, "dist_v13") if hasattr(sys, "_MEIPASS") \
|
|
# else "../client/demo_v13/dist"
|
|
return frontend_path
|