Update app.py

app.py

@@ -1,54 +1,37 @@
 import gradio as gr
 import torch
-import torch.nn as nn
-from torch.nn import functional as F
 import time
 
-# --- НАСТРОЙКИ МОДЕЛИ ---
-n_embd = 64
-block_size = 32
-n_head = 4
-n_layer = 4
-vocab_size = 100 # Примерный размер словаря
+# Кастомный CSS для создания градиента качества в тексте.
+# Мы применяем градиент только к markdown-тексту в описании.
+# linear-gradient(135deg, #FF00CC 0%, #9932CC 40%, #0000FF 70%, #00FFFF 100%) - это цвета как на твоем изображении.
+css = """
+.gr-markdown {
+    background: linear-gradient(135deg, #FF00CC 0%, #9932CC 40%, #0000FF 70%, #00FFFF 100%);
+    -webkit-background-clip: text;
+    -webkit-text-fill-color: transparent;
+    font-weight: bold;
+    font-size: 1.2em;
+}
+"""
 
-# --- ТУТ ТВОЯ АРХИТЕКТУРА ТРАНСФОРМЕРА (сокращенно для запуска) ---
-class AxisBrain(nn.Module):
-    def __init__(self):
-        super().__init__()
-        self.token_embedding_table = nn.Embedding(vocab_size, n_embd)
-        self.blocks = nn.Sequential(*[nn.Linear(n_embd, n_embd) for _ in range(n_layer)])
-        self.lm_head = nn.Linear(n_embd, vocab_size)
-
-    def forward(self, idx):
-        x = self.token_embedding_table(idx)
-        x = self.blocks(x)
-        logits = self.lm_head(x)
-        return logits
-
-# Создаем экземпляр модели "мозга"
-model = AxisBrain()
-
-# --- ЛОГИКА ИНТЕРФЕЙСА ---
+# Временная логика (заглушка)
 def axis_response(message, history, progress=gr.Progress()):
-    progress(0.0, desc="[0%] Инициализация нейронных связей...")
+    # Прогресс-бар для атмосферы
+    progress(0.0, desc="[0%] Проверка качества...")
     time.sleep(0.5)
+    progress(0.5, desc="[50%] Анализ данных...")
+    time.sleep(0.5)
+    progress(1.0, desc="[100%] Анализ завершен.")
     
-    # Имитация работы модели
-    progress(0.5, desc="[50%] Прогон тензоров через слои Transformer...")
-    time.sleep(0.8)
-    
-    progress(0.9, desc="[90%] Генерация ответа...")
-    
-    if message.lower() == "/status":
-        return "[System] Архитектура Transformer загружена. Параметры: n_layer=4, n_head=4."
-    
-    return f"Project Axis принял запрос. Архитектура готова к обучению. Входные данные: '{message}'"
+    return f"Axis Terminal: '{message}'. Мы подключили градиент качества в описание чата. Жду твои 'мозги' для ИИ."
 
-# Запуск чата
+# Создаем интерфейс с кастомным CSS и обновленным описанием
 demo = gr.ChatInterface(
     fn=axis_response,
-    title="Project Axis — Transformer Terminal",
-    description="Система запущена на базе архитектуры Transformer."
+    title="Project Axis — Терминал Качества",
+    description="Интерфейс для взаимодействия с прототипом Project Axis. Этот градиент символизирует **качество**.",
+    css=css # Подключаем кастомный CSS
 )
 
 if __name__ == "__main__":