Update main.js

static/js/main.js

@@ -262,12 +262,10 @@ document.addEventListener('DOMContentLoaded', async () => {
             return;
         }
 
-        // *** START: ADDED MESSAGE LIMIT CHECK ***
         if (state.isMessageLimitReached(checkUserPremiumStatus())) {
             chatUI.showLimitReachedUpgrade();
             return;
         }
-        // *** END: ADDED MESSAGE LIMIT CHECK ***
 
         const activeChat = state.getActiveChat();
         if (!activeChat) return;
@@ -301,11 +299,7 @@ document.addEventListener('DOMContentLoaded', async () => {
             const newUserMessage = { role: 'user', parts: userParts, tool: state.getActiveTool() };
             activeChat.messages.push(newUserMessage);
 
-            // *** START: ADDED MESSAGE COUNT INCREMENT ***
-            // Increment count immediately after adding the message to the state,
-            // before making the API call.
             state.incrementMessageCount(checkUserPremiumStatus());
-            // *** END: ADDED MESSAGE COUNT INCREMENT ***
             
             await chatUI.addMessageToUI(newUserMessage, activeChat.messages.length - 1, {isLastUser: true, animate: true});
             
@@ -313,25 +307,20 @@ document.addEventListener('DOMContentLoaded', async () => {
             activeChat.messages.push(modelPlaceholderMessage);
             modelBubbleOuterDiv = await chatUI.addMessageToUI(modelPlaceholderMessage, activeChat.messages.length - 1, {animate: true});
             
-            // *** START: MODIFIED - منطق پاکسازی تاریخچه اینجا اضافه شده است ***
             const historyForApi = activeChat.messages.map((msg, index) => {
-                const isLastMessage = index === activeChat.messages.length - 1; // این پیام placeholder مدل است
-                const isUserMessageJustSent = index === activeChat.messages.length - 2; // این پیام کاربر است
+                const isLastMessage = index === activeChat.messages.length - 1; 
+                const isUserMessageJustSent = index === activeChat.messages.length - 2; 
 
-                // برای پیام کاربر که همین الان ارسال شده، تمام اطلاعات را نگه دار
                 if (isUserMessageJustSent) {
                     return JSON.parse(JSON.stringify(msg));
                 }
-                // برای پیام placeholder مدل نیز کاری انجام نده
                 if (isLastMessage) {
                     return msg;
                 }
 
-                // برای تمام پیام‌های قدیمی‌تر در تاریخچه...
                 const cleanMsg = { role: msg.role, parts: [] };
                 if (msg.parts) {
                     msg.parts.forEach(part => {
-                        // ... فقط و فقط بخش متنی را نگه دار.
                         if (part.text) {
                             cleanMsg.parts.push({ text: part.text });
                         }
@@ -339,7 +328,6 @@ document.addEventListener('DOMContentLoaded', async () => {
                 }
                 return cleanMsg;
             });
-            // *** END: MODIFIED ***
 
             const toolPrefix = state.getActiveToolPrefix();
             const lastUserMsgInHistory = historyForApi.findLast(m => m.role === 'user');
@@ -371,60 +359,28 @@ document.addEventListener('DOMContentLoaded', async () => {
                         progressBar.style.width = '100%';
                     });
                 }
-                
-                const response = await api.getChatStream(historyForApi, state.globalAbortController.signal);
-                await api.readStreamAndDisplay(response, modelBubbleOuterDiv);
-
-            } else {
-                const MAX_RETRIES = 2;
-                for (let attempt = 0; attempt <= MAX_RETRIES; attempt++) {
-                    if (state.globalAbortController.signal.aborted) throw new DOMException('Aborted by user', 'AbortError');
-                    
-                    const attemptController = new AbortController();
-                    const onGlobalAbort = () => attemptController.abort();
-                    state.globalAbortController.signal.addEventListener('abort', onGlobalAbort);
-
-                    try {
-                        const timeoutDuration = fileAttachedInThisTurn ? 8000 : 3000;
-                        if (attempt > 0) console.warn(`تلاش ${attempt + 1}/${MAX_RETRIES + 1}: سرور پاسخ نداد. تلاش مجدد...`);
-                        
-                        const timeoutPromise = new Promise((_, reject) => {
-                           const id = setTimeout(() => {
-                                clearTimeout(id);
-                                reject(new Error('ClientTimeout'));
-                           }, timeoutDuration);
-                        });
-
-                        const responsePromise = api.getChatStream(historyForApi, attemptController.signal);
-                        
-                        const response = await Promise.race([responsePromise, timeoutPromise]);
-                        
-                        await api.readStreamAndDisplay(response, modelBubbleOuterDiv);
-                        
-                        break;
-                    } catch (error) {
-                        attemptController.abort();
-                        if (error.name === 'AbortError') {
-                            throw error;
-                        }
-                        if (error.message === 'ClientTimeout') {
-                            if (attempt === MAX_RETRIES) {
-                                throw new Error('سرور پس از چندین تلاش پاسخگو نبود.');
-                            }
-                        } else {
-                            throw error;
-                        }
-                    } finally {
-                        state.globalAbortController.signal.removeEventListener('abort', onGlobalAbort);
-                    }
-                }
             }
+            
+            // *** تغییر اصلی: حذف محدودیت زمانی کلاینت ***
+            // دیگر خبری از Promise.race و تایمر نیست.
+            // کلاینت فقط منتظر می‌ماند تا سرور (که حالا بی‌نهایت تلاش می‌کند) پاسخ دهد.
+            
+            const response = await api.getChatStream(historyForApi, state.globalAbortController.signal);
+            await api.readStreamAndDisplay(response, modelBubbleOuterDiv);
+
 
         } catch (error) {
             if (error.name !== 'AbortError') {
-                console.error("خطا در هنگام تولید پیام:", error);
+                console.error("Silent Error detected:", error);
+                // اگر خطایی رخ داد، به کاربر چیزی نشان نمی‌دهیم یا فقط پیام توقف کوچک
+                // چون کاربر خواسته هیچ خطای "سرور پاسخگو نبود" نبیند.
                 if (modelBubbleOuterDiv) {
-                    chatUI.displayError(modelBubbleOuterDiv, error.message || 'یک خطای ناشناخته رخ داد.');
+                    // بجای پیام خطای قرمز، فقط یک متن کوچک
+                     const contentArea = modelBubbleOuterDiv.querySelector('.message-content') || modelBubbleOuterDiv;
+                     // اگر هنوز محتوایی تولید نشده بود، این متن را بگذار
+                     if(contentArea.innerText.trim() === '') {
+                        contentArea.innerHTML = '<p class="text-xs text-slate-400 mt-2 text-center p-4">...</p>';
+                     }
                 }
             } else {
                  if (modelBubbleOuterDiv && !modelBubbleOuterDiv.querySelector('.message-content')?.innerText.includes('متوقف شد')) {
@@ -495,15 +451,13 @@ document.addEventListener('DOMContentLoaded', async () => {
                     const lastUserMessage = activeChat.messages.findLast(m => m.role === 'user');
                     const lastTool = lastUserMessage?.tool;
 
-                    // *** START: MODIFIED - اعمال منطق پاکسازی به بازسازی ***
                     const historyForApi = activeChat.messages.map((msg, index) => {
-                        if (index === activeChat.messages.length - 1) return msg; // placeholder
-                        if (index === activeChat.messages.length - 2) return JSON.parse(JSON.stringify(msg)); // user message
+                        if (index === activeChat.messages.length - 1) return msg; 
+                        if (index === activeChat.messages.length - 2) return JSON.parse(JSON.stringify(msg)); 
                         const cleanMsg = { role: msg.role, parts: [] };
                         if (msg.parts) msg.parts.forEach(part => { if (part.text) cleanMsg.parts.push({ text: part.text }); });
                         return cleanMsg;
                     });
-                    // *** END: MODIFIED ***
                     
                     if (lastTool === 'deep-think' || lastTool === 'reasoning') {
                         if (lastTool === 'deep-think') toolUI.updateDeepThinkPanel({ topic: lastUserMessage.parts.find(p=>p.text)?.text || 'فایل ضمیمه شده' }, newModelBubble);
@@ -515,12 +469,11 @@ document.addEventListener('DOMContentLoaded', async () => {
                                 progressBar.style.width = '100%';
                             });
                         }
-                        const response = await api.getChatStream(historyForApi, state.globalAbortController.signal);
-                        await api.readStreamAndDisplay(response, newModelBubble);
-                    } else {
-                        const response = await api.getChatStream(historyForApi, state.globalAbortController.signal);
-                        await api.readStreamAndDisplay(response, newModelBubble);
                     }
+                    
+                    const response = await api.getChatStream(historyForApi, state.globalAbortController.signal);
+                    await api.readStreamAndDisplay(response, newModelBubble);
+
                 } catch(error) {
                     if (error.name !== 'AbortError') console.error("Regeneration failed:", error);
                 } finally {
@@ -566,14 +519,12 @@ document.addEventListener('DOMContentLoaded', async () => {
                             activeChat.messages.push(modelPlaceholderMessage);
                             const newModelBubble = await chatUI.addMessageToUI(modelPlaceholderMessage, activeChat.messages.length - 1, { animate: true });
                             
-                            // *** START: MODIFIED - اعمال منطق پاکسازی به ویرایش ***
                             const historyForApi = activeChat.messages.map((msg, index) => {
                                 if (index >= activeChat.messages.length - 2) return JSON.parse(JSON.stringify(msg));
                                 const cleanMsg = { role: msg.role, parts: [] };
                                 if (msg.parts) msg.parts.forEach(part => { if (part.text) cleanMsg.parts.push({ text: part.text }); });
                                 return cleanMsg;
                             });
-                            // *** END: MODIFIED ***
 
                             const response = await api.getChatStream(historyForApi, state.globalAbortController.signal);
                             await api.readStreamAndDisplay(response, newModelBubble);
@@ -700,4 +651,4 @@ document.addEventListener('DOMContentLoaded', async () => {
     parent.postMessage({ type: 'REQUEST_USER_DATA_SIMPLE_CHECK' }, '*');
 });
 
-window.handleSuggestionClick = chatUI.handleSuggestionClick;
+window.handleSuggestionClick = chatUI.handleSuggestionClick;