Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 # Спеціальна системна інструкція, щоб ШІ знав своє ім'я
 SYSTEM_PROMPT = "Твоє ім'я — Flare. Ти персональний ШІ-помічник. Відповідай українською мовою у ввічливому та дружньому стилі."
 
-# Налаштування стилів (Чорний фон, Сині кнопки, Жовті акценти)
+# Налаштування кольорів
 theme = gr.themes.Soft(
     primary_hue="blue",
     secondary_hue="yellow",
@@ -17,28 +17,29 @@ theme = gr.themes.Soft(
     input_background_fill="*neutral_800",
 )
 
-# Додатковий CSS для тонкого налаштування кольорів
 css = """
 footer {visibility: hidden} 
 h1 {color: #FFD700 !important;}
 .sidebar-text {color: #ffffff !important;}
 """
 
+# Завантажуємо модель один раз при запуску, щоб не перевантажувати пам'ять
+client = gr.load("models/moonshotai/Kimi-K2-Thinking", provider="novita")
+
 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:
+# Створюємо інтерфейс без передачі theme та css сюди (вимога Gradio 6)
+with gr.Blocks(fill_height=True) as demo:
     with gr.Sidebar():
         gr.Markdown("# FlareAI")
         gr.Markdown("### Flare — твій персональний помічник")
@@ -46,10 +47,10 @@ with gr.Blocks(theme=theme, css=css, fill_height=True) as demo:
         button = gr.LoginButton("Увійти")
         gr.Markdown("🇺🇦 Зроблено з повагою до кольорів свободи.")
 
-    # Створення чату
+    # Створення чату (прибрано bubble_full_width)
     gr.ChatInterface(
         predict,
-        chatbot=gr.Chatbot(label="Чат з Flare", show_label=False, bubble_full_width=False, height=600),
+        chatbot=gr.Chatbot(label="Чат з Flare", show_label=False, height=600),
         textbox=gr.Textbox(placeholder="Напишіть повідомлення для Flare...", container=False, scale=7),
         submit_btn="Надіслати",
         stop_btn="Зупинити",
@@ -60,4 +61,5 @@ with gr.Blocks(theme=theme, css=css, fill_height=True) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # У Gradio 6 theme та css передаються САМЕ ТУТ
+    demo.launch(theme=theme, css=css)