Update app.py

app.py

@@ -1,58 +1,54 @@
 import gradio as gr
 
-# Системна інструкція
-SYSTEM_PROMPT = "Твоє ім'я — Flare. Ти персональний ШІ-помічник. Відповідай українською мовою у ввічливому та дружньому стилі."
+# 1. Твоя системна інструкція для Flare
+SYSTEM_PROMPT = "Твоє ім'я — Flare. Ти персональний ШІ-помічник. Відповідай українською мовою."
 
-# Налаштування кольорів (Чорний, Синій, Жовтий)
+# 2. Дизайн: Синій, Жовтий, Чорний
 theme = gr.themes.Soft(
     primary_hue="blue",
     secondary_hue="yellow",
     neutral_hue="slate",
 ).set(
-    body_background_fill="*neutral_950",        # Глибокий чорний
-    block_background_fill="*neutral_900",       # Темно-сірі блоки
+    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)
+    block_title_text_color="#FFD700",           # Жовтий текст (UA Gold)
     input_background_fill="*neutral_800",
 )
 
-css = """
-footer {visibility: hidden} 
-h1 {color: #FFD700 !important;}
-.sidebar-text {color: #ffffff !important;}
-"""
-
-# Завантажуємо модель один раз на рівні модуля
-client = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
+css = "footer {visibility: hidden} h1 {color: #FFD700 !important;}"
 
+# Функція для роботи чату (замість gr.load)
 def predict(message, history):
-    # Навчаємо модель пам'ятати історію та тримати образ "Flare"
-    prompt = f"Системна інструкція: {SYSTEM_PROMPT}\n\n"
+    # Підключаємо модель через провайдера novita (як у тебе в коді)
+    client = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
+    
+    # Формуємо запит так, щоб модель знала, що вона Flare
+    full_prompt = f"System: {SYSTEM_PROMPT}\n"
     for human, ai in history:
-        prompt += f"Користувач: {human}\nFlare: {ai}\n"
-    prompt += f"Користувач: {message}\nFlare:"
+        full_prompt += f"User: {human}\nAssistant: {ai}\n"
+    full_prompt += f"User: {message}\nAssistant:"
     
-    # Викликаємо модель
-    response = client(prompt)
-    return response
+    return client(full_prompt)
 
-# Створюємо блоки
+# ТВОЯ СТРУКТУРА КОДУ:
 with gr.Blocks(fill_height=True) as demo:
     with gr.Sidebar():
         gr.Markdown("# FlareAI")
-        gr.Markdown("### Flare — твій персональний помічник")
-        gr.Markdown("---")
+        gr.Markdown("Flare — твій персональний помічник")
+        # Твоя кнопка
         button = gr.LoginButton("Увійти")
-        gr.Markdown("🇺🇦 Зроблено з повагою до кольорів свободи.")
+        gr.Markdown("---")
+        gr.Markdown("🇺🇦 Український інтерфейс")
 
-    # У Gradio 6 прибрано параметри кастомних кнопок. Залишаємо чистий та робочий функціонал
+    # Твій чат (адаптований під Gradio 6)
     gr.ChatInterface(
         predict,
-        chatbot=gr.Chatbot(label="Чат з Flare", show_label=False, height=600),
-        textbox=gr.Textbox(placeholder="Напишіть повідомлення для Flare...", container=False, scale=7),
+        chatbot=gr.Chatbot(label="Чат Flare", show_label=False, height=600),
+        textbox=gr.Textbox(placeholder="Запитай Flare про що завгодно...", container=False, scale=7),
     )
 
 if __name__ == "__main__":
-    # Тема та стилі тепер живуть тільки в launch()
+    # У Gradio 6 theme та css передаються сюди
     demo.launch(theme=theme, css=css)