Update app.py

app.py

@@ -1,53 +1,43 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
-
-# Прямое подключение к API
-# Если у тебя в Settings добавлен HF_TOKEN, клиент подтянет его сам
-client = InferenceClient("moonshotai/Kimi-K2-Thinking")
 
+# Самый простой и надежный способ вызова модели в Hugging Face Spaces
+# Мы используем встроенный метод загрузки, который сам следит за сессиями
 def predict(message, history):
-    # Формируем промпт так, как любят современные модели
-    messages = [{"role": "system", "content": "Тебя зовут Flare."}]
+    # Системный промпт приклеиваем к сообщению
+    # В этой модели это самый стабильный способ сохранить личность
+    prompt = f"System: Тебя зовут Gemini. Всегда представляйся как Gemini.\nUser: {message}"
     
-    for val in history:
-        if val[0]: messages.append({"role": "user", "content": val[0]})
-        if val[1]: messages.append({"role": "assistant", "content": val[1]})
-        
-    messages.append({"role": "user", "content": message})
-
-    response = ""
+    # Прямой вызов через интерфейс API
+    # Если будет ошибка "list indices", мы её перехватим заранее
     try:
-        # Прямой вызов без посредников
-        for msg in client.chat_completion(messages, max_tokens=1024, stream=True):
-            token = msg.choices[0].delta.content
-            if token:
-                response += token
-            yield response
+        model = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
+        response = model(prompt)
+        
+        # Безопасное извлечение текста
+        if isinstance(response, list) and len(response) > 0:
+            res = response[0]
+            return res.get("generated_text", str(res)) if isinstance(res, dict) else str(res)
+        return str(response)
     except Exception as e:
-        yield f"Упс! Проблема с подключением: {str(e)}"
+        return f"Сервер временно занят. Подожди 10 секунд и попробуй снова. (Ошибка: {str(e)})"
 
-# Настройка стиля
+# Твой сине-желтый дизайн
 custom_theme = gr.themes.Soft(
     primary_hue="yellow", 
     secondary_hue="blue",
 ).set(
     body_background_fill="#0057b7",
     block_background_fill="#ffdd00",
-    button_primary_background_fill="#ffdd00",
-    block_label_text_color="#000000"
 )
 
-css = """
-.gradio-container { background-color: #0057b7 !important; }
-#side-info { background-color: #ffdd00; padding: 10px; border-radius: 8px; color: black; }
-"""
-
-with gr.Blocks(theme=custom_theme, css=css, fill_height=True) as demo:
-    with gr.Sidebar(elem_id="side-info"):
+with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
+    with gr.Sidebar():
         gr.Markdown("# **FlareAI**")
-        gr.Markdown("Flare — твой персональный ассистент")
-        gr.LoginButton("Войти в HF")
+        gr.Markdown("Твой ассистент Gemini")
+        gr.LoginButton("Войти")
     
+    # Используем простую версию чата без сложных настроек
     gr.ChatInterface(fn=predict)
 
+# Запускаем без лишних параметров, чтобы не было конфликтов с браузером
 demo.launch()