Fix GUI i18n (#38)

gui.py

@@ -1,6 +1,6 @@
 import PySimpleGUI as sg
 import sounddevice as sd
-#import noisereduce as nr
+import noisereduce as nr
 import numpy as np
 from fairseq import checkpoint_utils
 import librosa,torch,parselmouth,faiss,time,threading
@@ -147,6 +147,7 @@ class Config:
         self.threhold:int=-30
         self.crossfade_time:float=0.08
         self.extra_time:float=0.04
+        self.noise_reduce=False
 
 class GUI:
     def __init__(self) -> None:
@@ -181,7 +182,8 @@ class GUI:
                 sg.Frame(layout=[
                     [sg.Text(i18n("采样长度/Sample Length")),sg.Slider(range=(0.1,3.0),key='block_time',resolution=0.1,orientation='h',default_value=1.0)],
                     [sg.Text(i18n("淡入淡出长度/Crossfade Length")),sg.Slider(range=(0.01,0.15),key='crossfade_length',resolution=0.01,orientation='h',default_value=0.08)],
-                    [sg.Text(i18n("额外推理时长/Extra Length")),sg.Slider(range=(0.05,3.00),key='extra_time',resolution=0.01,orientation='h',default_value=0.05)]
+                    [sg.Text(i18n("额外推理时长/Extra Length")),sg.Slider(range=(0.05,3.00),key='extra_time',resolution=0.01,orientation='h',default_value=0.05)],
+                    [sg.Checkbox(i18n('输出降噪/Output Noisereduce'),key='noise_reduce')]
                 ],title=i18n("性能设置/Performance"))
             ],
             [sg.Button(i18n("开始音频转换"),key='start_vc'),sg.Button(i18n("停止音频转换"),key='stop_vc')]
@@ -217,7 +219,7 @@ class GUI:
         self.config.block_time=values['block_time']
         self.config.crossfade_time=values['crossfade_length']
         self.config.extra_time=values['extra_time']
-
+        self.config.noise_reduce=values['noise_reduce']
    
     def start_vc(self):
         torch.cuda.empty_cache()
@@ -290,6 +292,9 @@ class GUI:
         else:
             self.sola_buffer[:] = infer_wav[- self.crossfade_frame :]* self.fade_out_window
         
+        if self.config.noise_reduce:
+            self.output_wav[:]=nr.reduce_noise(y=self.output_wav,sr=self.config.samplerate)
+        
         outdata[:]=np.array([self.output_wav,self.output_wav]).T
         print('infer time:'+str(time.perf_counter()-start_time))
         
@@ -325,4 +330,6 @@ class GUI:
         sd.default.device[0]=input_device_indices[input_devices.index(input_device)]
         sd.default.device[1]=output_device_indices[output_devices.index(output_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()

locale/zh_CN.json

@@ -64,8 +64,21 @@
     "提取": "提取",
     "招募音高曲线前端编辑器": "招募音高曲线前端编辑器",
     "点击查看交流、问题反馈群号": "点击查看交流、问题反馈群号",
-    "输入设备/Input Device": "输入设备/Input Device",
-    "输出设备/Output Device": "输出设备/Output Device",
-    "音频设备(请使用同种类驱动)/Audio Devices": "音频设备(请使用同种类驱动)/Audio Devices",
-    "音调设置/Pitch Offset": "音调设置/Pitch Offset"
+    "加载模型/Load Model":"加载模型",
+    "选择.pth文件/.pth File":"选择.pth文件",
+    "选择.index文件/.index File":"选择.index文件",
+    "选择.npy文件/.npy File":"选择.npy文件",
+    "输入设备/Input Device": "输入设备",
+    "输出设备/Output Device": "输出设备",
+    "音频设备(请使用同种类驱动)/Audio Devices": "音频设备(请使用同种类驱动)",
+    "响应阈值/Silence Threhold":"响应阈值",
+    "音调设置/Pitch Offset": "音调设置",
+    "常规设置/Common":"常规设置",
+    "采样长度/Sample Length":"采样长度",
+    "淡入淡出长度/Crossfade Length":"淡入淡出长度",
+    "额外推理时长/Extra Length":"额外推理时长",
+    "输出降噪/Output Noisereduce":"输出降噪",
+    "性能设置/Performance":"性能设置",
+    "开始音频转换":"开始音频转换",
+    "停止音频转换":"停止音频转换"
 }

requirements-win-for-realtime_vc_gui.txt

@@ -25,3 +25,4 @@ wave
 PySimpleGUI
 sounddevice
 gradio
+noisereduce