Update app.py

app.py

@@ -1,10 +1,63 @@
 import gradio as gr
 
-with gr.Blocks(fill_height=True) as demo:
+# Спеціальна системна інструкція, щоб ШІ знав своє ім'я
+SYSTEM_PROMPT = "Твоє ім'я — Flare. Ти персональний ШІ-помічник. Відповідай українською мовою у ввічливому та дружньому стилі."
+
+# Налаштування стилів (Чорний фон, Сині кнопки, Жовті акценти)
+theme = gr.themes.Soft(
+    primary_hue="blue",
+    secondary_hue="yellow",
+    neutral_hue="slate",
+).set(
+    body_background_fill="*neutral_950",        # Глибокий чорний колір фону
+    block_background_fill="*neutral_900",       # Темно-сірі блоки
+    button_primary_background_fill="#0057B7",   # Синій колір кнопок (UA Blue)
+    button_primary_text_color="white",
+    block_title_text_color="#FFD700",           # Жовті заголовки (UA Gold)
+    input_background_fill="*neutral_800",
+)
+
+# Додатковий CSS для тонкого налаштування кольорів
+css = """
+footer {visibility: hidden} 
+h1 {color: #FFD700 !important;}
+.sidebar-text {color: #ffffff !important;}
+"""
+
+def predict(message, history):
+    # Додаємо системну інструкцію до контексту
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+    for human, ai in history:
+        messages.append({"role": "user", "content": human})
+        messages.append({"role": "assistant", "content": ai})
+    messages.append({"role": "user", "content": message})
+    
+    # Виклик моделі через провайдера novita
+    # Примітка: gr.load("models/...") у ChatInterface працює через функцію-обгортку
+    client = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
+    response = client(message, history=history)
+    return response
+
+with gr.Blocks(theme=theme, css=css, 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")
-    
-demo.launch()
+        gr.Markdown("### Flare — твій персональний помічник")
+        gr.Markdown("---")
+        button = gr.LoginButton("Увійти")
+        gr.Markdown("🇺🇦 Зроблено з повагою до кольорів свободи.")
+
+    # Створення чату
+    gr.ChatInterface(
+        predict,
+        chatbot=gr.Chatbot(label="Чат з Flare", show_label=False, bubble_full_width=False, height=600),
+        textbox=gr.Textbox(placeholder="Напишіть повідомлення для Flare...", container=False, scale=7),
+        submit_btn="Надіслати",
+        stop_btn="Зупинити",
+        retry_btn="Повторити",
+        undo_btn="Відмінити",
+        clear_btn="Очистити",
+        additional_inputs=None,
+    )
+
+if __name__ == "__main__":
+    demo.launch()