mirror of
https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git
synced 2025-02-07 06:02:49 +08:00
Format code (#456)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot]
GitHub
parent
1b307a4222
commit
fada942ecd
102
gui.py
102
gui.py
@ -253,31 +253,27 @@ class GUI:
|
|||||||
|
|
||||||
self.launcher()
|
self.launcher()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def load(self):
|
def load(self):
|
||||||
input_devices, output_devices, _, _ = self.get_devices()
|
input_devices, output_devices, _, _ = self.get_devices()
|
||||||
try:
|
try:
|
||||||
with open('values1.json', 'r') as j:
|
with open("values1.json", "r") as j:
|
||||||
data = json.load(j)
|
data = json.load(j)
|
||||||
except:
|
except:
|
||||||
with open('values1.json', 'w') as j:
|
with open("values1.json", "w") as j:
|
||||||
data = {
|
data = {
|
||||||
"pth_path": " ",
|
"pth_path": " ",
|
||||||
"index_path": " ",
|
"index_path": " ",
|
||||||
"sg_input_device": input_devices[sd.default.device[0]],
|
"sg_input_device": input_devices[sd.default.device[0]],
|
||||||
"sg_output_device": output_devices[sd.default.device[1]],
|
"sg_output_device": output_devices[sd.default.device[1]],
|
||||||
"threshold": '-60',
|
"threshold": "-60",
|
||||||
"pitch": '12',
|
"pitch": "12",
|
||||||
"index_rate": '0.5',
|
"index_rate": "0.5",
|
||||||
"block_time": '1',
|
"block_time": "1",
|
||||||
"crossfade_length": '0.15',
|
"crossfade_length": "0.15",
|
||||||
"extra_time": '1.1', }
|
"extra_time": "1.1",
|
||||||
|
}
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def launcher(self):
|
def launcher(self):
|
||||||
data = self.load()
|
data = self.load()
|
||||||
sg.theme("LightBlue3")
|
sg.theme("LightBlue3")
|
||||||
@ -288,28 +284,50 @@ class GUI:
|
|||||||
title=i18n("加载模型"),
|
title=i18n("加载模型"),
|
||||||
layout=[
|
layout=[
|
||||||
[
|
[
|
||||||
sg.Input(default_text="hubert_base.pt", key="hubert_path",disabled=True),
|
sg.Input(
|
||||||
sg.FileBrowse(i18n("Hubert模型"), initial_folder=os.path.join(os.getcwd()),file_types=((". pt"),)),
|
default_text="hubert_base.pt",
|
||||||
],
|
key="hubert_path",
|
||||||
[
|
disabled=True,
|
||||||
sg.Input(default_text=data.get("pth_path", ''), key="pth_path",),
|
),
|
||||||
sg.FileBrowse(i18n("选择.pth文件"),initial_folder=os.path.join(os.getcwd(), "weights"), file_types=((". pth"),))
|
sg.FileBrowse(
|
||||||
|
i18n("Hubert模型"),
|
||||||
|
initial_folder=os.path.join(os.getcwd()),
|
||||||
|
file_types=((". pt"),),
|
||||||
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
sg.Input(
|
sg.Input(
|
||||||
default_text=data.get('index_path', ''),
|
default_text=data.get("pth_path", ""),
|
||||||
key="index_path",
|
key="pth_path",
|
||||||
|
),
|
||||||
|
sg.FileBrowse(
|
||||||
|
i18n("选择.pth文件"),
|
||||||
|
initial_folder=os.path.join(os.getcwd(), "weights"),
|
||||||
|
file_types=((". pth"),),
|
||||||
|
),
|
||||||
|
],
|
||||||
|
[
|
||||||
|
sg.Input(
|
||||||
|
default_text=data.get("index_path", ""),
|
||||||
|
key="index_path",
|
||||||
|
),
|
||||||
|
sg.FileBrowse(
|
||||||
|
i18n("选择.index文件"),
|
||||||
|
initial_folder=os.path.join(os.getcwd(), "logs"),
|
||||||
|
file_types=((". index"),),
|
||||||
),
|
),
|
||||||
sg.FileBrowse(i18n("选择.index文件"), initial_folder=os.path.join(os.getcwd(), "logs"),file_types=((". index"),)),
|
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
sg.Input(
|
sg.Input(
|
||||||
default_text="你不需要填写这个You don't need write this.",
|
default_text="你不需要填写这个You don't need write this.",
|
||||||
key="npy_path",
|
key="npy_path",
|
||||||
disabled=True
|
disabled=True,
|
||||||
|
),
|
||||||
|
sg.FileBrowse(
|
||||||
|
i18n("选择.npy文件"),
|
||||||
|
initial_folder=os.path.join(os.getcwd(), "logs"),
|
||||||
|
file_types=((". npy"),),
|
||||||
),
|
),
|
||||||
sg.FileBrowse(i18n("选择.npy文件"), initial_folder=os.path.join(os.getcwd(), "logs"),file_types=((". npy"),)),
|
|
||||||
],
|
],
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
@ -322,7 +340,7 @@ class GUI:
|
|||||||
sg.Combo(
|
sg.Combo(
|
||||||
input_devices,
|
input_devices,
|
||||||
key="sg_input_device",
|
key="sg_input_device",
|
||||||
default_value=data.get('sg_input_device', ''),
|
default_value=data.get("sg_input_device", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
@ -330,7 +348,7 @@ class GUI:
|
|||||||
sg.Combo(
|
sg.Combo(
|
||||||
output_devices,
|
output_devices,
|
||||||
key="sg_output_device",
|
key="sg_output_device",
|
||||||
default_value=data.get('sg_output_device', ''),
|
default_value=data.get("sg_output_device", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
],
|
],
|
||||||
@ -347,7 +365,7 @@ class GUI:
|
|||||||
key="threhold",
|
key="threhold",
|
||||||
resolution=1,
|
resolution=1,
|
||||||
orientation="h",
|
orientation="h",
|
||||||
default_value=data.get('threhold', '')
|
default_value=data.get("threhold", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
@ -357,7 +375,7 @@ class GUI:
|
|||||||
key="pitch",
|
key="pitch",
|
||||||
resolution=1,
|
resolution=1,
|
||||||
orientation="h",
|
orientation="h",
|
||||||
default_value=data.get('pitch', ''),
|
default_value=data.get("pitch", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
@ -367,7 +385,7 @@ class GUI:
|
|||||||
key="index_rate",
|
key="index_rate",
|
||||||
resolution=0.01,
|
resolution=0.01,
|
||||||
orientation="h",
|
orientation="h",
|
||||||
default_value=data.get('index_rate', ''),
|
default_value=data.get("index_rate", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
],
|
],
|
||||||
@ -382,7 +400,7 @@ class GUI:
|
|||||||
key="block_time",
|
key="block_time",
|
||||||
resolution=0.1,
|
resolution=0.1,
|
||||||
orientation="h",
|
orientation="h",
|
||||||
default_value=data.get('block_time', ''),
|
default_value=data.get("block_time", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
@ -392,7 +410,7 @@ class GUI:
|
|||||||
key="crossfade_length",
|
key="crossfade_length",
|
||||||
resolution=0.01,
|
resolution=0.01,
|
||||||
orientation="h",
|
orientation="h",
|
||||||
default_value=data.get('crossfade_length', ''),
|
default_value=data.get("crossfade_length", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
@ -402,7 +420,7 @@ class GUI:
|
|||||||
key="extra_time",
|
key="extra_time",
|
||||||
resolution=0.01,
|
resolution=0.01,
|
||||||
orientation="h",
|
orientation="h",
|
||||||
default_value=data.get('extra_time', ''),
|
default_value=data.get("extra_time", ""),
|
||||||
),
|
),
|
||||||
],
|
],
|
||||||
[
|
[
|
||||||
@ -436,8 +454,6 @@ class GUI:
|
|||||||
settings = {
|
settings = {
|
||||||
"pth_path": values["pth_path"],
|
"pth_path": values["pth_path"],
|
||||||
"index_path": values["index_path"],
|
"index_path": values["index_path"],
|
||||||
|
|
||||||
|
|
||||||
"sg_input_device": values["sg_input_device"],
|
"sg_input_device": values["sg_input_device"],
|
||||||
"sg_output_device": values["sg_output_device"],
|
"sg_output_device": values["sg_output_device"],
|
||||||
"threhold": values["threhold"],
|
"threhold": values["threhold"],
|
||||||
@ -447,25 +463,14 @@ class GUI:
|
|||||||
"crossfade_length": values["crossfade_length"],
|
"crossfade_length": values["crossfade_length"],
|
||||||
"extra_time": values["extra_time"],
|
"extra_time": values["extra_time"],
|
||||||
}
|
}
|
||||||
with open('values1.json', 'w') as j:
|
with open("values1.json", "w") as j:
|
||||||
json.dump(settings, j)
|
json.dump(settings, j)
|
||||||
if event == "stop_vc" and self.flag_vc == True:
|
if event == "stop_vc" and self.flag_vc == True:
|
||||||
self.flag_vc = False
|
self.flag_vc = False
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def set_values(self, values):
|
def set_values(self, values):
|
||||||
self.set_devices(values["sg_input_device"], values["sg_output_device"])
|
self.set_devices(values["sg_input_device"], values["sg_output_device"])
|
||||||
self.config.hubert_path = os.path.join(current_dir, 'hubert_base.pt')
|
self.config.hubert_path = os.path.join(current_dir, "hubert_base.pt")
|
||||||
self.config.pth_path = values["pth_path"]
|
self.config.pth_path = values["pth_path"]
|
||||||
self.config.index_path = values["index_path"]
|
self.config.index_path = values["index_path"]
|
||||||
self.config.npy_path = values["npy_path"]
|
self.config.npy_path = values["npy_path"]
|
||||||
@ -672,4 +677,5 @@ class GUI:
|
|||||||
print("input device:" + str(sd.default.device[0]) + ":" + str(input_device))
|
print("input device:" + str(sd.default.device[0]) + ":" + str(input_device))
|
||||||
print("output device:" + str(sd.default.device[1]) + ":" + str(output_device))
|
print("output device:" + str(sd.default.device[1]) + ":" + str(output_device))
|
||||||
|
|
||||||
|
|
||||||
gui = GUI()
|
gui = GUI()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user