Update chat.js

static/js/ui/chat.js

@@ -361,26 +361,34 @@ export async function addMessageToUI(message, index, options = {}, existingEleme
 
     let finalElement = existingElement;
 
-    // *** منطق جدید و دقیق برای پاکسازی دکمه‌های پیام‌های قبلی ***
+    // *** منطق جدید و دقیق برای پاکسازی دکمه‌ها ***
     if (!existingElement) {
-        // 1. حذف دکمه ویرایش از آخرین پیام کاربر (اگر وجود دارد)
-        const lastUserMsg = dom.chatWindow.querySelector('.message-entry.user:last-of-type');
-        if (lastUserMsg) {
-             const idx = parseInt(lastUserMsg.dataset.index, 10);
-             const chat = state.getActiveChat();
-             if(chat && chat.messages[idx]) {
-                 updateMessageActions(lastUserMsg, chat.messages[idx], false, false);
-             }
-        }
+        const activeChat = state.getActiveChat();
+
+        if (isUser) {
+            // وقتی کاربر پیام جدید میده:
+            // 1. دکمه ویرایش از تمام پیام‌های قبلی کاربر حذف بشه
+            const prevUserMsgs = dom.chatWindow.querySelectorAll('.message-entry.user');
+            prevUserMsgs.forEach(el => {
+                const idx = parseInt(el.dataset.index, 10);
+                if (activeChat.messages[idx]) updateMessageActions(el, activeChat.messages[idx], false, false);
+            });
 
-        // 2. حذف دکمه تلاش مجدد از آخرین پیام مدل (اگر وجود دارد)
-        const lastModelMsg = dom.chatWindow.querySelector('.message-entry.model:last-of-type');
-        if (lastModelMsg) {
-             const idx = parseInt(lastModelMsg.dataset.index, 10);
-             const chat = state.getActiveChat();
-             if(chat && chat.messages[idx]) {
-                 updateMessageActions(lastModelMsg, chat.messages[idx], false, false);
-             }
+            // 2. دکمه تلاش مجدد از تمام پیام‌های قبلی هوش مصنوعی حذف بشه (چون بحث ادامه پیدا کرده)
+            const prevModelMsgs = dom.chatWindow.querySelectorAll('.message-entry.model');
+            prevModelMsgs.forEach(el => {
+                const idx = parseInt(el.dataset.index, 10);
+                if (activeChat.messages[idx]) updateMessageActions(el, activeChat.messages[idx], false, false);
+            });
+        } 
+        else {
+            // وقتی هوش مصنوعی پیام جدید میده:
+            // 1. دکمه تلاش مجدد از پیام‌های قبلی هوش مصنوعی حذف بشه
+            const prevModelMsgs = dom.chatWindow.querySelectorAll('.message-entry.model');
+            prevModelMsgs.forEach(el => {
+                const idx = parseInt(el.dataset.index, 10);
+                if (activeChat.messages[idx]) updateMessageActions(el, activeChat.messages[idx], false, false);
+            });
         }
     }