Update chat.js

static/js/ui/chat.js

@@ -361,21 +361,26 @@ export async function addMessageToUI(message, index, options = {}, existingEleme
 
     let finalElement = existingElement;
 
-    // *** منطق جدید: پاک کردن دکمه‌های پیام قبلی ***
-    // اگر این پیام جدید، "آخرین" پیام کاربر یا مدل است، دکمه‌های پیام‌های قبلی همین نقش را حذف کن
-    if (!existingElement && (isLastUser || isLastModel)) {
-        const roleClass = isUser ? 'user' : 'model';
-        const allMessagesOfRole = dom.chatWindow.querySelectorAll(`.message-entry.${roleClass}`);
-        if (allMessagesOfRole.length > 0) {
-            const lastMsgElement = allMessagesOfRole[allMessagesOfRole.length - 1];
-            const lastMsgIndex = parseInt(lastMsgElement.dataset.index, 10);
-            const activeChat = state.getActiveChat();
-            
-            // مطمئن شویم پیام معتبر است
-            if (activeChat && activeChat.messages[lastMsgIndex]) {
-                // بروزرسانی پیام قبلی با flagهای false برای حذف دکمه‌ها
-                updateMessageActions(lastMsgElement, activeChat.messages[lastMsgIndex], false, false);
-            }
+    // *** منطق جدید و دقیق برای پاکسازی دکمه‌های پیام‌های قبلی ***
+    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);
+             }
+        }
+
+        // 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);
+             }
         }
     }