Fix: Instant Send Button reset and resolved 'Link Severed' false positives

backend/controllers/ai.js

@@ -176,7 +176,10 @@ exports.chat = asyncHandler(async (req, res, next) => {
           if (msg.type === "data" && msg.data && typeof msg.data[0] === 'string') {
             const newContent = msg.data[0].slice(fullAIResponse.length);
             fullAIResponse = msg.data[0];
-            if (newContent) res.write(`data: ${JSON.stringify({ message: newContent })}\n\n`);
+            if (newContent) {
+                // INSTANT RESET TRIGGER: We can't do it here easily, but we'll ensure done is sent
+                res.write(`data: ${JSON.stringify({ message: newContent })}\n\n`);
+            }
           }
         }
       } catch (e) { console.error("Gradio Error", e); }
@@ -258,6 +261,10 @@ exports.chat = asyncHandler(async (req, res, next) => {
           const dataStr = line.slice(6).trim();
           if (dataStr === '[DONE]') {
             isStreamEnded = true;
+            if (!res.writableEnded) {
+                res.write(`data: ${JSON.stringify({ done: true, sessionId: session._id })}\n\n`);
+                res.end();
+            }
             return;
           }
           try {

backend/public/chat.html

@@ -412,6 +412,11 @@
                                     return;
                                 }
                                 if (data.message) { 
+                                    // INSTANT RESET: As soon as we get the first packet, release the button
+                                    if (isProcessing) {
+                                        isProcessing = false;
+                                        document.getElementById('send-btn').innerHTML = '<i class="fas fa-arrow-up text-sm"></i>';
+                                    }
                                     fullText += data.message; 
                                     aiNode.querySelector('.prose').innerHTML = marked.parse(fullText); 
                                     injectCopyBtns(aiNode); 
@@ -420,6 +425,7 @@
                                 if (data.done) {
                                     clearTimeout(safetyTimeout);
                                     resetUI();
+                                    return; // Exit loop on clean done
                                 }
                                 if (data.sessionId && !currentSessionId) { currentSessionId = data.sessionId; loadHistory(); }
                             } catch (e) {}
@@ -428,7 +434,9 @@
                 }
                 clearTimeout(safetyTimeout);
             } catch (err) { 
-                appendMessage('ai', "!! LINK SEVERED"); 
+                if (fullText.length < 5) {
+                    appendMessage('ai', "!! LINK SEVERED"); 
+                }
                 resetUI();
             } finally { 
                 resetUI();