@kirosnn/mosaic 0.71.0 → 0.73.0

package/src/components/Main.tsx

@@ -1,19 +1,16 @@
 import { useState, useEffect, useRef } from "react";
 import type { ImagePart, TextPart, UserContent } from "ai";
-import { useKeyboard } from "@opentui/react";
-import { Agent } from "../agent";
-import { saveConversation, addInputToHistory, type ConversationHistory, type ConversationStep } from "../utils/history";
-import { readConfig } from "../utils/config";
-import { DEFAULT_MAX_TOOL_LINES, formatToolMessage, formatErrorMessage, parseToolHeader } from '../utils/toolFormatting';
-import { initializeCommands, isCommand, executeCommand } from '../utils/commands';
-import type { InputSubmitMeta } from './CustomInput';
-
-import { subscribeQuestion, type QuestionRequest } from "../utils/questionBridge";
-import { subscribeApprovalAccepted, type ApprovalAccepted } from "../utils/approvalBridge";
-import { subscribeUndoRedo } from "../utils/undoRedoBridge";
-import { setExploreAbortController, setExploreToolCallback, abortExplore } from "../utils/exploreBridge";
-import { initializeSession, saveState } from "../utils/undoRedo";
-import { resetFileChanges } from "../utils/fileChangeTracker";
+import { useKeyboard } from "@opentui/react";
+import { Agent } from "../agent";
+import { saveConversation, addInputToHistory, type ConversationHistory, type ConversationStep } from "../utils/history";
+import { readConfig } from "../utils/config";
+import { DEFAULT_MAX_TOOL_LINES, formatToolMessage, formatErrorMessage, parseToolHeader } from '../utils/toolFormatting';
+import { initializeCommands, isCommand, executeCommand } from '../utils/commands';
+import type { InputSubmitMeta } from './CustomInput';
+
+import { subscribeQuestion, type QuestionRequest } from "../utils/questionBridge";
+import { subscribeApprovalAccepted } from "../utils/approvalBridge";
+import { setExploreAbortController, setExploreToolCallback } from "../utils/exploreBridge";
 import { getCurrentQuestion, cancelQuestion } from "../utils/questionBridge";
 import { getCurrentApproval, cancelApproval } from "../utils/approvalBridge";
 import { BLEND_WORDS, type MainProps, type Message } from "./main/types";
@@ -21,65 +18,210 @@ import { HomePage } from './main/HomePage';
 import { ChatPage } from './main/ChatPage';
 import type { ImageAttachment } from "../utils/images";
 import { subscribeImageCommand, setImageSupport } from "../utils/imageBridge";
