Update app.py

app.py

@@ -6,6 +6,8 @@ import cv2
 import numpy as np
 from pydub import AudioSegment
 from langdetect import detect
+from gtts import gTTS
+import os
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -43,6 +45,13 @@ def detect_language(text):
     except:
         return "en"  # По умолчанию английский
 
+# Функция для преобразования текста в речь (TTS)
+def text_to_speech(text, language):
+    tts = gTTS(text=text, lang=language, slow=False)
+    audio_file = "output.mp3"
+    tts.save(audio_file)
+    return audio_file
+
 # Функция для обработки сообщений, истории и файлов
 def respond(
     message,
@@ -103,6 +112,13 @@ def respond(
         response += token
         yield response
 
+    # Сохраняем ответ для TTS
+    if response:
+        audio_file = text_to_speech(response, language)
+        return response, gr.Audio.update(value=audio_file, visible=True)
+    else:
+        return response, gr.Audio.update(visible=False)
+
 # Функция для сброса истории чата
 def reset_chat():
     return []
@@ -134,9 +150,19 @@ with gr.Blocks() as demo:
         ],
     )
 
+    # Аудиоплеер для TTS
+    audio_player = gr.Audio(label="Reader", visible=False)
+
     # Привязка кнопки "Новый чат" к функции сброса истории
     new_chat_button.click(fn=reset_chat, outputs=chat_interface.chatbot)
 
+    # Обновление интерфейса после ответа бота
+    chat_interface.chatbot.change(
+        fn=lambda response: (response, gr.Audio.update(visible=bool(response))),
+        inputs=chat_interface.chatbot,
+        outputs=[chat_interface.chatbot, audio_player],
+    )
+
 # Запуск интерфейса
 if __name__ == "__main__":
     demo.launch()