mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2025-04-05 04:22:46 +08:00
198 lines
8.6 KiB
Python
198 lines
8.6 KiB
Python
import os
|
|
import multiprocessing
|
|
import Ref_Audio_Selector.config_param.config_params as params
|
|
import Ref_Audio_Selector.tool.audio_inference as audio_inference
|
|
import Ref_Audio_Selector.common.common as common
|
|
|
|
rw_param = params.config_manager.get_rw_param()
|
|
# -------------------基本信息---------------------------
|
|
|
|
# 角色所在工作目录
|
|
base_dir_default = None
|
|
# 工作目录
|
|
text_work_space_dir_default = None
|
|
# 角色名称
|
|
text_role_default = None
|
|
# 参考音频所在目录
|
|
text_refer_audio_file_dir_default = None
|
|
# 推理音频所在目录
|
|
text_inference_audio_file_dir_default = None
|
|
|
|
# -------------------第一步------------------------------
|
|
|
|
# 参考音频抽样目录
|
|
text_sample_dir_default = None
|
|
# 分段数
|
|
slider_subsection_num_default = None
|
|
# 每段随机抽样个数
|
|
slider_sample_num_default = None
|
|
|
|
# -------------------第二步------------------------------
|
|
|
|
# api服务模型切换接口地址
|
|
text_api_set_model_base_url_default = None
|
|
# GPT模型参数名
|
|
text_api_gpt_param_default = None
|
|
# SoVITS模型参数名
|
|
text_api_sovits_param_default = None
|
|
# api服务GPT模型切换接口地址
|
|
text_api_v2_set_gpt_model_base_url_default = None
|
|
# GPT模型参数名
|
|
text_api_v2_gpt_model_param_default = None
|
|
# api服务SoVITS模型切换接口地址
|
|
text_api_v2_set_sovits_model_base_url_default = None
|
|
# SoVITS模型参数名
|
|
text_api_v2_sovits_model_param_default = None
|
|
# 推理服务请求地址与参数
|
|
text_url_default = None
|
|
# 推理服务请求完整地址
|
|
text_whole_url_default = None
|
|
# 文本参数名
|
|
text_text_default = None
|
|
# 参考参数类型
|
|
dropdown_refer_type_param_default = None
|
|
# 参考音频路径参数名
|
|
text_ref_path_default = None
|
|
# 参考音频文本参数名
|
|
text_ref_text_default = None
|
|
# 角色情绪参数名
|
|
text_emotion_default = None
|
|
# 待推理文本路径
|
|
text_test_content_default = None
|
|
# 请求并发数
|
|
slider_request_concurrency_num_default = 3
|
|
# 最大并发数
|
|
slider_request_concurrency_max_num = None
|
|
|
|
# -------------------第三步------------------------------
|
|
|
|
# 待asr的音频所在目录
|
|
text_asr_audio_dir_default = None
|
|
# 待分析的文件路径
|
|
text_text_similarity_analysis_path_default = None
|
|
# 文本相似度放大边界
|
|
slider_text_similarity_amplification_boundary_default = 0.90
|
|
# 文本相似度分析结果文件所在路径
|
|
text_text_similarity_result_path_default = None
|
|
|
|
# -------------------第四步------------------------------
|
|
# -------------------第五步------------------------------
|
|
# 模板内容
|
|
text_template_default = None
|
|
|
|
|
|
def empty_default(vale, default_value):
|
|
if vale is None or vale == "":
|
|
return default_value
|
|
else:
|
|
return vale
|
|
|
|
|
|
def init_base():
|
|
global text_work_space_dir_default, text_role_default, base_dir_default, text_refer_audio_file_dir_default, text_inference_audio_file_dir_default
|
|
|
|
text_work_space_dir_default = rw_param.read(rw_param.work_dir)
|
|
text_role_default = rw_param.read(rw_param.role)
|
|
base_dir_default = os.path.join(text_work_space_dir_default, text_role_default)
|
|
|
|
text_refer_audio_file_dir_default = common.check_path_existence_and_return(
|
|
os.path.join(base_dir_default, params.reference_audio_dir))
|
|
|
|
text_inference_audio_file_dir_default = common.check_path_existence_and_return(
|
|
os.path.join(base_dir_default, params.inference_audio_dir))
|
|
|
|
|
|
def init_first():
|
|
global text_sample_dir_default, slider_subsection_num_default, slider_sample_num_default
|
|
|
|
text_sample_dir_default = common.check_path_existence_and_return(
|
|
os.path.join(base_dir_default, params.list_to_convert_reference_audio_dir))
|
|
|
|
slider_subsection_num_default = int(empty_default(rw_param.read(rw_param.subsection_num), 10))
|
|
|
|
slider_sample_num_default = (empty_default(rw_param.read(rw_param.sample_num), 4))
|
|
|
|
|
|
def init_second():
|
|
global text_api_set_model_base_url_default, text_api_gpt_param_default, text_api_sovits_param_default, text_api_v2_set_gpt_model_base_url_default, text_api_v2_gpt_model_param_default
|
|
global text_api_v2_set_sovits_model_base_url_default, text_api_v2_sovits_model_param_default, text_url_default, text_whole_url_default, text_text_default, dropdown_refer_type_param_default, text_ref_path_default
|
|
global text_ref_text_default, text_emotion_default, text_test_content_default, slider_request_concurrency_num_default, slider_request_concurrency_max_num
|
|
|
|
text_api_set_model_base_url_default = empty_default(rw_param.read(rw_param.api_set_model_base_url),
|
|
'http://localhost:9880/set_model')
|
|
text_api_gpt_param_default = empty_default(rw_param.read(rw_param.api_gpt_param), 'gpt_model_path')
|
|
text_api_sovits_param_default = empty_default(rw_param.read(rw_param.api_sovits_param), 'sovits_model_path')
|
|
|
|
text_api_v2_set_gpt_model_base_url_default = empty_default(rw_param.read(rw_param.api_v2_set_gpt_model_base_url),
|
|
'http://localhost:9880/set_gpt_weights')
|
|
text_api_v2_gpt_model_param_default = empty_default(rw_param.read(rw_param.api_v2_gpt_model_param), 'weights_path')
|
|
|
|
text_api_v2_set_sovits_model_base_url_default = empty_default(
|
|
rw_param.read(rw_param.api_v2_set_sovits_model_base_url), 'http://localhost:9880/set_sovits_weights')
|
|
text_api_v2_sovits_model_param_default = empty_default(rw_param.read(rw_param.api_v2_sovits_model_param), 'weights_path')
|
|
|
|
text_url_default = empty_default(rw_param.read(rw_param.text_url),
|
|
'http://localhost:9880?prompt_language=中文&text_language=中文&cut_punc=,.;?!、，。？！;：…')
|
|
text_text_default = empty_default(rw_param.read(rw_param.text_param), 'text')
|
|
dropdown_refer_type_param_default = empty_default(rw_param.read(rw_param.refer_type_param), '参考音频')
|
|
|
|
text_ref_path_default = empty_default(rw_param.read(rw_param.ref_path_param), 'refer_wav_path')
|
|
text_ref_text_default = empty_default(rw_param.read(rw_param.ref_text_param), 'prompt_text')
|
|
text_emotion_default = empty_default(rw_param.read(rw_param.emotion_param), 'emotion')
|
|
|
|
text_whole_url_default = whole_url(text_url_default, dropdown_refer_type_param_default, text_text_default,
|
|
text_ref_path_default, text_ref_text_default, text_emotion_default)
|
|
|
|
text_test_content_default = empty_default(rw_param.read(rw_param.test_content_path), params.default_test_text_path)
|
|
|
|
slider_request_concurrency_max_num = multiprocessing.cpu_count()
|
|
|
|
slider_request_concurrency_num_default = empty_default(rw_param.read(rw_param.request_concurrency_num), 3)
|
|
|
|
slider_request_concurrency_num_default = min(int(slider_request_concurrency_num_default), slider_request_concurrency_max_num)
|
|
|
|
|
|
# 基于请求路径和参数，合成完整的请求路径
|
|
def whole_url(text_url, dropdown_refer_type_param, text_text, text_ref_path, text_ref_text, text_emotion):
|
|
url_composer = audio_inference.TTSURLComposer(text_url, dropdown_refer_type_param, text_emotion, text_text,
|
|
text_ref_path, text_ref_text)
|
|
if url_composer.is_emotion():
|
|
text_whole_url = url_composer.build_url_with_emotion('测试内容', '情绪类型', False)
|
|
else:
|
|
text_whole_url = url_composer.build_url_with_ref('测试内容', '参考路径', '参考文本', False)
|
|
return text_whole_url
|
|
|
|
|
|
def init_third():
|
|
global text_asr_audio_dir_default, text_text_similarity_analysis_path_default, slider_text_similarity_amplification_boundary_default, text_text_similarity_result_path_default
|
|
|
|
text_asr_audio_dir_default = common.check_path_existence_and_return(
|
|
os.path.join(base_dir_default, params.inference_audio_dir, params.inference_audio_text_aggregation_dir))
|
|
text_text_similarity_analysis_path_default = common.check_path_existence_and_return(
|
|
os.path.join(base_dir_default, params.asr_filename + '.list'))
|
|
slider_text_similarity_amplification_boundary_default = empty_default(
|
|
rw_param.read(rw_param.text_similarity_amplification_boundary), 0.90)
|
|
text_text_similarity_result_path_default = common.check_path_existence_and_return(
|
|
os.path.join(base_dir_default, params.text_emotion_average_similarity_report_filename + '.txt'))
|
|
|
|
|
|
def init_fourth():
|
|
pass
|
|
|
|
|
|
def init_fifth():
|
|
global text_template_default
|
|
|
|
default_template_path = params.default_template_path
|
|
text_template_default = empty_default(rw_param.read(rw_param.text_template),
|
|
common.read_file(default_template_path))
|
|
|
|
|
|
def init_all():
|
|
init_base()
|
|
init_first()
|
|
init_second()
|
|
init_third()
|
|
init_fourth()
|
|
init_fifth()
|