| import logging
|
| import os
|
|
|
|
|
| import gradio as gr
|
| from dotenv import load_dotenv
|
|
|
| from configs.config import Config
|
| from i18n.i18n import I18nAuto
|
| from infer.modules.vc.modules import VC
|
|
|
| logging.getLogger("numba").setLevel(logging.WARNING)
|
| logging.getLogger("markdown_it").setLevel(logging.WARNING)
|
| logging.getLogger("urllib3").setLevel(logging.WARNING)
|
| logging.getLogger("matplotlib").setLevel(logging.WARNING)
|
| logger = logging.getLogger(__name__)
|
|
|
| i18n = I18nAuto()
|
| logger.info(i18n)
|
|
|
| load_dotenv()
|
| config = Config()
|
| vc = VC(config)
|
|
|
| weight_root = os.getenv("weight_root")
|
| weight_uvr5_root = os.getenv("weight_uvr5_root")
|
| index_root = os.getenv("index_root")
|
| names = []
|
| hubert_model = None
|
| for name in os.listdir(weight_root):
|
| if name.endswith(".pth"):
|
| names.append(name)
|
| index_paths = []
|
| for root, dirs, files in os.walk(index_root, topdown=False):
|
| for name in files:
|
| if name.endswith(".index") and "trained" not in name:
|
| index_paths.append("%s/%s" % (root, name))
|
|
|
|
|
| app = gr.Blocks()
|
| with app:
|
| with gr.Tabs():
|
| with gr.TabItem("在线demo"):
|
| gr.Markdown(
|
| value="""
|
| RVC 在线demo
|
| """
|
| )
|
| sid = gr.Dropdown(label=i18n("推理音色"), choices=sorted(names))
|
| with gr.Column():
|
| spk_item = gr.Slider(
|
| minimum=0,
|
| maximum=2333,
|
| step=1,
|
| label=i18n("请选择说话人id"),
|
| value=0,
|
| visible=False,
|
| interactive=True,
|
| )
|
| sid.change(fn=vc.get_vc, inputs=[sid], outputs=[spk_item])
|
| gr.Markdown(
|
| value=i18n(
|
| "男转女推荐+12key, 女转男推荐-12key, 如果音域爆炸导致音色失真也可以自己调整到合适音域. "
|
| )
|
| )
|
| vc_input3 = gr.Audio(label="上传音频（长度小于90秒）")
|
| vc_transform0 = gr.Number(
|
| label=i18n("变调(整数, 半音数量, 升八度12降八度-12)"), value=0
|
| )
|
| f0method0 = gr.Radio(
|
| label=i18n(
|
| "选择音高提取算法,输入歌声可用pm提速,harvest低音好但巨慢无比,crepe效果好但吃GPU"
|
| ),
|
| choices=["pm", "harvest", "crepe", "rmvpe"],
|
| value="pm",
|
| interactive=True,
|
| )
|
| filter_radius0 = gr.Slider(
|
| minimum=0,
|
| maximum=7,
|
| label=i18n(
|
| ">=3则使用对harvest音高识别的结果使用中值滤波，数值为滤波半径，使用可以削弱哑音"
|
| ),
|
| value=3,
|
| step=1,
|
| interactive=True,
|
| )
|
| with gr.Column():
|
| file_index1 = gr.Textbox(
|
| label=i18n("特征检索库文件路径,为空则使用下拉的选择结果"),
|
| value="",
|
| interactive=False,
|
| visible=False,
|
| )
|
| file_index2 = gr.Dropdown(
|
| label=i18n("自动检测index路径,下拉式选择(dropdown)"),
|
| choices=sorted(index_paths),
|
| interactive=True,
|
| )
|
| index_rate1 = gr.Slider(
|
| minimum=0,
|
| maximum=1,
|
| label=i18n("检索特征占比"),
|
| value=0.88,
|
| interactive=True,
|
| )
|
| resample_sr0 = gr.Slider(
|
| minimum=0,
|
| maximum=48000,
|
| label=i18n("后处理重采样至最终采样率，0为不进行重采样"),
|
| value=0,
|
| step=1,
|
| interactive=True,
|
| )
|
| rms_mix_rate0 = gr.Slider(
|
| minimum=0,
|
| maximum=1,
|
| label=i18n(
|
| "输入源音量包络替换输出音量包络融合比例，越靠近1越使用输出包络"
|
| ),
|
| value=1,
|
| interactive=True,
|
| )
|
| protect0 = gr.Slider(
|
| minimum=0,
|
| maximum=0.5,
|
| label=i18n(
|
| "保护清辅音和呼吸声，防止电音撕裂等artifact，拉满0.5不开启，调低加大保护力度但可能降低索引效果"
|
| ),
|
| value=0.33,
|
| step=0.01,
|
| interactive=True,
|
| )
|
| f0_file = gr.File(
|
| label=i18n("F0曲线文件, 可选, 一行一个音高, 代替默认F0及升降调")
|
| )
|
| but0 = gr.Button(i18n("转换"), variant="primary")
|
| vc_output1 = gr.Textbox(label=i18n("输出信息"))
|
| vc_output2 = gr.Audio(label=i18n("输出音频(右下角三个点,点了可以下载)"))
|
| but0.click(
|
| vc.vc_single,
|
| [
|
| spk_item,
|
| vc_input3,
|
| vc_transform0,
|
| f0_file,
|
| f0method0,
|
| file_index1,
|
| file_index2,
|
|
|
| index_rate1,
|
| filter_radius0,
|
| resample_sr0,
|
| rms_mix_rate0,
|
| protect0,
|
| ],
|
| [vc_output1, vc_output2],
|
| )
|
|
|
|
|
| app.launch()
|
|
|
