update rewritten ui: flagging think or nothink

index.html

@@ -344,15 +344,20 @@
             }
         }
 
-        function splitThinking(fullText) {
-            const text = fullText.replace('<think>', '');
+        function splitThinking(fullText, thinkingEnabled) {
+            const text = fullText.replace('<think>', '').replace('<|im_end|>', '');
+
+            if (!thinkingEnabled) {
+                return { thinking: '', answer: text.trim() };
+            }
+
             const pos = text.indexOf(THINK_CLOSE);
             if (pos === -1) {
                 return { thinking: text.trim(), answer: '' };
             }
             return {
                 thinking: text.slice(0, pos).trim(),
-                answer: text.slice(pos + THINK_CLOSE.length).replace('<|im_end|>', '').trim()
+                answer: text.slice(pos + THINK_CLOSE.length).trim()
             };
         }
 
@@ -379,7 +384,8 @@
         function updateBotMessage(div, fullText) {
             const thinkingContainer = div.querySelector('.thinking-container');
             const contentContainer = div.querySelector('.content-container');
-            const { thinking, answer } = splitThinking(fullText);
+            const thinkingEnabled = thinkingToggle.classList.contains('active');
+            const { thinking, answer } = splitThinking(fullText, thinkingEnabled);
 
             if (thinking) {
                 thinkingContainer.classList.remove('hidden');
@@ -391,16 +397,16 @@
 
             if (answer) {
                 contentContainer.innerHTML = marked.parse(answer);
-            } else if (!thinking) {
-                contentContainer.innerHTML = marked.parse(fullText.replace('<|im_end|>', ''));
-            } else {
+            } else if (thinkingEnabled && thinking) {
                 contentContainer.innerHTML = '<div class="flex gap-1.5 py-1"><div class="typing-dot"></div><div class="typing-dot"></div><div class="typing-dot"></div></div>';
+            } else {
+                contentContainer.innerHTML = '';
             }
 
             renderMath(thinkingContainer);
             renderMath(contentContainer);
             chatScrollArea.scrollTo({ top: chatScrollArea.scrollHeight, behavior: 'smooth' });
-            return answer || fullText.replace(THINK_CLOSE, '').replace('<|im_end|>', '').trim();
+            return answer;
         }
 
         async function sendMessage() {