Update app.py

app.py

@@ -1,54 +1,10 @@
 import gradio as gr
-from huggingface_hub import InferenceClient
-import os
 
-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. Отвечай кратко и понятно на русском языке."}]
-    
-    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.Blocks(fill_height=True) 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)
+        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")
+    
+demo.launch()