Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 import cv2
 import numpy as np
 from pydub import AudioSegment
+from langdetect import detect
 
 # Инициализация клиента для модели HuggingFaceH4/zephyr-7b-beta
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
@@ -35,6 +36,13 @@ def process_audio(file):
     audio = AudioSegment.from_file(file)
     return f"Аудио: длительность {len(audio) / 1000:.2f} секунд, частота {audio.frame_rate} Гц"
 
+# Функция для определения языка текста
+def detect_language(text):
+    try:
+        return detect(text)
+    except:
+        return "en"  # По умолчанию английский
+
 # Функция для обработки сообщений, истории и файлов
 def respond(
     message,
@@ -44,6 +52,7 @@ def respond(
     temperature,
     top_p,
     file=None,
+    language="auto",
 ):
     # Если загружен файл, обрабатываем его
     if file is not None:
@@ -60,6 +69,26 @@ def respond(
             file_info = "Неизвестный тип файла"
         message += f"\n[Пользователь загрузил файл: {file.name}]\n{file_info}"
 
+    # Определяем язык сообщения
+    if language == "auto":
+        language = detect_language(message)
+    else:
+        language = language
+
+    # Добавляем системное сообщение с учетом языка
+    if language == "ru":
+        system_message = "Вы дружелюбный чат-бот, который понимает русский язык."
+    elif language == "en":
+        system_message = "You are a friendly chatbot that understands English."
+    elif language == "es":
+        system_message = "Eres un chatbot amigable que entiende español."
+    elif language == "fr":
+        system_message = "Vous êtes un chatbot sympathique qui comprend le français."
+    elif language == "de":
+        system_message = "Sie sind ein freundlicher Chatbot, der Deutsch versteht."
+    else:
+        system_message = "You are a friendly chatbot."
+
     # Добавляем системное сообщение
     messages = [{"role": "system", "content": system_message}]
 
@@ -93,12 +122,20 @@ def reset_chat():
 # Создание интерфейса с использованием ChatInterface
 with gr.Blocks() as demo:
     gr.Markdown("# Felguk v0")
-    gr.Markdown("Чат-бот на основе модели felguk v0. Задавайте вопросы и загружайте файлы (PDF, изображения, видео, аудио)!")
+    gr.Markdown("Чат-бот Felguk v0. Поддерживает несколько языков. Задавайте вопросы и загружайте файлы (PDF, изображения, видео, аудио)!")
     
     # Кнопка "Новый чат"
     with gr.Row():
         new_chat_button = gr.Button("Новый чат", variant="secondary")
 
+    # Выбор языка
+    with gr.Row():
+        language_dropdown = gr.Dropdown(
+            choices=["auto", "en", "ru", "es", "fr", "de"],
+            value="auto",
+            label="Выберите язык (или оставьте 'auto' для автоматического определения)",
+        )
+
     # Интерфейс чата
     chat_interface = gr.ChatInterface(
         respond,
@@ -114,6 +151,7 @@ with gr.Blocks() as demo:
                 label="Top-p (nucleus sampling)",
             ),
             gr.File(label="Загрузите файл (опционально)"),  # Поле для загрузки файлов
+            language_dropdown,  # Выбор языка
         ],
     )