feature: 集成bert_vits

app/public.py

@@ -30,12 +30,12 @@ class SentenceSegmentation():
         return self.__sentenceSegmentation(llm_chunk)
 
 class Recorder:
-    def __init__(self, user_id):
+    def __init__(self, user_id, input_sr, output_sr):
         self.input_wav_path = 'storage/wav/'+ datetime.now().strftime('%Y%m%d%H%M%S') + 'U' + user_id + 'i.wav'
         self.output_wav_path = 'storage/wav/'+ datetime.now().strftime('%Y%m%d%H%M%S') + 'U' + user_id + 'o.wav'
         self.out_put_text_path = 'storage/record/'+ datetime.now().strftime('%Y%m%d%H%M%S') + 'U' + user_id + 'o.txt'
-        self.input_sr = 16000
-        self.output_sr = 22050
+        self.input_sr = input_sr
+        self.output_sr = output_sr
         self.user_audio = b''
         self.tts_audio = b''
         self.input_text = ""