Update app.py

app.py

@@ -12,22 +12,15 @@ from pydub import AudioSegment
 from pydub.playback import play
 
 
-# PATH_TO_MODEL = os.getcwd() + "/vosk-model-tts-ru-0.4-multi/"
-
-
-# with open(os.path.join(PATH_TO_MODEL, 'dictionary')) as file:
-#     for i, val in enumerate(file):
-#         if i == 100:
-#             break
-#         print(val)
-
-
+PATH_TO_MODEL = os.path.join(os.getcwd(), "vosk-model-tts-ru-0.4-multi")
 PATH_TO_OUTPUT = os.path.join(os.getcwd(), "content")
 k = "sk-YOVNQzHmpga9My3dwlSo9BQN907TuPZQXcHn50ztigTwm3I2"
 YA_GPT_KEY = "AQVNyVqBi-XoJ1cAo7VIxq6ztgXm3owqowtso5Qb"
 instruction = """
 Ответь на запрос так, как ответил бы на него Павел Воля. Используй данные из биографии Павла Воли, если это потребуется. Отвечай на запрос в его стиле. Ответ должен содержать не болеее 10 предложений.
 """
+
+
 #files = [
 #    ("input_face", open("C:\\Users\\user\\Desktop\\deepfake_sirius\\materials\\scale_1200.jpg", "rb")), #TODO: IT
 #    ("input_audio", open("C:\\Users\\user\\Desktop\\deepfake_sirius\\materials\\audio\\output.wav", "rb")), #TODO: IT
@@ -63,8 +56,7 @@ instruction = """
 class VoiceGenerator:
     
     def __init__(self):
-        # print(PATH_TO_MODEL)
-        self.model = Model("/home/user/app/vosk-model-tts-ru-0.4-multi/")
+        self.model = Model(PATH_TO_MODEL)
 
         
     def generate(self, text, file_name='output.wav'):
@@ -118,7 +110,7 @@ def result(audio):
     file_path = os.path.join(PATH_TO_OUTPUT, file_name)
     write(file_path, sample_rate, data)
     text_from_audio = recognize_audio(file_path)
-    generated_audio = vg.generate(text_from_audio)
+    generated_audio = vg.generate(ask_gpt(text_from_audio))
     return generated_audio