-import { findModelsDevModelById, modelAcceptsImages } from "../utils/models";
-
-function extractTitle(content: string, alreadyResolved: boolean): { title: string | null; cleanContent: string; isPending: boolean; noTitle: boolean } {
-  const trimmed = content.trimStart();
-
-  const titleMatch = trimmed.match(/^<title>(.*?)<\/title>\s*/s);
-  if (titleMatch) {
-    const title = alreadyResolved ? null : (titleMatch[1]?.trim() || null);
-    const cleanContent = trimmed.replace(/^<title>.*?<\/title>\s*/s, '');
-    return { title, cleanContent, isPending: false, noTitle: false };
-  }
-
-  if (alreadyResolved) {
-    return { title: null, cleanContent: content, isPending: false, noTitle: false };
-  }
-
-  const partialTitlePattern = /^<(t(i(t(l(e(>.*)?)?)?)?)?)?$/i;
-  if (partialTitlePattern.test(trimmed) || (trimmed.startsWith('<title>') && !trimmed.includes('</title>'))) {
-    return { title: null, cleanContent: '', isPending: true, noTitle: false };
-  }
-
-  return { title: null, cleanContent: content, isPending: false, noTitle: true };
-}
-
-function setTerminalTitle(title: string) {
-  process.title = `⁘ ${title}`;
-}
-
-export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsOpen = false, commandsOpen = false, initialMessage }: MainProps) {
-  const [currentPage, setCurrentPage] = useState<"home" | "chat">(initialMessage ? "chat" : "home");
-  const [messages, setMessages] = useState<Message[]>([]);
-  const [isProcessing, setIsProcessing] = useState(false);
-  const [processingStartTime, setProcessingStartTime] = useState<number | null>(null);
-  const [currentTokens, setCurrentTokens] = useState(0);
-  const [scrollOffset, setScrollOffset] = useState(0);
-  const [terminalHeight, setTerminalHeight] = useState(process.stdout.rows || 24);
-  const [terminalWidth, setTerminalWidth] = useState(process.stdout.columns || 80);
+import { findModelsDevModelById, modelAcceptsImages, getModelsDevContextLimit } from "../utils/models";
+import { DEFAULT_SYSTEM_PROMPT, processSystemPrompt } from "../agent/prompts/systemPrompt";
+import { estimateTokensFromText, estimateTokensForContent, getDefaultContextBudget } from "../utils/tokenEstimator";
+
+type CompactableMessage = Pick<Message, "role" | "content" | "thinkingContent" | "toolName">;
+
+function extractTitle(content: string, alreadyResolved: boolean): { title: string | null; cleanContent: string; isPending: boolean; noTitle: boolean } {
+  const trimmed = content.trimStart();
+
+  const titleMatch = trimmed.match(/^<title>(.*?)<\/title>\s*/s);
+  if (titleMatch) {
+    const title = alreadyResolved ? null : (titleMatch[1]?.trim() || null);
+    const cleanContent = trimmed.replace(/^<title>.*?<\/title>\s*/s, '');
+    return { title, cleanContent, isPending: false, noTitle: false };
+  }
+
+  if (alreadyResolved) {
+    return { title: null, cleanContent: content, isPending: false, noTitle: false };
+  }
+
+  const partialTitlePattern = /^<(t(i(t(l(e(>.*)?)?)?)?)?)?$/i;
+  if (partialTitlePattern.test(trimmed) || (trimmed.startsWith('<title>') && !trimmed.includes('</title>'))) {
+    return { title: null, cleanContent: '', isPending: true, noTitle: false };
+  }
+
+  return { title: null, cleanContent: content, isPending: false, noTitle: true };
+}
+
+function setTerminalTitle(title: string) {
+  process.title = `⁘ ${title}`;
+}
+
+export function normalizeWhitespace(text: string): string {
+  return text.replace(/\s+/g, " ").trim();
+}
+
+export function truncateText(text: string, maxChars: number): string {
+  if (text.length <= maxChars) return text;
+  return text.slice(0, Math.max(0, maxChars - 3)) + "...";
+}
+
+export function estimateTokensForMessage(message: CompactableMessage): number {
+  return estimateTokensForContent(message.content || "", message.thinkingContent || undefined);
+}
+
+export function estimateTokensForMessages(messages: CompactableMessage[]): number {
+  return messages.reduce((sum, message) => sum + estimateTokensForMessage(message), 0);
+}
+
+export function estimateTotalTokens(messages: CompactableMessage[], systemPrompt: string): number {
+  const systemTokens = estimateTokensFromText(systemPrompt) + 8;
+  return systemTokens + estimateTokensForMessages(messages);
+}
+
+export function shouldAutoCompact(totalTokens: number, maxContextTokens: number): boolean {
+  if (!Number.isFinite(maxContextTokens) || maxContextTokens <= 0) return false;
+  const threshold = Math.floor(maxContextTokens * 0.95);
+  return totalTokens >= threshold;
+}
+
+export function summarizeMessage(message: CompactableMessage, isLastUser: boolean): string {
+  if (message.role === "tool") {
+    const name = message.toolName || "tool";
+    const text = message.content || "";
+    const isError = text.toLowerCase().includes('error') || text.toLowerCase().includes('failed');
+    const status = isError ? 'FAILED' : 'OK';
+    const cleaned = normalizeWhitespace(text);
+    return `[tool:${name} ${status}] ${truncateText(cleaned, 120)}`;
+  }
+
+  if (message.role === "assistant") {
+    const cleaned = normalizeWhitespace(message.content || "");
+    const sentenceMatch = cleaned.match(/^[^.!?\n]{10,}[.!?]/);
+    const summary = sentenceMatch ? sentenceMatch[0] : cleaned;
+    return `assistant: ${truncateText(summary, 200)}`;
+  }
+
+  const cleaned = normalizeWhitespace(message.content || "");
+  const limit = isLastUser ? cleaned.length : 400;
+  return `user: ${truncateText(cleaned, limit)}`;
+}
+
+export function buildSummary(messages: CompactableMessage[], maxTokens: number): string {
+  const maxChars = Math.max(0, maxTokens * 3);
+  const header = "Résumé de conversation (compact):";
+  let charCount = header.length + 1;
+  const lines: string[] = [];
+
+  let lastUserIndex = -1;
+  for (let i = messages.length - 1; i >= 0; i--) {
+    if (messages[i]!.role === 'user') { lastUserIndex = i; break; }
+  }
+
+  for (let i = 0; i < messages.length; i++) {
+    if (charCount >= maxChars) break;
+    const line = `- ${summarizeMessage(messages[i]!, i === lastUserIndex)}`;
+    charCount += line.length + 1;
+    lines.push(line);
+  }
+  const body = lines.join("\n");
+  const full = `${header}\n${body}`.trim();
+  return truncateText(full, maxChars);
+}
+
+export function collectContextFiles(messages: Message[]): string[] {
+  const files = new Set<string>();
+  for (const message of messages) {
+    if (message.role !== "tool") continue;
+    if (!message.toolArgs) continue;
+    const toolName = message.toolName || "";
+    if (!["read", "write", "edit", "list", "grep"].includes(toolName)) continue;
+    const path = message.toolArgs.path;
+    if (typeof path === "string" && path.trim()) {
+      files.add(path.trim());
+    }
+    const pattern = message.toolArgs.pattern;
+    if (toolName === "grep" && typeof pattern === "string" && pattern.trim()) {
+      files.add(pattern.trim());
+    }
+  }
+  return Array.from(files.values()).sort((a, b) => a.localeCompare(b));
+}
+
+export function appendContextFiles(summary: string, files: string[], maxTokens: number): string {
+  if (files.length === 0) return summary;
+  const maxChars = Math.max(0, maxTokens * 4);
+  const list = files.map(f => `- ${f}`).join("\n");
+  const block = `\n\nFichiers conservés après compaction:\n${list}`;
+  return truncateText(`${summary}${block}`, maxChars);
+}
+
+export function compactMessagesForUi(
+  messages: Message[],
+  systemPrompt: string,
+  maxContextTokens: number,
+  createId: () => string,
+  summaryOnly: boolean
+): { messages: Message[]; estimatedTokens: number; didCompact: boolean } {
+  const systemTokens = estimateTokensFromText(systemPrompt) + 8;
+  const totalTokens = systemTokens + estimateTokensForMessages(messages);
+  if (totalTokens <= maxContextTokens && !summaryOnly) {
+    return { messages, estimatedTokens: totalTokens - systemTokens, didCompact: false };
+  }
+
+  const summaryTokens = Math.min(2000, Math.max(400, Math.floor(maxContextTokens * 0.2)));
+  const recentBudget = Math.max(500, maxContextTokens - summaryTokens);
+
+  let recentTokens = 0;
+  const recent: Message[] = [];
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const message = messages[i]!;
+    const msgTokens = estimateTokensForMessage(message);
+    if (recentTokens + msgTokens > recentBudget && recent.length > 0) break;
+    recent.unshift(message);
+    recentTokens += msgTokens;
+  }
+
+  const cutoff = messages.length - recent.length;
+  const older = cutoff > 0 ? messages.slice(0, cutoff) : [];
+  const files = collectContextFiles(messages);
+  const summaryBase = buildSummary(summaryOnly ? messages : (older.length > 0 ? older : messages), summaryTokens);
+  const summary = appendContextFiles(summaryBase, files, summaryTokens);
+  const summaryMessage: Message = {
+    id: createId(),
+    role: "assistant",
+    content: summary
+  };
+
+  const nextMessages = summaryOnly ? [summaryMessage] : [summaryMessage, ...recent];
+  const estimatedTokens = estimateTokensForMessages(nextMessages);
+  return { messages: nextMessages, estimatedTokens, didCompact: true };
+}
+
+export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsOpen = false, commandsOpen = false, initialMessage }: MainProps) {
+  const [currentPage, setCurrentPage] = useState<"home" | "chat">(initialMessage ? "chat" : "home");
+  const [messages, setMessages] = useState<Message[]>([]);
+  const [isProcessing, setIsProcessing] = useState(false);
+  const [processingStartTime, setProcessingStartTime] = useState<number | null>(null);
+  const [currentTokens, setCurrentTokens] = useState(0);
+  const [scrollOffset, setScrollOffset] = useState(0);
+  const [terminalHeight, setTerminalHeight] = useState(process.stdout.rows || 24);
+  const [terminalWidth, setTerminalWidth] = useState(process.stdout.columns || 80);
   const [questionRequest, setQuestionRequest] = useState<QuestionRequest | null>(null);
   const [currentTitle, setCurrentTitle] = useState<string | null>(null);
   const [pendingImages, setPendingImages] = useState<ImageAttachment[]>([]);
   const [imagesSupported, setImagesSupported] = useState(false);
-  const currentTitleRef = useRef<string | null>(null);
-  const titleExtractedRef = useRef(false);
-  const shouldAutoScroll = useRef(true);
-  const abortControllerRef = useRef<AbortController | null>(null);
-  const currentPageRef = useRef(currentPage);
-  const shortcutsOpenRef = useRef(shortcutsOpen);
-  const commandsOpenRef = useRef(commandsOpen);
-  const questionRequestRef = useRef<QuestionRequest | null>(questionRequest);
-  const initialMessageProcessed = useRef(false);
-  const exploreMessageIdRef = useRef<string | null>(null);
-  const exploreToolsRef = useRef<Array<{ tool: string; info: string; success: boolean }>>([]);
-  const explorePurposeRef = useRef<string>('');
-
-  const createId = () => `${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
-
+  const currentTitleRef = useRef<string | null>(null);
+  const titleExtractedRef = useRef(false);
+  const shouldAutoScroll = useRef(true);
+  const abortControllerRef = useRef<AbortController | null>(null);
+  const currentPageRef = useRef(currentPage);
+  const shortcutsOpenRef = useRef(shortcutsOpen);
+  const commandsOpenRef = useRef(commandsOpen);
+  const questionRequestRef = useRef<QuestionRequest | null>(questionRequest);
+  const initialMessageProcessed = useRef(false);
+  const lastPromptTokensRef = useRef<number>(0);
+  const exploreMessageIdRef = useRef<string | null>(null);
+  const exploreToolsRef = useRef<Array<{ tool: string; info: string; success: boolean }>>([]);
+  const explorePurposeRef = useRef<string>('');
+
+  const createId = () => `${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;
+
   useEffect(() => {
     initializeCommands();
-    initializeSession();
   }, []);
 
   useEffect(() => {
@@ -102,50 +244,41 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
     };
     loadSupport();
   }, []);
-
-  useEffect(() => {
-    let lastExploreTokens = 0;
-    setExploreToolCallback((toolName, args, result, totalTokens) => {
-      const info = (args.path || args.pattern || args.query || '') as string;
-      const shortInfo = info.length > 40 ? info.substring(0, 37) + '...' : info;
-      exploreToolsRef.current.push({ tool: toolName, info: shortInfo, success: result.success });
-
-      const tokenDelta = totalTokens - lastExploreTokens;
-      lastExploreTokens = totalTokens;
-      if (tokenDelta > 0) {
-        setCurrentTokens(prev => prev + tokenDelta);
-      }
-
-      if (exploreMessageIdRef.current) {
-        setMessages((prev: Message[]) => {
-          const newMessages = [...prev];
-          const idx = newMessages.findIndex(m => m.id === exploreMessageIdRef.current);
-          if (idx !== -1) {
-            const toolLines = exploreToolsRef.current.map(t => {
-              const icon = t.success ? '→' : '-';
-              return `  ${icon} ${t.tool}(${t.info})`;
-            });
-            const purpose = explorePurposeRef.current;
-            const newContent = `Explore (${purpose})\n${toolLines.join('\n')}`;
-            newMessages[idx] = { ...newMessages[idx]!, content: newContent };
-          }
-          return newMessages;
-        });
-      }
-    });
-
-    return () => {
-      setExploreToolCallback(null);
-    };
-  }, []);
-
+
   useEffect(() => {
-    return subscribeUndoRedo((state, action) => {
-      if (state) {
-        setMessages(state.messages);
-        resetFileChanges();
+    let lastExploreTokens = 0;
+    setExploreToolCallback((toolName, args, result, totalTokens) => {
+      const info = (args.path || args.pattern || args.query || '') as string;
+      const shortInfo = info.length > 40 ? info.substring(0, 37) + '...' : info;
+      exploreToolsRef.current.push({ tool: toolName, info: shortInfo, success: result.success });
+
+      const tokenDelta = totalTokens - lastExploreTokens;
+      lastExploreTokens = totalTokens;
+      if (tokenDelta > 0) {
+        setCurrentTokens(prev => prev + tokenDelta);
+      }
+
+      if (exploreMessageIdRef.current) {
+        setMessages((prev: Message[]) => {
+          const newMessages = [...prev];
+          const idx = newMessages.findIndex(m => m.id === exploreMessageIdRef.current);
+          if (idx !== -1) {
+            const toolLines = exploreToolsRef.current.map(t => {
+              const icon = t.success ? '→' : '-';
+              return `  ${icon} ${t.tool}(${t.info})`;
+            });
+            const purpose = explorePurposeRef.current;
+            const newContent = `Explore (${purpose})\n${toolLines.join('\n')}`;
+            newMessages[idx] = { ...newMessages[idx]!, content: newContent };
+          }
+          return newMessages;
+        });
       }
     });
+
+    return () => {
+      setExploreToolCallback(null);
+    };
   }, []);
 
   useEffect(() => {
@@ -168,146 +301,145 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
       setPendingImages([]);
     }
   }, [imagesSupported]);
-
-  useEffect(() => {
-    const handleResize = () => {
-      const newWidth = process.stdout.columns || 80;
-      const newHeight = process.stdout.rows || 24;
-      const oldWidth = terminalWidth;
-      const oldHeight = terminalHeight;
-
-      setTerminalWidth(newWidth);
-      setTerminalHeight(newHeight);
-
-      if (shouldAutoScroll.current) {
-        setScrollOffset(0);
-      } else if (oldHeight !== newHeight) {
-        const heightDiff = newHeight - oldHeight;
-        setScrollOffset(prev => Math.max(0, prev - heightDiff));
-      }
-    };
-    process.stdout.on('resize', handleResize);
-    return () => {
-      process.stdout.off('resize', handleResize);
-    };
-  }, [terminalWidth, terminalHeight]);
-
-  useEffect(() => {
-    return subscribeQuestion(setQuestionRequest);
-  }, []);
-
-  useEffect(() => {
-    return subscribeApprovalAccepted((accepted) => {
-      const isBashTool = accepted.toolName === 'bash';
-
-      if (isBashTool) {
-        const { name: toolDisplayName, info: toolInfo } = parseToolHeader(accepted.toolName, accepted.args);
-        const runningContent = toolInfo ? `${toolDisplayName} (${toolInfo})` : toolDisplayName;
-
-        setMessages((prev: Message[]) => {
-          const newMessages = [...prev];
-          newMessages.push({
-            id: createId(),
-            role: "tool",
-            content: runningContent,
-            toolName: accepted.toolName,
-            toolArgs: accepted.args,
-            success: true,
-            isRunning: true,
-            runningStartTime: Date.now()
-          });
-          return newMessages;
-        });
-      }
-    });
-  }, []);
-
-  useEffect(() => {
-    currentPageRef.current = currentPage;
-  }, [currentPage]);
-
-  useEffect(() => {
-    shortcutsOpenRef.current = shortcutsOpen;
-  }, [shortcutsOpen]);
-
-  useEffect(() => {
-    commandsOpenRef.current = commandsOpen;
-  }, [commandsOpen]);
-
-  useEffect(() => {
-    questionRequestRef.current = questionRequest;
-  }, [questionRequest]);
-
-  useEffect(() => {
-    if (questionRequest) {
-      shouldAutoScroll.current = true;
-      setScrollOffset(0);
-    }
-  }, [questionRequest]);
-
-  useEffect(() => {
-    if (currentPage !== "chat") return;
-
-    process.stdin.setRawMode(true);
-    process.stdout.write('\x1b[?1000h');
-    process.stdout.write('\x1b[?1003h');
-    process.stdout.write('\x1b[?1006h');
-
-    const handleData = (data: Buffer) => {
-      const str = data.toString();
-
-      if (str.match(/\x1b\[<(\d+);(\d+);(\d+)([mM])/)) {
-        const match = str.match(/\x1b\[<(\d+);(\d+);(\d+)([mM])/);
-        if (match) {
-          const button = parseInt(match[1] || '0');
-
-          if (button === 64) {
-            shouldAutoScroll.current = false;
-            setScrollOffset((prev) => prev + 1);
-          } else if (button === 65) {
-            setScrollOffset((prev) => {
-              const newOffset = Math.max(0, prev - 1);
-              if (newOffset === 0) {
-                shouldAutoScroll.current = true;
-              }
-              return newOffset;
-            });
-          }
-        }
-      }
-    };
-
-    process.stdin.on('data', handleData);
-
-    return () => {
-      process.stdin.off('data', handleData);
-      process.stdout.write('\x1b[?1000l');
-      process.stdout.write('\x1b[?1003l');
-      process.stdout.write('\x1b[?1006l');
-    };
-  }, [currentPage]);
-
-  useEffect(() => {
-    if (currentPage === "chat") {
-      setScrollOffset((prevOffset) => {
-        if (shouldAutoScroll.current || prevOffset < 5) {
-          shouldAutoScroll.current = true;
-          return 0;
-        }
-        return prevOffset;
-      });
-    }
-  }, [messages, currentPage]);
-
-  useEffect(() => {
-    if (copyRequestId > 0 && onCopy && messages.length > 0) {
-      const lastAssistantMessage = messages.slice().reverse().find(m => m.role === 'assistant');
-      if (lastAssistantMessage) {
-        onCopy(lastAssistantMessage.content);
-      }
-    }
-  }, [copyRequestId, onCopy, messages]);
-
+
+  useEffect(() => {
+    const handleResize = () => {
+      const newWidth = process.stdout.columns || 80;
+      const newHeight = process.stdout.rows || 24;
+      const oldHeight = terminalHeight;
+
+      setTerminalWidth(newWidth);
+      setTerminalHeight(newHeight);
+
+      if (shouldAutoScroll.current) {
+        setScrollOffset(0);
+      } else if (oldHeight !== newHeight) {
+        const heightDiff = newHeight - oldHeight;
+        setScrollOffset(prev => Math.max(0, prev - heightDiff));
+      }
+    };
+    process.stdout.on('resize', handleResize);
+    return () => {
+      process.stdout.off('resize', handleResize);
+    };
+  }, [terminalWidth, terminalHeight]);
+
+  useEffect(() => {
+    return subscribeQuestion(setQuestionRequest);
+  }, []);
+
+  useEffect(() => {
+    return subscribeApprovalAccepted((accepted) => {
+      const isBashTool = accepted.toolName === 'bash';
+
+      if (isBashTool) {
+        const { name: toolDisplayName, info: toolInfo } = parseToolHeader(accepted.toolName, accepted.args);
+        const runningContent = toolInfo ? `${toolDisplayName} (${toolInfo})` : toolDisplayName;
+
+        setMessages((prev: Message[]) => {
+          const newMessages = [...prev];
+          newMessages.push({
+            id: createId(),
+            role: "tool",
+            content: runningContent,
+            toolName: accepted.toolName,
+            toolArgs: accepted.args,
+            success: true,
+            isRunning: true,
+            runningStartTime: Date.now()
+          });
+          return newMessages;
+        });
+      }
+    });
+  }, []);
+
+  useEffect(() => {
+    currentPageRef.current = currentPage;
+  }, [currentPage]);
+
+  useEffect(() => {
+    shortcutsOpenRef.current = shortcutsOpen;
+  }, [shortcutsOpen]);
+
+  useEffect(() => {
+    commandsOpenRef.current = commandsOpen;
+  }, [commandsOpen]);
+
+  useEffect(() => {
+    questionRequestRef.current = questionRequest;
+  }, [questionRequest]);
+
+  useEffect(() => {
+    if (questionRequest) {
+      shouldAutoScroll.current = true;
+      setScrollOffset(0);
+    }
+  }, [questionRequest]);
+
+  useEffect(() => {
+    if (currentPage !== "chat") return;
+
+    process.stdin.setRawMode(true);
+    process.stdout.write('\x1b[?1000h');
+    process.stdout.write('\x1b[?1003h');
+    process.stdout.write('\x1b[?1006h');
+
+    const handleData = (data: Buffer) => {
+      const str = data.toString();
+
+      if (str.match(/\x1b\[<(\d+);(\d+);(\d+)([mM])/)) {
+        const match = str.match(/\x1b\[<(\d+);(\d+);(\d+)([mM])/);
+        if (match) {
+          const button = parseInt(match[1] || '0');
+
+          if (button === 64) {
+            shouldAutoScroll.current = false;
+            setScrollOffset((prev) => prev + 1);
+          } else if (button === 65) {
+            setScrollOffset((prev) => {
+              const newOffset = Math.max(0, prev - 1);
+              if (newOffset === 0) {
+                shouldAutoScroll.current = true;
+              }
+              return newOffset;
+            });
+          }
+        }
+      }
+    };
+
+    process.stdin.on('data', handleData);
+
+    return () => {
+      process.stdin.off('data', handleData);
+      process.stdout.write('\x1b[?1000l');
+      process.stdout.write('\x1b[?1003l');
+      process.stdout.write('\x1b[?1006l');
+    };
+  }, [currentPage]);
+
+  useEffect(() => {
+    if (currentPage === "chat") {
+      setScrollOffset((prevOffset) => {
+        if (shouldAutoScroll.current || prevOffset < 5) {
+          shouldAutoScroll.current = true;
+          return 0;
+        }
+        return prevOffset;
+      });
+    }
+  }, [messages, currentPage]);
+
+  useEffect(() => {
+    if (copyRequestId > 0 && onCopy && messages.length > 0) {
+      const lastAssistantMessage = messages.slice().reverse().find(m => m.role === 'assistant');
+      if (lastAssistantMessage) {
+        onCopy(lastAssistantMessage.content);
+      }
+    }
+  }, [copyRequestId, onCopy, messages]);
+
   useKeyboard((key) => {
     if ((key.name === 'c' && key.ctrl) || key.sequence === '\x03') {
       if (getCurrentQuestion()) {
@@ -324,11 +456,11 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
       if (getCurrentQuestion()) {
         cancelQuestion();
       }
-      if (getCurrentApproval()) {
-        cancelApproval();
-      }
-      abortControllerRef.current?.abort();
-      return;
+      if (getCurrentApproval()) {
+        cancelApproval();
+      }
+      abortControllerRef.current?.abort();
+      return;
     }
   });
 
@@ -360,343 +492,376 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
     const hasPastedContent = Boolean(meta?.isPaste && meta.pastedContent);
     const hasImages = imagesSupported && pendingImages.length > 0;
     if (!value.trim() && !hasPastedContent && !hasImages) return;
-
-    if (isCommand(value)) {
-      const result = await executeCommand(value);
-      if (result) {
-        if (result.shouldAddToHistory === true) {
-          addInputToHistory(value.trim());
-
-          saveState(messages);
-
-          const userMessage: Message = {
-            id: createId(),
-            role: "user",
-            content: result.content,
-            displayContent: value,
-          };
-
-          setMessages((prev: Message[]) => [...prev, userMessage]);
-          setIsProcessing(true);
-          const localStartTime = Date.now();
-          setProcessingStartTime(localStartTime);
-          setCurrentTokens(0);
-          shouldAutoScroll.current = true;
-
-          const conversationId = createId();
-          const conversationSteps: ConversationStep[] = [];
-          let totalTokens = { prompt: 0, completion: 0, total: 0 };
-          let stepCount = 0;
-          let totalChars = 0;
-          for (const m of messages) {
-            if (m.role === 'assistant') {
-              totalChars += m.content.length;
-              if (m.thinkingContent) totalChars += m.thinkingContent.length;
-            } else if (m.role === 'tool') {
-              totalChars += m.content.length;
-            }
-          }
-
-          const estimateTokens = () => Math.ceil(totalChars / 4);
-          setCurrentTokens(estimateTokens());
-          const config = readConfig();
-          const abortController = new AbortController();
-          abortControllerRef.current = abortController;
-          let abortNotified = false;
-          const notifyAbort = () => {
-            if (abortNotified) return;
-            abortNotified = true;
-            setMessages((prev: Message[]) => {
-              const newMessages = [...prev];
-              newMessages.push({
-                id: createId(),
-                role: "tool",
-                success: false,
-                content: "Request interrupted by user. \n↪ What should Mosaic do instead?"
-              });
-              return newMessages;
-            });
-          };
-
-          conversationSteps.push({
-            type: 'user',
-            content: result.content,
-            timestamp: Date.now()
-          });
-
+
+    if (isCommand(value)) {
+      const result = await executeCommand(value);
+      if (result) {
+        if (result.shouldClearMessages === true) {
+          const commandMessage: Message = {
+            id: createId(),
+            role: "slash",
+            content: result.content,
+            isError: !result.success
+          };
+          setMessages([commandMessage]);
+          return;
+        }
+
+        if (result.shouldCompactMessages === true) {
+          const config = readConfig();
+          const rawSystemPrompt = config.systemPrompt || DEFAULT_SYSTEM_PROMPT;
+          const systemPrompt = processSystemPrompt(rawSystemPrompt, true);
+          let maxContextTokens = result.compactMaxTokens ?? config.maxContextTokens;
+          if (!maxContextTokens && config.provider && config.model) {
+            const resolved = await getModelsDevContextLimit(config.provider, config.model);
+            if (typeof resolved === "number") {
+              maxContextTokens = resolved;
+            }
+          }
+          const targetTokens = maxContextTokens ?? getDefaultContextBudget(config.provider);
+          let nextTokens = currentTokens;
+          setMessages(prev => {
+            const compacted = compactMessagesForUi(prev, systemPrompt, targetTokens, createId, true);
+            nextTokens = compacted.estimatedTokens;
+            return compacted.messages;
+          });
+          setCurrentTokens(nextTokens);
+          return;
+        }
+        
+        if (result.shouldAddToHistory === true) {
+          addInputToHistory(value.trim());
+
+          const userMessage: Message = {
+            id: createId(),
+            role: "user",
+            content: result.content,
+            displayContent: value,
+          };
+
+          setMessages((prev: Message[]) => [...prev, userMessage]);
+          setIsProcessing(true);
+          const localStartTime = Date.now();
+          setProcessingStartTime(localStartTime);
+          setCurrentTokens(0);
+          lastPromptTokensRef.current = 0;
+          shouldAutoScroll.current = true;
+
+          const conversationId = createId();
+          const conversationSteps: ConversationStep[] = [];
+          let totalTokens = { prompt: 0, completion: 0, total: 0 };
+          let stepCount = 0;
+          let totalChars = 0;
+          for (const m of messages) {
+            if (m.role === 'assistant') {
+              totalChars += m.content.length;
+              if (m.thinkingContent) totalChars += m.thinkingContent.length;
+            } else if (m.role === 'tool') {
+              totalChars += m.content.length;
+            }
+          }
+
+          const estimateTokens = () => Math.ceil(totalChars / 4);
+          setCurrentTokens(estimateTokens());
+          const config = readConfig();
+          const abortController = new AbortController();
+          abortControllerRef.current = abortController;
+          let abortNotified = false;
+          const notifyAbort = () => {
+            if (abortNotified) return;
+            abortNotified = true;
+            setMessages((prev: Message[]) => {
+              const newMessages = [...prev];
+              newMessages.push({
+                id: createId(),
+                role: "tool",
+                success: false,
+                content: "Request interrupted by user. \n↪ What should Mosaic do instead?"
+              });
+              return newMessages;
+            });
+          };
+
+          conversationSteps.push({
+            type: 'user',
+            content: result.content,
+            timestamp: Date.now()
+          });
+
           let responseDuration: number | null = null;
-          let responseBlendWord: string | null = null;
+          let responseBlendWord: string | undefined = undefined;
 
           try {
             const providerStatus = await Agent.ensureProviderReady();
             if (!providerStatus.ready) {
-              setMessages((prev: Message[]) => {
-                const newMessages = [...prev];
-                newMessages.push({
-                  id: createId(),
-                  role: "assistant",
-                  content: `Ollama error: ${providerStatus.error || 'Could not start Ollama. Make sure Ollama is installed.'}`,
-                  isError: true
-                });
-                return newMessages;
-              });
-              setIsProcessing(false);
-              return;
-            }
-
-            const agent = new Agent();
+              setMessages((prev: Message[]) => {
+                const newMessages = [...prev];
+                newMessages.push({
+                  id: createId(),
+                  role: "assistant",
+                  content: `Ollama error: ${providerStatus.error || 'Could not start Ollama. Make sure Ollama is installed.'}`,
+                  isError: true
+                });
+                return newMessages;
+              });
+              setIsProcessing(false);
+              return;
+            }
+
+            const agent = new Agent();
             const conversationHistory = buildConversationHistory([...messages, userMessage], imagesSupported);
-            let assistantChunk = '';
-            let thinkingChunk = '';
-            const pendingToolCalls = new Map<string, { toolName: string; args: Record<string, unknown>; messageId?: string }>();
-            let assistantMessageId: string | null = null;
-            let streamHadError = false;
-            titleExtractedRef.current = false;
-
-            for await (const event of agent.streamMessages(conversationHistory, { abortSignal: abortController.signal })) {
-              if (event.type === 'reasoning-delta') {
-                thinkingChunk += event.content;
-                totalChars += event.content.length;
-                setCurrentTokens(estimateTokens());
-
-                if (assistantMessageId === null) {
-                  assistantMessageId = createId();
-                }
-
-                const currentMessageId = assistantMessageId;
-                setMessages((prev: Message[]) => {
-                  const newMessages = [...prev];
-                  const messageIndex = newMessages.findIndex(m => m.id === currentMessageId);
-
-                  if (messageIndex === -1) {
-                    newMessages.push({ id: currentMessageId, role: "assistant", content: '', thinkingContent: thinkingChunk });
-                  } else {
-                    newMessages[messageIndex] = {
-                      ...newMessages[messageIndex]!,
-                      thinkingContent: thinkingChunk
-                    };
-                  }
-                  return newMessages;
-                });
-              } else if (event.type === 'text-delta') {
-                assistantChunk += event.content;
-                totalChars += event.content.length;
-                setCurrentTokens(estimateTokens());
-
-                const { title, cleanContent, isPending, noTitle } = extractTitle(assistantChunk, titleExtractedRef.current);
-
-                if (title) {
-                  titleExtractedRef.current = true;
-                  currentTitleRef.current = title;
-                  setCurrentTitle(title);
-                  setTerminalTitle(title);
-                } else if (noTitle) {
-                  titleExtractedRef.current = true;
-                }
-
-                if (isPending) continue;
-
-                if (assistantMessageId === null) {
-                  assistantMessageId = createId();
-                }
-
-                const displayContent = cleanContent;
-                const currentMessageId = assistantMessageId;
-                setMessages((prev: Message[]) => {
-                  const newMessages = [...prev];
-                  const messageIndex = newMessages.findIndex(m => m.id === currentMessageId);
-
-                  if (messageIndex === -1) {
-                    newMessages.push({ id: currentMessageId, role: "assistant", content: displayContent, thinkingContent: thinkingChunk });
-                  } else {
-                    newMessages[messageIndex] = {
-                      ...newMessages[messageIndex]!,
-                      content: displayContent
-                    };
-                  }
-                  return newMessages;
-                });
-              } else if (event.type === 'step-start') {
-                stepCount++;
-              } else if (event.type === 'tool-call-end') {
-                totalChars += JSON.stringify(event.args).length;
-                setCurrentTokens(estimateTokens());
-
-                const needsApproval = event.toolName === 'write' || event.toolName === 'edit' || event.toolName === 'bash';
-                const isExploreTool = event.toolName === 'explore';
-                const showRunning = event.toolName === 'bash';
-                let runningMessageId: string | undefined;
-
-                if (isExploreTool) {
-                  setExploreAbortController(abortController);
-                  exploreToolsRef.current = [];
-                  const purpose = (event.args.purpose as string) || 'exploring...';
-                  explorePurposeRef.current = purpose;
-                }
-
-                if (!needsApproval) {
-                  runningMessageId = createId();
-                  const { name: toolDisplayName, info: toolInfo } = parseToolHeader(event.toolName, event.args);
-                  const runningContent = toolInfo ? `${toolDisplayName} (${toolInfo})` : toolDisplayName;
-
-                  if (isExploreTool) {
-                    exploreMessageIdRef.current = runningMessageId;
-                  }
-
-                  setMessages((prev: Message[]) => {
-                    const newMessages = [...prev];
-                    newMessages.push({
-                      id: runningMessageId!,
-                      role: "tool",
-                      content: runningContent,
-                      toolName: event.toolName,
-                      toolArgs: event.args,
-                      success: true,
-                      isRunning: showRunning || isExploreTool,
-                      runningStartTime: (showRunning || isExploreTool) ? Date.now() : undefined
-                    });
-                    return newMessages;
-                  });
-                }
-
-                pendingToolCalls.set(event.toolCallId, {
-                  toolName: event.toolName,
-                  args: event.args,
-                  messageId: runningMessageId
-                });
-
-              } else if (event.type === 'tool-result') {
-                const pending = pendingToolCalls.get(event.toolCallId);
-                const toolName = pending?.toolName ?? event.toolName;
-                const toolArgs = pending?.args ?? {};
-                const runningMessageId = pending?.messageId;
-                pendingToolCalls.delete(event.toolCallId);
-
-                if (toolName === 'explore') {
-                  exploreMessageIdRef.current = null;
-                  setExploreAbortController(null);
-                }
-
-                const { content: toolContent, success } = formatToolMessage(
-                  toolName,
-                  toolArgs,
-                  event.result,
-                  { maxLines: DEFAULT_MAX_TOOL_LINES }
-                );
-
-                const toolResultStr = typeof event.result === 'string' ? event.result : JSON.stringify(event.result);
-                totalChars += toolResultStr.length;
-                setCurrentTokens(estimateTokens());
-
-                if (assistantChunk.trim()) {
-                  conversationSteps.push({
-                    type: 'assistant',
-                    content: assistantChunk,
-                    timestamp: Date.now()
-                  });
-                }
-
-                conversationSteps.push({
-                  type: 'tool',
-                  content: toolContent,
-                  toolName,
-                  toolArgs,
-                  toolResult: event.result,
-                  timestamp: Date.now()
-                });
-
-                setMessages((prev: Message[]) => {
-                  const newMessages = [...prev];
-
-                  let runningIndex = -1;
-                  if (runningMessageId) {
-                    runningIndex = newMessages.findIndex(m => m.id === runningMessageId);
-                  } else if (toolName === 'bash' || toolName === 'explore') {
-                    runningIndex = newMessages.findIndex(m => m.toolName === toolName && m.isRunning === true);
-                  }
-
-                  if (runningIndex !== -1) {
-                    newMessages[runningIndex] = {
-                      ...newMessages[runningIndex]!,
-                      content: toolContent,
-                      toolArgs: toolArgs,
-                      toolResult: event.result,
-                      success,
-                      isRunning: false,
-                      runningStartTime: undefined,
-                      timestamp: Date.now()
-                    };
-                    return newMessages;
-                  }
-
-                  newMessages.push({
-                    id: createId(),
-                    role: "tool",
-                    content: toolContent,
-                    toolName,
-                    toolArgs: toolArgs,
-                    toolResult: event.result,
-                    success: success,
-                    timestamp: Date.now()
-                  });
-                  return newMessages;
-                });
-
-                assistantChunk = '';
-                assistantMessageId = null;
-              } else if (event.type === 'error') {
-                if (abortController.signal.aborted) {
-                  notifyAbort();
-                  streamHadError = true;
-                  break;
-                }
-                if (assistantChunk.trim()) {
-                  conversationSteps.push({
-                    type: 'assistant',
-                    content: assistantChunk,
-                    timestamp: Date.now()
-                  });
-                }
-
-                const errorContent = formatErrorMessage('API', event.error);
-                conversationSteps.push({
-                  type: 'assistant',
-                  content: errorContent,
-                  timestamp: Date.now()
-                });
-
-                setMessages((prev: Message[]) => {
-                  const newMessages = [...prev];
-                  newMessages.push({
-                    id: createId(),
-                    role: 'assistant',
-                    content: errorContent,
-                    isError: true,
-                  });
-                  return newMessages;
-                });
-
-                assistantChunk = '';
-                assistantMessageId = null;
-                streamHadError = true;
-                break;
-              } else if (event.type === 'finish') {
-                if (event.usage && event.usage.totalTokens > 0) {
-                  totalTokens = {
-                    prompt: event.usage.promptTokens,
-                    completion: event.usage.completionTokens,
-                    total: event.usage.totalTokens
-                  };
-                  setCurrentTokens(event.usage.totalTokens);
-                }
-              }
-            }
-
-            if (abortController.signal.aborted) {
-              notifyAbort();
-              return;
-            }
-
+            let assistantChunk = '';
+            let thinkingChunk = '';
+            const pendingToolCalls = new Map<string, { toolName: string; args: Record<string, unknown>; messageId?: string }>();
+            let assistantMessageId: string | null = null;
+            let streamHadError = false;
+            titleExtractedRef.current = false;
+
+            for await (const event of agent.streamMessages(conversationHistory, { abortSignal: abortController.signal })) {
+              if (event.type === 'reasoning-delta') {
+                thinkingChunk += event.content;
+                totalChars += event.content.length;
+                setCurrentTokens(estimateTokens());
+
+                if (assistantMessageId === null) {
+                  assistantMessageId = createId();
+                }
+
+                const currentMessageId = assistantMessageId;
+                setMessages((prev: Message[]) => {
+                  const newMessages = [...prev];
+                  const messageIndex = newMessages.findIndex(m => m.id === currentMessageId);
+
+                  if (messageIndex === -1) {
+                    newMessages.push({ id: currentMessageId, role: "assistant", content: '', thinkingContent: thinkingChunk });
+                  } else {
+                    newMessages[messageIndex] = {
+                      ...newMessages[messageIndex]!,
+                      thinkingContent: thinkingChunk
+                    };
+                  }
+                  return newMessages;
+                });
+              } else if (event.type === 'text-delta') {
+                assistantChunk += event.content;
+                totalChars += event.content.length;
+                setCurrentTokens(estimateTokens());
+
+                const { title, cleanContent, isPending, noTitle } = extractTitle(assistantChunk, titleExtractedRef.current);
+
+                if (title) {
+                  titleExtractedRef.current = true;
+                  currentTitleRef.current = title;
+                  setCurrentTitle(title);
+                  setTerminalTitle(title);
+                } else if (noTitle) {
+                  titleExtractedRef.current = true;
+                }
+
+                if (isPending) continue;
+
+                if (assistantMessageId === null) {
+                  assistantMessageId = createId();
+                }
+
+                const displayContent = cleanContent;
+                const currentMessageId = assistantMessageId;
+                setMessages((prev: Message[]) => {
+                  const newMessages = [...prev];
+                  const messageIndex = newMessages.findIndex(m => m.id === currentMessageId);
+
+                  if (messageIndex === -1) {
+                    newMessages.push({ id: currentMessageId, role: "assistant", content: displayContent, thinkingContent: thinkingChunk });
+                  } else {
+                    newMessages[messageIndex] = {
+                      ...newMessages[messageIndex]!,
+                      content: displayContent
+                    };
+                  }
+                  return newMessages;
+                });
+              } else if (event.type === 'step-start') {
+                stepCount++;
+              } else if (event.type === 'tool-call-end') {
+                totalChars += JSON.stringify(event.args).length;
+                setCurrentTokens(estimateTokens());
+
+                const needsApproval = event.toolName === 'write' || event.toolName === 'edit' || event.toolName === 'bash';
+                const isExploreTool = event.toolName === 'explore';
+                const showRunning = event.toolName === 'bash';
+                let runningMessageId: string | undefined;
+
+                if (isExploreTool) {
+                  setExploreAbortController(abortController);
+                  exploreToolsRef.current = [];
+                  const purpose = (event.args.purpose as string) || 'exploring...';
+                  explorePurposeRef.current = purpose;
+                }
+
+                if (!needsApproval) {
+                  runningMessageId = createId();
+                  const { name: toolDisplayName, info: toolInfo } = parseToolHeader(event.toolName, event.args);
+                  const runningContent = toolInfo ? `${toolDisplayName} (${toolInfo})` : toolDisplayName;
+
+                  if (isExploreTool) {
+                    exploreMessageIdRef.current = runningMessageId;
+                  }
+
+                  setMessages((prev: Message[]) => {
+                    const newMessages = [...prev];
+                    newMessages.push({
+                      id: runningMessageId!,
+                      role: "tool",
+                      content: runningContent,
+                      toolName: event.toolName,
+                      toolArgs: event.args,
+                      success: true,
+                      isRunning: showRunning || isExploreTool,
+                      runningStartTime: (showRunning || isExploreTool) ? Date.now() : undefined
+                    });
+                    return newMessages;
+                  });
+                }
+
+                pendingToolCalls.set(event.toolCallId, {
+                  toolName: event.toolName,
+                  args: event.args,
+                  messageId: runningMessageId
+                });
+
+              } else if (event.type === 'tool-result') {
+                const pending = pendingToolCalls.get(event.toolCallId);
+                const toolName = pending?.toolName ?? event.toolName;
+                const toolArgs = pending?.args ?? {};
+                const runningMessageId = pending?.messageId;
+                pendingToolCalls.delete(event.toolCallId);
+
+                if (toolName === 'explore') {
+                  exploreMessageIdRef.current = null;
+                  setExploreAbortController(null);
+                }
+
+                const { content: toolContent, success } = formatToolMessage(
+                  toolName,
+                  toolArgs,
+                  event.result,
+                  { maxLines: DEFAULT_MAX_TOOL_LINES }
+                );
+
+                const toolResultStr = typeof event.result === 'string' ? event.result : JSON.stringify(event.result);
+                totalChars += toolResultStr.length;
+                setCurrentTokens(estimateTokens());
+
+                if (assistantChunk.trim()) {
+                  conversationSteps.push({
+                    type: 'assistant',
+                    content: assistantChunk,
+                    timestamp: Date.now()
+                  });
+                }
+
+                conversationSteps.push({
+                  type: 'tool',
+                  content: toolContent,
+                  toolName,
+                  toolArgs,
+                  toolResult: event.result,
+                  timestamp: Date.now()
+                });
+
+                setMessages((prev: Message[]) => {
+                  const newMessages = [...prev];
+
+                  let runningIndex = -1;
+                  if (runningMessageId) {
+                    runningIndex = newMessages.findIndex(m => m.id === runningMessageId);
+                  } else if (toolName === 'bash' || toolName === 'explore') {
+                    runningIndex = newMessages.findIndex(m => m.toolName === toolName && m.isRunning === true);
+                  }
+
+                  if (runningIndex !== -1) {
+                    newMessages[runningIndex] = {
+                      ...newMessages[runningIndex]!,
+                      content: toolContent,
+                      toolArgs: toolArgs,
+                      toolResult: event.result,
+                      success,
+                      isRunning: false,
+                      runningStartTime: undefined,
+                      timestamp: Date.now()
+                    };
+                    return newMessages;
+                  }
+
+                  newMessages.push({
+                    id: createId(),
+                    role: "tool",
+                    content: toolContent,
+                    toolName,
+                    toolArgs: toolArgs,
+                    toolResult: event.result,
+                    success: success,
+                    timestamp: Date.now()
+                  });
+                  return newMessages;
+                });
+
+                assistantChunk = '';
+                assistantMessageId = null;
+              } else if (event.type === 'error') {
+                if (abortController.signal.aborted) {
+                  notifyAbort();
+                  streamHadError = true;
+                  break;
+                }
+                if (assistantChunk.trim()) {
+                  conversationSteps.push({
+                    type: 'assistant',
+                    content: assistantChunk,
+                    timestamp: Date.now()
+                  });
+                }
+
+                const errorContent = formatErrorMessage('API', event.error);
+                conversationSteps.push({
+                  type: 'assistant',
+                  content: errorContent,
+                  timestamp: Date.now()
+                });
+
+                setMessages((prev: Message[]) => {
+                  const newMessages = [...prev];
+                  newMessages.push({
+                    id: createId(),
+                    role: 'assistant',
+                    content: errorContent,
+                    isError: true,
+                  });
+                  return newMessages;
+                });
+
+                assistantChunk = '';
+                assistantMessageId = null;
+                streamHadError = true;
+                break;
+              } else if (event.type === 'finish') {
+                if (event.usage && event.usage.totalTokens > 0) {
+                  totalTokens = {
+                    prompt: event.usage.promptTokens,
+                    completion: event.usage.completionTokens,
+                    total: event.usage.totalTokens
+                  };
+                  lastPromptTokensRef.current = event.usage.promptTokens;
+                  setCurrentTokens(event.usage.totalTokens);
+                }
+              }
+            }
+
+            if (abortController.signal.aborted) {
+              notifyAbort();
+              return;
+            }
+
             if (!streamHadError && assistantChunk.trim()) {
               conversationSteps.push({
                 type: 'assistant',
@@ -735,35 +900,35 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
             saveConversation(conversationData);
 
           } catch (error) {
-            if (abortController.signal.aborted) {
-              notifyAbort();
-              return;
-            }
-            const errorMessage = error instanceof Error ? error.message : 'An unknown error occurred';
-            const errorContent = formatErrorMessage('Mosaic', errorMessage);
-            setMessages((prev: Message[]) => {
-              const newMessages = [...prev];
-              if (newMessages[newMessages.length - 1]?.role === 'assistant' && newMessages[newMessages.length - 1]?.content === '') {
-                newMessages[newMessages.length - 1] = {
-                  id: newMessages[newMessages.length - 1]!.id,
-                  role: "assistant",
-                  content: errorContent,
-                  isError: true
-                };
-              } else {
-                newMessages.push({
-                  id: createId(),
-                  role: "assistant",
-                  content: errorContent,
-                  isError: true
-                });
-              }
-              return newMessages;
-            });
-          } finally {
-            if (abortControllerRef.current === abortController) {
-              abortControllerRef.current = null;
-            }
+            if (abortController.signal.aborted) {
+              notifyAbort();
+              return;
+            }
+            const errorMessage = error instanceof Error ? error.message : 'An unknown error occurred';
+            const errorContent = formatErrorMessage('Mosaic', errorMessage);
+            setMessages((prev: Message[]) => {
+              const newMessages = [...prev];
+              if (newMessages[newMessages.length - 1]?.role === 'assistant' && newMessages[newMessages.length - 1]?.content === '') {
+                newMessages[newMessages.length - 1] = {
+                  id: newMessages[newMessages.length - 1]!.id,
+                  role: "assistant",
+                  content: errorContent,
+                  isError: true
+                };
+              } else {
+                newMessages.push({
+                  id: createId(),
+                  role: "assistant",
+                  content: errorContent,
+                  isError: true
+                });
+              }
+              return newMessages;
+            });
+          } finally {
+            if (abortControllerRef.current === abortController) {
+              abortControllerRef.current = null;
+            }
             const duration = responseDuration ?? (Date.now() - localStartTime);
             if (duration >= 60000) {
               const blendWord = responseBlendWord ?? BLEND_WORDS[Math.floor(Math.random() * BLEND_WORDS.length)];
@@ -772,44 +937,42 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
                 for (let i = newMessages.length - 1; i >= 0; i--) {
                   if (newMessages[i]?.role === 'assistant') {
                     newMessages[i] = { ...newMessages[i]!, responseDuration: duration, blendWord };
-                    break;
-                  }
-                }
-                return newMessages;
-              });
-            }
-            setIsProcessing(false);
-            setProcessingStartTime(null);
-          }
-
-          return;
-        }
-
-        const commandMessage: Message = {
-          id: createId(),
-          role: "slash",
-          content: result.content,
-          isError: !result.success
-        };
-
-        setMessages((prev: Message[]) => [...prev, commandMessage]);
-
-        if (result.shouldAddToHistory !== false) {
-          addInputToHistory(value.trim());
-        }
-
-        return;
-      }
-    }
-
-    const composedContent = hasPastedContent
-      ? `${meta!.pastedContent!}${value.trim() ? `\n\n${value}` : ''}`
-      : value;
-
+                    break;
+                  }
+                }
+                return newMessages;
+              });
+            }
+            setIsProcessing(false);
+            setProcessingStartTime(null);
+          }
+
+          return;
+        }
+
+        const commandMessage: Message = {
+          id: createId(),
+          role: "slash",
+          content: result.content,
+          isError: !result.success
+        };
+
+        setMessages((prev: Message[]) => [...prev, commandMessage]);
+
+        if (result.shouldAddToHistory !== false) {
+          addInputToHistory(value.trim());
+        }
+
+        return;
+      }
+    }
+
+    const composedContent = hasPastedContent
+      ? `${meta!.pastedContent!}${value.trim() ? `\n\n${value}` : ''}`
+      : value;
+
     addInputToHistory(value.trim() || (hasPastedContent ? '[Pasted text]' : (hasImages ? '[Image]' : value)));
-
-    saveState(messages);
-
+
     const imagesForMessage = imagesSupported ? pendingImages : [];
 
     const userMessage: Message = {
@@ -824,76 +987,89 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
       setPendingImages([]);
     }
 
-    setMessages((prev: Message[]) => [...prev, userMessage]);
-    setIsProcessing(true);
-    const localStartTime = Date.now();
-    setProcessingStartTime(localStartTime);
-    setCurrentTokens(0);
-    shouldAutoScroll.current = true;
-
-    const conversationId = createId();
-    const conversationSteps: ConversationStep[] = [];
-    let totalTokens = { prompt: 0, completion: 0, total: 0 };
-    let stepCount = 0;
-    let totalChars = 0;
-    for (const m of messages) {
-      if (m.role === 'assistant') {
-        totalChars += m.content.length;
-        if (m.thinkingContent) totalChars += m.thinkingContent.length;
-      } else if (m.role === 'tool') {
-        totalChars += m.content.length;
-      }
-    }
-
-    const estimateTokens = () => Math.ceil(totalChars / 4);
-    setCurrentTokens(estimateTokens());
-    const config = readConfig();
-    const abortController = new AbortController();
-    abortControllerRef.current = abortController;
-    let abortNotified = false;
-    const notifyAbort = () => {
-      if (abortNotified) return;
-      abortNotified = true;
-      setMessages((prev: Message[]) => {
-        const newMessages = [...prev];
-        newMessages.push({
-          id: createId(),
-          role: "tool",
-          success: false,
-          content: "Generation aborted. \n↪ What should Mosaic do instead?"
-        });
-        return newMessages;
-      });
-    };
-
+    setMessages((prev: Message[]) => [...prev, userMessage]);
+    setIsProcessing(true);
+    const localStartTime = Date.now();
+    setProcessingStartTime(localStartTime);
+    setCurrentTokens(0);
+    lastPromptTokensRef.current = 0;
+    shouldAutoScroll.current = true;
+
+    const conversationId = createId();
+    const conversationSteps: ConversationStep[] = [];
+    let totalTokens = { prompt: 0, completion: 0, total: 0 };
+    let stepCount = 0;
+    let totalChars = 0;
+    for (const m of messages) {
+      if (m.role === 'assistant') {
+        totalChars += m.content.length;
+        if (m.thinkingContent) totalChars += m.thinkingContent.length;
+      } else if (m.role === 'tool') {
+        totalChars += m.content.length;
+      }
+    }
+
+    const estimateTokens = () => Math.ceil(totalChars / 4);
+    setCurrentTokens(estimateTokens());
+    const config = readConfig();
+    const resolveMaxContextTokens = async () => {
+      if (config.maxContextTokens) return config.maxContextTokens;
+      if (config.provider && config.model) {
+        const resolved = await getModelsDevContextLimit(config.provider, config.model);
+        if (typeof resolved === "number") return resolved;
+      }
+      return undefined;
+    };
+    const buildSystemPrompt = () => {
+      const rawSystemPrompt = config.systemPrompt || DEFAULT_SYSTEM_PROMPT;
+      return processSystemPrompt(rawSystemPrompt, true);
+    };
+    const abortController = new AbortController();
+    abortControllerRef.current = abortController;
+    let abortNotified = false;
+    const notifyAbort = () => {
+      if (abortNotified) return;
+      abortNotified = true;
+      setMessages((prev: Message[]) => {
+        const newMessages = [...prev];
+        newMessages.push({
+          id: createId(),
+          role: "tool",
+          success: false,
+          content: "Generation aborted. \n↪ What should Mosaic do instead?"
+        });
+        return newMessages;
+      });
+    };
+
     conversationSteps.push({
       type: 'user',
       content: composedContent,
       timestamp: Date.now(),
       images: imagesForMessage.length > 0 ? imagesForMessage : undefined
     });
-
+
           let responseDuration: number | null = null;
-          let responseBlendWord: string | null = null;
+          let responseBlendWord: string | undefined = undefined;
 
           try {
             const providerStatus = await Agent.ensureProviderReady();
             if (!providerStatus.ready) {
-        setMessages((prev: Message[]) => {
-          const newMessages = [...prev];
-          newMessages.push({
-            id: createId(),
-            role: "assistant",
-            content: `Ollama error: ${providerStatus.error || 'Could not start Ollama. Make sure Ollama is installed.'}`,
-            isError: true
-          });
-          return newMessages;
-        });
-        setIsProcessing(false);
-        return;
-      }
-
-      const agent = new Agent();
+        setMessages((prev: Message[]) => {
+          const newMessages = [...prev];
+          newMessages.push({
+            id: createId(),
+            role: "assistant",
+            content: `Ollama error: ${providerStatus.error || 'Could not start Ollama. Make sure Ollama is installed.'}`,
+            isError: true
+          });
+          return newMessages;
+        });
+        setIsProcessing(false);
+        return;
+      }
+
+      const agent = new Agent();
       const conversationHistory = buildConversationHistory([...messages, userMessage], imagesSupported);
       let assistantChunk = '';
       let thinkingChunk = '';
@@ -931,25 +1107,25 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
           assistantChunk += event.content;
           totalChars += event.content.length;
           setCurrentTokens(estimateTokens());
-
-          const { title, cleanContent, isPending, noTitle } = extractTitle(assistantChunk, titleExtractedRef.current);
-
-          if (title) {
-            titleExtractedRef.current = true;
-            currentTitleRef.current = title;
-            setCurrentTitle(title);
-            setTerminalTitle(title);
-          } else if (noTitle) {
-            titleExtractedRef.current = true;
-          }
-
-          if (isPending) continue;
-
-          if (assistantMessageId === null) {
-            assistantMessageId = createId();
-          }
-
-          const displayContent = cleanContent;
+
+          const { title, cleanContent, isPending, noTitle } = extractTitle(assistantChunk, titleExtractedRef.current);
+
+          if (title) {
+            titleExtractedRef.current = true;
+            currentTitleRef.current = title;
+            setCurrentTitle(title);
+            setTerminalTitle(title);
+          } else if (noTitle) {
+            titleExtractedRef.current = true;
+          }
+
+          if (isPending) continue;
+
+          if (assistantMessageId === null) {
+            assistantMessageId = createId();
+          }
+
+          const displayContent = cleanContent;
           const currentMessageId = assistantMessageId;
           setMessages((prev: Message[]) => {
             const newMessages = [...prev];
@@ -966,122 +1142,122 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
             }
             return newMessages;
           });
-        } else if (event.type === 'step-start') {
-          stepCount++;
-        } else if (event.type === 'tool-call-end') {
-          totalChars += JSON.stringify(event.args).length;
-          setCurrentTokens(estimateTokens());
-
-          const isExploreTool = event.toolName === 'explore';
-          let runningMessageId: string | undefined;
-
-          if (isExploreTool) {
-            setExploreAbortController(abortController);
-            exploreToolsRef.current = [];
-            const purpose = (event.args.purpose as string) || 'exploring...';
-            explorePurposeRef.current = purpose;
-            runningMessageId = createId();
-            exploreMessageIdRef.current = runningMessageId;
-            const { name: toolDisplayName, info: toolInfo } = parseToolHeader(event.toolName, event.args);
-            const runningContent = toolInfo ? `${toolDisplayName} (${toolInfo})` : toolDisplayName;
-
-            setMessages((prev: Message[]) => {
-              const newMessages = [...prev];
-              newMessages.push({
-                id: runningMessageId!,
-                role: "tool",
-                content: runningContent,
-                toolName: event.toolName,
-                toolArgs: event.args,
-                success: true,
-                isRunning: true,
-                runningStartTime: Date.now()
-              });
-              return newMessages;
-            });
-          }
-
-          pendingToolCalls.set(event.toolCallId, {
-            toolName: event.toolName,
-            args: event.args,
-            messageId: runningMessageId
-          });
-
-        } else if (event.type === 'tool-result') {
-          const pending = pendingToolCalls.get(event.toolCallId);
-          const toolName = pending?.toolName ?? event.toolName;
-          const toolArgs = pending?.args ?? {};
-          const runningMessageId = pending?.messageId;
-          pendingToolCalls.delete(event.toolCallId);
-
-          if (toolName === 'explore') {
-            exploreMessageIdRef.current = null;
-            setExploreAbortController(null);
-          }
-
-          const { content: toolContent, success } = formatToolMessage(
-            toolName,
-            toolArgs,
-            event.result,
-            { maxLines: DEFAULT_MAX_TOOL_LINES }
-          );
-
-          const toolResultStr = typeof event.result === 'string' ? event.result : JSON.stringify(event.result);
-          totalChars += toolResultStr.length;
-          setCurrentTokens(estimateTokens());
-
-          if (assistantChunk.trim()) {
-            conversationSteps.push({
-              type: 'assistant',
-              content: assistantChunk,
-              timestamp: Date.now()
-            });
-          }
-
-          conversationSteps.push({
-            type: 'tool',
-            content: toolContent,
-            toolName,
-            toolArgs,
-            toolResult: event.result,
-            timestamp: Date.now()
-          });
-
-          setMessages((prev: Message[]) => {
-            const newMessages = [...prev];
-
-            let runningIndex = -1;
-            if (runningMessageId) {
-              runningIndex = newMessages.findIndex(m => m.id === runningMessageId);
-            } else if (toolName === 'bash' || toolName === 'explore') {
-              runningIndex = newMessages.findIndex(m => m.isRunning && m.toolName === toolName);
-            }
-
-            if (runningIndex !== -1) {
-              newMessages[runningIndex] = {
-                ...newMessages[runningIndex]!,
-                content: toolContent,
-                toolArgs: toolArgs,
-                toolResult: event.result,
-                success,
-                isRunning: false,
-                runningStartTime: undefined
-              };
-              return newMessages;
-            }
-
-            newMessages.push({
-              id: createId(),
-              role: "tool",
-              content: toolContent,
-              toolName,
-              toolArgs: toolArgs,
-              toolResult: event.result,
-              success: success
-            });
-            return newMessages;
-          });
-
+        } else if (event.type === 'step-start') {
+          stepCount++;
+        } else if (event.type === 'tool-call-end') {
+          totalChars += JSON.stringify(event.args).length;
+          setCurrentTokens(estimateTokens());
+
+          const isExploreTool = event.toolName === 'explore';
+          let runningMessageId: string | undefined;
+
+          if (isExploreTool) {
+            setExploreAbortController(abortController);
+            exploreToolsRef.current = [];
+            const purpose = (event.args.purpose as string) || 'exploring...';
+            explorePurposeRef.current = purpose;
+            runningMessageId = createId();
+            exploreMessageIdRef.current = runningMessageId;
+            const { name: toolDisplayName, info: toolInfo } = parseToolHeader(event.toolName, event.args);
+            const runningContent = toolInfo ? `${toolDisplayName} (${toolInfo})` : toolDisplayName;
+
+            setMessages((prev: Message[]) => {
+              const newMessages = [...prev];
+              newMessages.push({
+                id: runningMessageId!,
+                role: "tool",
+                content: runningContent,
+                toolName: event.toolName,
+                toolArgs: event.args,
+                success: true,
+                isRunning: true,
+                runningStartTime: Date.now()
+              });
+              return newMessages;
+            });
+          }
+
+          pendingToolCalls.set(event.toolCallId, {
+            toolName: event.toolName,
+            args: event.args,
+            messageId: runningMessageId
+          });
+
+        } else if (event.type === 'tool-result') {
+          const pending = pendingToolCalls.get(event.toolCallId);
+          const toolName = pending?.toolName ?? event.toolName;
+          const toolArgs = pending?.args ?? {};
+          const runningMessageId = pending?.messageId;
+          pendingToolCalls.delete(event.toolCallId);
+
+          if (toolName === 'explore') {
+            exploreMessageIdRef.current = null;
+            setExploreAbortController(null);
+          }
+
+          const { content: toolContent, success } = formatToolMessage(
+            toolName,
+            toolArgs,
+            event.result,
+            { maxLines: DEFAULT_MAX_TOOL_LINES }
+          );
+
+          const toolResultStr = typeof event.result === 'string' ? event.result : JSON.stringify(event.result);
+          totalChars += toolResultStr.length;
+          setCurrentTokens(estimateTokens());
+
+          if (assistantChunk.trim()) {
+            conversationSteps.push({
+              type: 'assistant',
+              content: assistantChunk,
+              timestamp: Date.now()
+            });
+          }
+
+          conversationSteps.push({
+            type: 'tool',
+            content: toolContent,
+            toolName,
+            toolArgs,
+            toolResult: event.result,
+            timestamp: Date.now()
+          });
+
+          setMessages((prev: Message[]) => {
+            const newMessages = [...prev];
+
+            let runningIndex = -1;
+            if (runningMessageId) {
+              runningIndex = newMessages.findIndex(m => m.id === runningMessageId);
+            } else if (toolName === 'bash' || toolName === 'explore') {
+              runningIndex = newMessages.findIndex(m => m.isRunning && m.toolName === toolName);
+            }
+
+            if (runningIndex !== -1) {
+              newMessages[runningIndex] = {
+                ...newMessages[runningIndex]!,
+                content: toolContent,
+                toolArgs: toolArgs,
+                toolResult: event.result,
+                success,
+                isRunning: false,
+                runningStartTime: undefined
+              };
+              return newMessages;
+            }
+
+            newMessages.push({
+              id: createId(),
+              role: "tool",
+              content: toolContent,
+              toolName,
+              toolArgs: toolArgs,
+              toolResult: event.result,
+              success: success
+            });
+            return newMessages;
+          });
+
           assistantChunk = '';
           thinkingChunk = '';
           assistantMessageId = null;
@@ -1091,54 +1267,55 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
             streamHadError = true;
             break;
           }
