Update app.py

app.py

@@ -1,10 +1,54 @@
 import gradio as gr
+from huggingface_hub import InferenceClient
+import os
 
-with gr.Blocks(fill_height=True) as demo:
-    with gr.Sidebar():
-        gr.Markdown("# FlareAI")
-        gr.Markdown("Flare - your personal assistant")
-        button = gr.LoginButton("Sign in")
-    gr.load("models/moonshotai/Kimi-K2-Thinking", accept_token=button, provider="novita")
+MODEL_ID = "Qwen/Qwen2.5-72B-Instruct"
+
+def predict(message, history):
+    token = os.getenv("HF_TOKEN")
+    client = InferenceClient(MODEL_ID, token=token)
+
+    messages = [{"role": "system", "content": "Тебя зовут Gemini. Ты ассистент FlareAI. Отвечай кратко и понятно на русском языке."}]
     
-demo.launch()
+    for user_msg, bot_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": bot_msg})
+    
+    messages.append({"role": "user", "content": message})
+
+    response = ""
+    try:
+        for message_chunk in client.chat_completion(messages, max_tokens=1024, stream=True):
+            token_text = message_chunk.choices[0].delta.content
+            if token_text:
+                response += token_text
+            yield response
+    except Exception as e:
+        yield f"⚠️ Ошибка: {str(e)}. Подожди 10 секунд и попробуй снова."
+
+custom_theme = gr.themes.Soft(
+    primary_hue="yellow",
+    secondary_hue="blue",
+).set(
+    body_background_fill="#0057b7", 
+    block_background_fill="#ffdd00", 
+    block_title_text_color="#000000",
+    button_primary_text_color="#000000"
+)
+
+# ИСПРАВЛЕНИЕ 1: Убрали theme отсюда
+with gr.Blocks(title="FlareAI") as demo:
+    with gr.Sidebar():
+        gr.Markdown("# **FlareAI**")
+        gr.Markdown("---")
+        gr.Markdown("Твой ассистент на базе Gemini")
+        gr.LoginButton() 
+        gr.Markdown("---")
+        gr.Markdown("Если бот молчит, нажми 'Login' или проверь HF_TOKEN в настройках.")
+
+    # ИСПРАВЛЕНИЕ 2: Убрали параметры type и save_history, оставили только функцию
+    gr.ChatInterface(fn=predict)
+
+if __name__ == "__main__":
+    # ИСПРАВЛЕНИЕ 3: Перенесли theme сюда, как просило предупреждение в логах
+    demo.launch(theme=custom_theme)