Update app.py

app.py

@@ -2,56 +2,46 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import os
 
-# 1. Инициализация клиента с использованием токена из секретов
-# Убедись, что в Settings -> Secrets добавлен HF_TOKEN
+# 1. Инициализация клиента
 token = os.getenv("HF_TOKEN")
-client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.2", token=token)
-
-def respond(message, history: list[tuple[str, str]]):
-    """
-    Функция для обработки сообщений.
-    history — это список пар [пользователь, бот], который Gradio ведет сам.
-    """
-    
-    # Формируем системный промпт, чтобы задать роль боту
-    messages = [{"role": "system", "content": "Ты — полезный ИИ-ассистент. Отвечай кратко и по делу на русском языке."}]
-
-    # Добавляем историю переписки в контекст
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    # Добавляем текущее сообщение пользователя
+# Используем Llama 3 для стабильности, но можно вернуть и Mistral
+client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct", token=token)
+
+def respond(message, history):
+    messages = [{"role": "system", "content": "Ты — полезный ИИ-ассистент. Отвечай на русском языке."}]
+
+    # Добавляем историю
+    for user_msg, assistant_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
+
     messages.append({"role": "user", "content": message})
 
     response = ""
 
-    # Запрос к модели через Inference API
     try:
-        # Используем stream=True для эффекта печатающегося текста
+        # Стриминг ответа
         for message_chunk in client.chat_completion(
             messages=messages,
             max_tokens=1000,
             stream=True,
             temperature=0.7,
-            top_p=0.95,
         ):
             token_text = message_chunk.choices[0].delta.content
             if token_text:
                 response += token_text
                 yield response
     except Exception as e:
-        yield f"Произошла ошибка: {str(e)}. Проверь, добавлен ли HF_TOKEN в Settings -> Secrets."
+        yield f"Ошибка доступа: {str(e)}. Проверь HF_TOKEN в Settings -> Secrets."
 
-# 2. Создаем красивый интерфейс
+# 2. Создаем интерфейс (без аргумента theme, чтобы не было ошибки)
 demo = gr.ChatInterface(
     fn=respond,
     title="Мой ИИ Чат-бот",
-    description="Запущено в Hugging Face Spaces с использованием Mistral API",
-    examples=["Привет! Кто ты?", "Напиши код на Python для сортировки списка", "Расскажи шутку про программистов"],
-    theme="soft" # Можно выбрать: 'default', 'soft', 'monochrome'
+    examples=["Привет!", "Как дела?", "Напиши стих о роботах"],
+    textbox=gr.Textbox(placeholder="Введите сообщение...", container=False, scale=7),
 )
 
 if __name__ == "__main__":