-          if (assistantChunk.trim()) {
-            conversationSteps.push({
-              type: 'assistant',
-              content: assistantChunk,
-              timestamp: Date.now()
-            });
-          }
-
-          const errorContent = formatErrorMessage('API', event.error);
-          conversationSteps.push({
-            type: 'assistant',
-            content: errorContent,
-            timestamp: Date.now()
-          });
-
-          setMessages((prev: Message[]) => {
-            const newMessages = [...prev];
-            newMessages.push({
-              id: createId(),
-              role: 'assistant',
-              content: errorContent,
-              isError: true,
-            });
-            return newMessages;
-          });
-
+          if (assistantChunk.trim()) {
+            conversationSteps.push({
+              type: 'assistant',
+              content: assistantChunk,
+              timestamp: Date.now()
+            });
+          }
+
+          const errorContent = formatErrorMessage('API', event.error);
+          conversationSteps.push({
+            type: 'assistant',
+            content: errorContent,
+            timestamp: Date.now()
+          });
+
+          setMessages((prev: Message[]) => {
+            const newMessages = [...prev];
+            newMessages.push({
+              id: createId(),
+              role: 'assistant',
+              content: errorContent,
+              isError: true,
+            });
+            return newMessages;
+          });
+
           assistantChunk = '';
           thinkingChunk = '';
           assistantMessageId = null;
           streamHadError = true;
           break;
