OPENGIT/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2025-02-06 21:52:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1660 from RVC-Project/formatter-dev

Browse Source 
chore(format): run black on dev
This commit is contained in:
RVC-Boss 2023-12-28 22:48:08 +08:00 committed by GitHub
commit 4769472ba2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
gui_v1.py
View File

@ -14,6 +14,7 @@ import multiprocessing
flag_vc = False flag_vc = False
def printt(strr, *args): def printt(strr, *args):
if len(args) == 0: if len(args) == 0:
print(strr) print(strr)
@ -135,7 +136,7 @@ if __name__ == "__main__":
self.delay_time = 0 self.delay_time = 0
self.input_devices = None self.input_devices = None
self.output_devices = None self.output_devices = None
self.input_devices_indices = None self.input_devices_indices = None
self.output_devices_indices = None self.output_devices_indices = None
self.stream = None self.stream = None
self.update_devices() self.update_devices()
@ -153,16 +154,24 @@ if __name__ == "__main__":
data["rmvpe"] = data["f0method"] == "rmvpe" data["rmvpe"] = data["f0method"] == "rmvpe"
data["fcpe"] = data["f0method"] == "fcpe" data["fcpe"] = data["f0method"] == "fcpe"
if data["sg_input_device"] not in self.input_devices: if data["sg_input_device"] not in self.input_devices:
data["sg_input_device"] = self.input_devices[self.input_devices_indices.index(sd.default.device[0])] data["sg_input_device"] = self.input_devices[
self.input_devices_indices.index(sd.default.device[0])
]
if data["sg_output_device"] not in self.output_devices: if data["sg_output_device"] not in self.output_devices:
data["sg_output_device"] = self.output_devices[self.output_devices_indices.index(sd.default.device[1])] data["sg_output_device"] = self.output_devices[
self.output_devices_indices.index(sd.default.device[1])
]
except: except:
with open("configs/config.json", "w") as j: with open("configs/config.json", "w") as j:
data = { data = {
"pth_path": "", "pth_path": "",
"index_path": "", "index_path": "",
"sg_input_device": self.input_devices[self.input_devices_indices.index(sd.default.device[0])], "sg_input_device": self.input_devices[
"sg_output_device": self.output_devices[self.output_devices_indices.index(sd.default.device[1])], self.input_devices_indices.index(sd.default.device[0])
],
"sg_output_device": self.output_devices[
self.output_devices_indices.index(sd.default.device[1])
],
"sr_type": "sr_model", "sr_type": "sr_model",
"threhold": -60, "threhold": -60,
"pitch": 0, "pitch": 0,
@ -567,7 +576,7 @@ if __name__ == "__main__":
elif event == "stop_vc" or event != "start_vc": elif event == "stop_vc" or event != "start_vc":
# Other parameters do not support hot update # Other parameters do not support hot update
self.stop_stream() self.stop_stream()
def set_values(self, values): def set_values(self, values):
if len(values["pth_path"].strip()) == 0: if len(values["pth_path"].strip()) == 0:
sg.popup(i18n("请选择pth文件")) sg.popup(i18n("请选择pth文件"))
@ -726,7 +735,7 @@ if __name__ == "__main__":
sr=self.gui_config.samplerate, n_fft=4 * self.zc, prop_decrease=0.9 sr=self.gui_config.samplerate, n_fft=4 * self.zc, prop_decrease=0.9
).to(self.config.device) ).to(self.config.device)
self.start_stream() self.start_stream()
def start_stream(self): def start_stream(self):
global flag_vc global flag_vc
if not flag_vc: if not flag_vc:
@ -737,7 +746,8 @@ if __name__ == "__main__":
callback=self.audio_callback, callback=self.audio_callback,
blocksize=self.block_frame, blocksize=self.block_frame,
samplerate=self.gui_config.samplerate, samplerate=self.gui_config.samplerate,
dtype="float32") dtype="float32",
)
self.stream.start() self.stream.start()
def stop_stream(self): def stop_stream(self):
@ -748,7 +758,7 @@ if __name__ == "__main__":
self.stream.stop() self.stream.stop()
self.stream.close() self.stream.close()
self.stream = None self.stream = None
def audio_callback( def audio_callback(
self, indata: np.ndarray, outdata: np.ndarray, frames, times, status self, indata: np.ndarray, outdata: np.ndarray, frames, times, status
): ):
@ -937,7 +947,7 @@ if __name__ == "__main__":
for d in devices for d in devices
if d["max_output_channels"] > 0 if d["max_output_channels"] > 0
] ]
def set_devices(self, input_device, output_device): def set_devices(self, input_device, output_device):
"""设置输出设备""" """设置输出设备"""
sd.default.device[0] = self.input_devices_indices[ sd.default.device[0] = self.input_devices_indices[
@ -948,8 +958,10 @@ if __name__ == "__main__":
] ]
printt("Input device: %s:%s", str(sd.default.device[0]), input_device) printt("Input device: %s:%s", str(sd.default.device[0]), input_device)
printt("Output device: %s:%s", str(sd.default.device[1]), output_device) printt("Output device: %s:%s", str(sd.default.device[1]), output_device)
def get_device_samplerate(self): def get_device_samplerate(self):
return int(sd.query_devices(device=sd.default.device[0])['default_samplerate']) return int(
sd.query_devices(device=sd.default.device[0])["default_samplerate"]
)
gui = GUI() gui = GUI()