Update app.py

app.py

@@ -1,10 +1,38 @@
 import gradio as gr
 
-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")
+# Кастомный CSS для сине-желтого дизайна
+css = """
+.gradio-container {
+    background-color: #0057b7 !important; /* Синий фон */
+}
+#side-bar {
+    background-color: #ffdd00 !important; /* Желтый сайдбар */
+    padding: 15px;
+    border-radius: 10px;
+}
+footer {display: none !important;}
+"""
+
+def predict(message, history):
+    # Добавляем системную инструкцию перед сообщением пользователя
+    system_prompt = "Тебя зовут Flare "
+    full_prompt = f"{system_prompt} {message}"
     
+    # Загружаем модель (используем интерфейс через функцию для контроля промпта)
+    client = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
+    return client(full_prompt)
+
+with gr.Blocks(css=css, fill_height=True) as demo:
+    with gr.Sidebar(elem_id="side-bar"):
+        gr.Markdown("# **FlareAI**")
+        gr.Markdown("Flare — твой персональный ассистент")
+        button = gr.LoginButton("Войти")
+        
+    gr.ChatInterface(
+        predict,
+        additional_inputs=None,
+        # Настройка цветов текстовых пузырей через тему
+        theme=gr.themes.Soft(primary_hue="yellow", secondary_hue="blue")
+    )
+
 demo.launch()