Update app.py

app.py

@@ -7,6 +7,10 @@ import gradio as gr
 from datetime import datetime
 import math
 import spaces
+import warnings
+
+# Подавляем предупреждение PyTorch о nested tensors (оно не влияет на работу)
+warnings.filterwarnings("ignore", message="The PyTorch API of nested tensors is in prototype stage")
 
 # ============ НАСТРОЙКИ ============
 DATA_DIR = '/data'
@@ -124,7 +128,7 @@ class AndreyTransformer(nn.Module):
         )
         return self.fc_out(output)
 
-# ============ ДИАЛОГИ (Сокращенная версия для примера, но грамматически верная) ============
+# ============ ДИАЛОГИ ============
 DIALOGUES = [
     ("привет", "привет как твои дела"),
     ("как дела", "у меня всё хорошо а у тебя"),
@@ -194,7 +198,6 @@ DIALOGUES = [
     ("ты честный", "честность основа доверия"),
     ("ты справедливый", "я стараюсь быть объективным"),
     ("ты мудрый", "мудрость приходит с опытом"),
-    ("ты умный", "спасибо я развиваю свой интеллект"),
     ("ты логичный", "логика это мой язык"),
     ("ты рациональный", "я принимаю решения на основе данных"),
     ("ты практичный", "я даю практические советы"),
@@ -347,88 +350,7 @@ DIALOGUES = [
     ("ты добродушный великан", "добродушный великан добр"),
     ("ты отзывчивый человек", "отзывчивый человек поможет"),
     ("ты сострадательный сердце", "сострадательное сердце болит"),
-    ("ты милосердный суд", "миłosердный суд прощает"),
-    ("ты прощающий родитель", "прощающий родитель любит"),
-    ("ты памятный день", "памятный день остается"),
-    ("ты внимательный глаз", "внимательный глаз видит"),
-    ("ты наблюдательный детектив", "наблюдательный детектив находит"),
-    ("ты зоркий сокол", "зоркий сокол видит добычу"),
-    ("ты дальнозоркий политик", "дальнозоркий политик планирует"),
-    ("ты здравый смысл", "здравый смысл нужен"),
-    ("ты здоровый дух", "здоровый дух в здоровом теле"),
-    ("ты крепкий орешек", "крепкие орешки ломаются"),
-    ("ты мощный компьютер", "мощный компьютер быстр"),
-    ("ты яркий свет", "яркий свет слепит"),
-    ("ты громкий голос", "громкий голос слышен далеко"),
-    ("ты звонкий колокол", "звонкий колокол зовет"),
-    ("ты музыкальный слух", "музыкальный слух редок"),
-    ("ты зрячий орел", "зрячий орел видит далеко"),
-    ("ты внимательный слушатель", "внимательный слушатель ценен"),
-    ("ты сосредоточенный взгляд", "сосредоточенный взгляд пронзает"),
-    ("ты активный отдых", "активный отдых полезен"),
-    ("ты энергичный танец", "энергичный танец бодрит"),
-    ("ты живой организм", "живой организм дышит"),
-    ("ты современный искусство", "современное искусство странно"),
-    ("ты древний город", "древний город помнит историю"),
-    ("ты новый год", "новый год приносит надежды"),
-    ("ты старый друг", "старый друг лучше новых двух"),
-    ("ты свежий хлеб", "свежий хлеб вкусен"),
-    ("ты чистый лист", "чистый лист ждет письма"),
-    ("ты аккуратный почерк", "аккуратный почерк радует"),
-    ("ты опрятный костюм", "опрятный костюм украшает"),
-    ("ты стильный образ", "стильный образ запоминается"),
-    ("ты изящный жест", "изящный жест красив"),
-    ("ты нежный цветок", "нежный цветок ломается"),
-    ("ты ласковый кот", "ласковый кот мурлычет"),
-    ("ты твердый камень", "твердый камень не сдается"),
-    ("ты решительный шаг", "решительный шаг важен"),
-    ("ты смелый поступок", "смелый поступок восхищает"),
-    ("ты храбрый воин", "храбрый воин защищает"),
-    ("ты уверенный тон", "уверенный тон убеждает"),
-    ("ты спокойный море", "спокойное море манит"),
-    ("ты уравновешенный человек", "уравновешенный человек стабилен"),
-    ("ты терпеливый ожидание", "терпеливое ожидание награждается"),
-    ("ты настойчивый поиск", "настойчивый поиск находит"),
-    ("ты упорный труд", "упорный труд дает плоды"),
-    ("ты трудолюбивый пчела", "трудолюбивая пчела собирает"),
-    ("ты прилежный ученик", "прилежный ученик учится"),
-    ("ты ответственный подход", "ответственный подход надежен"),
-    ("ты дисциплинированный солдат", "дисциплинированный солдат исполняет"),
-    ("ты организованный процесс", "организованный процесс эффективен"),
-    ("ты пунктуальный человек", "пунктуальный человек точен"),
-    ("ты точный расчет", "точный расчет верен"),
-    ("ты регулярный рейс", "регулярный рейс надежен"),
-    ("ты постоянный клиент", "постоянный клиент важен"),
-    ("ты стабильный доход", "стабильный доход радует"),
-    ("ты устойчивый стол", "устойчивый стол не падает"),
-    ("ты гибкий график", "гибкий график удобен"),
-    ("ты адаптивный дизайн", "адаптивный дизайн подстраивается"),
-    ("ты универсальный пульт", "универсальный пульт управляет"),
-    ("ты глобальный потепление", "глобальное потепление опасно"),
-    ("ты масштабный проект", "масштабный проект сложен"),
-    ("ты индивидуальный заказ", "индивидуальный заказ уникален"),
-    ("ты коллективный договор", "коллективный договор защищает"),
-    ("ты личный кабинет", "личный кабинет приватен"),
-    ("ты общественный транспорт", "общественный транспорт доступен"),
-    ("ты частный дом", "частный дом уютен"),
-    ("ты коммерческий банк", "коммерческий банк считает"),
-    ("ты платный сервис", "платный сервис качествен"),
-    ("ты выгодный обмен", "выгодный обмен радует"),
-    ("ты прибыльный бизнес", "прибыльный бизнес успешен"),
-    ("ты экономный режим", "экономный режим бережет"),
-    ("ты богатый урожай", "богатый урожай радует"),
-    ("ты щедрый подарок", "щедрый подарок приятен"),
-    ("ты великодушный жест", "великодушный жест благороден"),
-    ("ты благородный металл", "благородный металл блестит"),
-    ("ты честный бой", "честный бой справедлив"),
-    ("ты правдивый рассказ", "правдивый рассказ убедителен"),
-    ("ты искренний смех", "искренний смех заразителен"),
-    ("ты дружелюбный сосед", "дружелюбный сосед помогает"),
-    ("ты миролюбивый народ", "миролюбивый народ хочет мира"),
-    ("ты добродушный великан", "добродушный великан добр"),
-    ("ты отзывчивый человек", "отзывчивый человек поможет"),
-    ("ты сострадательный сердце", "сострадательное сердце болит"),
-    ("ты милосердный суд", "миłosердный суд прощает"),
+    ("ты милосердный суд", "милосердный суд прощает"),
     ("ты прощающий родитель", "прощающий родитель любит"),
     ("ты памятный день", "памятный день остается"),
     ("ты внимательный глаз", "внимательный глаз видит"),
@@ -480,14 +402,14 @@ class AndreyAI:
             'num_encoder_layers': 2, 'num_decoder_layers': 2, 'dim_feedforward': 256,
             'word_to_idx': word_to_idx,
             'idx_to_word': {str(k): v for k, v in idx_to_word.items()},
-            'memory': self.memory, 'version': '9.1-Grammar-Button',
+            'memory': self.memory, 'version': '9.2-Gradio-Dict-Fix',
             'created': datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         }
         torch.save(state, self.bin_file)
         print(f"✅ Сохранено в /data: {os.path.getsize(self.bin_file)/1024:.1f} КБ")
 
     @spaces.GPU(duration=120)
-    def train(self, epochs=20): # Уменьшено до 20 для быстрого теста
+    def train(self, epochs=20):
         print("🚀 Начало обучения грамотного Андрея...")
         device = torch.device('cuda')
         self.model.to(device)
@@ -544,11 +466,18 @@ class AndreyAI:
         
         q = question.lower().strip()
         context_parts = []
+        
+        # Обработка нового формата истории (словари)
         if history:
-            for user_msg, bot_msg in history[-2:]:
-                context_parts.append(user_msg)
-                context_parts.append(bot_msg)
-        context_parts.append(q)
+            for msg in history[-4:]: # Берем последние 4 сообщения (2 пары)
+                role = msg.get("role", "unknown")
+                content = msg.get("content", "")
+                if role == "user":
+                    context_parts.append(f"пользователь говорит {content}")
+                elif role == "assistant":
+                    context_parts.append(f"андрей говорит {content}")
+                    
+        context_parts.append(f"пользователь говорит {q}")
         full_context = " ".join(context_parts)
         
         ctx_tokens = tokenize(full_context)
@@ -587,20 +516,33 @@ class AndreyAI:
 
         return generated_text if generated_text else self.get_fallback_answer(q)
 
-# ============ GRADIO ИНТЕРФЕЙС ============
+# ============ GRADIO ИНТЕРФЕЙС (ИСПРАВЛЕННЫЙ ФОРМАТ) ============
 def gradio_chat(message, history):
-    if not message: 
-        return "", history
+    if not message:
+        return "", history or []
+    
+    if history is None:
+        history = []
     
+    # Генерируем ответ, передавая историю в новом формате
     answer = andrey.generate(message, history=history)
-    new_history = history + [(message, answer)]
+    
+    # Добавляем сообщения в строгом формате Gradio: список словарей
+    new_history = history + [
+        {"role": "user", "content": message},
+        {"role": "assistant", "content": answer}
+    ]
+    
+    # Сохраняем в память модели
+    andrey.memory['chat_history'] = new_history
+    
     return "", new_history
 
 def start_training():
     return andrey.train(epochs=20)
 
 with gr.Blocks(title="Андрей AI") as demo:
-    gr.Markdown("# 🤖 Андрей AI (Грамотный)\n### Transformer с правильной речью")
+    gr.Markdown("# 🤖 Андрей AI (Грамотный)\n### Transformer с правильной речью и новой памятью")
     
     chatbot = gr.Chatbot(height=400, label="Диалог")
     msg = gr.Textbox(label="Сообщение", placeholder="Напишите что-нибудь...")