-        } else if (event.type === 'finish') {
-          if (event.usage && event.usage.totalTokens > 0) {
-            totalTokens = {
-              prompt: event.usage.promptTokens,
-              completion: event.usage.completionTokens,
-              total: event.usage.totalTokens
-            };
-            setCurrentTokens(event.usage.totalTokens);
-          }
-        }
-      }
-
-      if (abortController.signal.aborted) {
-        notifyAbort();
-        return;
-      }
-
+        } else if (event.type === 'finish') {
+          if (event.usage && event.usage.totalTokens > 0) {
+            totalTokens = {
+              prompt: event.usage.promptTokens,
+              completion: event.usage.completionTokens,
+              total: event.usage.totalTokens
+            };
+            lastPromptTokensRef.current = event.usage.promptTokens;
+            setCurrentTokens(event.usage.totalTokens);
+          }
+        }
+      }
+
+      if (abortController.signal.aborted) {
+        notifyAbort();
+        return;
+      }
+
           if (!streamHadError && assistantChunk.trim()) {
             conversationSteps.push({
               type: 'assistant',
@@ -1175,37 +1352,52 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
           };
 
           saveConversation(conversationData);
-
-    } catch (error) {
-      if (abortController.signal.aborted) {
-        notifyAbort();
-        return;
-      }
-      const errorMessage = error instanceof Error ? error.message : 'An unknown error occurred';
-      const errorContent = formatErrorMessage('Mosaic', errorMessage);
-      setMessages((prev: Message[]) => {
-        const newMessages = [...prev];
-        if (newMessages[newMessages.length - 1]?.role === 'assistant' && newMessages[newMessages.length - 1]?.content === '') {
-          newMessages[newMessages.length - 1] = {
-            id: newMessages[newMessages.length - 1]!.id,
-            role: "assistant",
-            content: errorContent,
-            isError: true
-          };
-        } else {
-          newMessages.push({
-            id: createId(),
-            role: "assistant",
-            content: errorContent,
-            isError: true
-          });
-        }
-        return newMessages;
-      });
-    } finally {
-      if (abortControllerRef.current === abortController) {
-        abortControllerRef.current = null;
-      }
+          const resolvedMax = await resolveMaxContextTokens();
+          const maxContextTokens = resolvedMax ?? getDefaultContextBudget(config.provider);
+          if (!abortController.signal.aborted) {
+            const realPromptTokens = lastPromptTokensRef.current;
+            const systemPrompt = buildSystemPrompt();
+            setMessages(prev => {
+              const usedTokens = realPromptTokens > 0
+                ? realPromptTokens
+                : estimateTotalTokens(prev, systemPrompt);
+              if (!shouldAutoCompact(usedTokens, maxContextTokens)) return prev;
+              const compacted = compactMessagesForUi(prev, systemPrompt, maxContextTokens, createId, true);
+              setCurrentTokens(compacted.estimatedTokens);
+              return compacted.messages;
+            });
+          }
+
+    } catch (error) {
+      if (abortController.signal.aborted) {
+        notifyAbort();
+        return;
+      }
+      const errorMessage = error instanceof Error ? error.message : 'An unknown error occurred';
+      const errorContent = formatErrorMessage('Mosaic', errorMessage);
+      setMessages((prev: Message[]) => {
+        const newMessages = [...prev];
+        if (newMessages[newMessages.length - 1]?.role === 'assistant' && newMessages[newMessages.length - 1]?.content === '') {
+          newMessages[newMessages.length - 1] = {
+            id: newMessages[newMessages.length - 1]!.id,
+            role: "assistant",
+            content: errorContent,
+            isError: true
+          };
+        } else {
+          newMessages.push({
+            id: createId(),
+            role: "assistant",
+            content: errorContent,
+            isError: true
+          });
+        }
+        return newMessages;
+      });
+    } finally {
+      if (abortControllerRef.current === abortController) {
+        abortControllerRef.current = null;
+      }
       const duration = responseDuration ?? (Date.now() - localStartTime);
       if (duration >= 60000) {
         const blendWord = responseBlendWord ?? BLEND_WORDS[Math.floor(Math.random() * BLEND_WORDS.length)];
@@ -1214,42 +1406,42 @@ export function Main({ pasteRequestId = 0, copyRequestId = 0, onCopy, shortcutsO
           for (let i = newMessages.length - 1; i >= 0; i--) {
             if (newMessages[i]?.role === 'assistant') {
               newMessages[i] = { ...newMessages[i]!, responseDuration: duration, blendWord };
-              break;
-            }
-          }
-          return newMessages;
-        });
-      }
-      setIsProcessing(false);
-      setProcessingStartTime(null);
-    }
-  };
-
-  useEffect(() => {
-    if (initialMessage && !initialMessageProcessed.current && currentPage === "chat") {
-      initialMessageProcessed.current = true;
-      handleSubmit(initialMessage);
-    }
-  }, [initialMessage, currentPage, handleSubmit]);
-
-  if (currentPage === "home") {
-    const handleHomeSubmit = (value: string, meta?: InputSubmitMeta) => {
-      const hasPastedContent = Boolean(meta?.isPaste && meta.pastedContent);
-      if (!value.trim() && !hasPastedContent) return;
-      setCurrentPage("chat");
-      handleSubmit(value, meta);
-    };
-
-    return (
-      <HomePage
-        onSubmit={handleHomeSubmit}
-        pasteRequestId={pasteRequestId}
-        shortcutsOpen={shortcutsOpen}
-      />
-    );
-  }
-
-  return (
+              break;
+            }
+          }
+          return newMessages;
+        });
+      }
+      setIsProcessing(false);
+      setProcessingStartTime(null);
+    }
+  };
+
+  useEffect(() => {
+    if (initialMessage && !initialMessageProcessed.current && currentPage === "chat") {
+      initialMessageProcessed.current = true;
+      handleSubmit(initialMessage);
+    }
+  }, [initialMessage, currentPage, handleSubmit]);
+
+  if (currentPage === "home") {
+    const handleHomeSubmit = (value: string, meta?: InputSubmitMeta) => {
+      const hasPastedContent = Boolean(meta?.isPaste && meta.pastedContent);
+      if (!value.trim() && !hasPastedContent) return;
+      setCurrentPage("chat");
+      handleSubmit(value, meta);
+    };
+
+    return (
+      <HomePage
+        onSubmit={handleHomeSubmit}
+        pasteRequestId={pasteRequestId}
+        shortcutsOpen={shortcutsOpen}
+      />
+    );
+  }
+
+  return (
     <ChatPage
       messages={messages}
       isProcessing={isProcessing}