@widgetic/chat 0.1.4

package/dist/components/ChatMessages.svelte

@@ -0,0 +1,996 @@
+<script lang="ts">
+  import type { ChatMessage, ChatContext } from '../types/index.js';
+  import { MESSAGE_COLORS } from '../constants/colors.js';
+  import { chatLog } from '../utils/logger.js';
+  import { AttachmentDisplay, SuggestionButtons } from './index.js';
+  import CodeBlock from './CodeBlock.svelte';
+  import LoadingIndicator from './LoadingIndicator.svelte';
+  import { onMount, tick } from 'svelte';
+
+  // Props
+  interface Props {
+    className?: className;
+    messages: ChatMessage[];
+    isLoading?: boolean;
+    context: ChatContext;
+    isCompleted?: boolean;
+    /**
+     * When true, heuristic code pattern detection (`detectCodePatterns`) runs
+     * for messages without explicit ``` fences. Default false — keeps chat UI
+     * clean when assistants return summaries that mention file paths or
+     * extensions (e.g. "Updated 4 files: ...svelte") which would otherwise be
+     * misclassified as JavaScript code blocks. Enable only for debugging.
+     */
+    enableAutoCodeDetection?: boolean;
+    /**
+     * When false, backend-generated "Updated N files: ..." summaries are
+     * hidden from the UI (they still exist in the DB as LLM context). Default
+     * false — noisy for end users. Typically bound to `debugMode` by hosts.
+     */
+    showGenerationSummaries?: boolean;
+    onSuggestionSelected?: (suggestion: string) => void;
+    onRestoreCheckpoint?: (messageId: string, messageIndex: number) => void;
+    onRetryMessage?: (content: string) => void;
+  }
+
+  let {
+    class: className = '',
+    messages,
+    isLoading = false,
+    context,
+    isCompleted = false,
+    enableAutoCodeDetection = false,
+    showGenerationSummaries = false,
+    onSuggestionSelected,
+    onRestoreCheckpoint,
+    onRetryMessage,
+  }: Props = $props();
+
+  // Matches "Updated N file(s)[...]: ..." assistant summaries emitted by the
+  // widget codegen backend. Anchored to first line only — regular assistant
+  // chat replies that happen to contain this phrase mid-message are kept.
+  const GENERATION_SUMMARY_REGEX = /^Updated \d+ files?(?:\s*\([^)]*\))?:/;
+
+  function isGenerationSummary(message: ChatMessage): boolean {
+    if (message.messageType !== 'assistant' || !message.messageContent) return false;
+    const trimmed = message.messageContent.trim();
+    if (/^Code generation failed/i.test(trimmed)) return false;
+    if (/^Code generation completed/i.test(trimmed)) return false;
+    if (/^Preview recompiled/i.test(trimmed)) return false;
+    return GENERATION_SUMMARY_REGEX.test(trimmed);
+  }
+
+  function isPreviewRecompiledMessage(message: ChatMessage): boolean {
+    return message.messageType === 'assistant'
+      && !!message.messageContent
+      && /^Preview recompiled/i.test(message.messageContent.trim());
+  }
+
+  function extractCommitFromContent(content: string): string | null {
+    const match = content.match(/(?:Commit:?\s+|\n\nCommit\s+|\(commit\s+)([a-f0-9]{7,40})/i);
+    return match?.[1]?.toLowerCase() ?? null;
+  }
+
+  function isBackendCodegenSummary(content: string): boolean {
+    const trimmed = content.trim();
+    return /^Updated \d+ files?/i.test(trimmed) && /\n\nCommit\s+[a-f0-9]{7,40}/i.test(trimmed);
+  }
+
+  /** Codegen completion with commit — not preview-rebuild status lines. */
+  function messageHasCommit(message: ChatMessage): boolean {
+    if (!message.messageContent || isPreviewRecompiledMessage(message)) return false;
+    const trimmed = message.messageContent.trim();
+    if (isCodegenFailure(message)) return false;
+    if (/^Code generation completed/i.test(trimmed) && /Commit:?\s+[a-f0-9]{7,40}/i.test(trimmed)) {
+      return true;
+    }
+    return isBackendCodegenSummary(trimmed);
+  }
+
+  function turnHasCodegenFailure(turn: ChatTurn): boolean {
+    return turn.replies.some(isCodegenFailure);
+  }
+
+  function rebalancePreviewRecompiledAcrossTurns(turns: ChatTurn[]): ChatTurn[] {
+    const result = turns.map((t) => ({ user: t.user, replies: [...t.replies] }));
+
+    for (let i = 0; i < result.length; i++) {
+      const turn = result[i];
+      const turnCodegenCommits = new Set(
+        turn.replies
+          .filter((r) => messageHasCommit(r))
+          .map((r) => extractCommitFromContent(r.messageContent || '')!)
+          .filter(Boolean),
+      );
+
+      const kept: ChatMessage[] = [];
+      const moved: ChatMessage[] = [];
+
+      for (const reply of turn.replies) {
+        if (!isPreviewRecompiledMessage(reply)) {
+          kept.push(reply);
+          continue;
+        }
+        const previewCommit = extractCommitFromContent(reply.messageContent || '');
+        const belongsHere =
+          !turnHasCodegenFailure(turn)
+          && (!previewCommit || turnCodegenCommits.size === 0 || turnCodegenCommits.has(previewCommit));
+        if (belongsHere) {
+          kept.push(reply);
+        } else {
+          moved.push(reply);
+        }
+      }
+
+      turn.replies = kept;
+
+      for (const msg of moved) {
+        const commit = extractCommitFromContent(msg.messageContent || '');
+        let placed = false;
+        for (let j = i - 1; j >= 0; j--) {
+          const prev = result[j];
+          const prevHasCommit = prev.replies.some(
+            (r) => commit && extractCommitFromContent(r.messageContent || '') === commit,
+          );
+          if (prevHasCommit || (j === i - 1 && !turnHasCodegenFailure(prev))) {
+            prev.replies.push(msg);
+            placed = true;
+            break;
+          }
+        }
+        if (!placed && i > 0) {
+          result[i - 1].replies.push(msg);
+        }
+      }
+    }
+
+    return result;
+  }
+
+  function turnHasWidgetHead(turn: ChatTurn): boolean {
+    return turn.replies.some((r) => r.isWidgetHead && messageHasCommit(r));
+  }
+
+  function isCodegenFailure(message: ChatMessage): boolean {
+    return message.messageType === 'assistant'
+      && !!message.messageContent
+      && /^Code generation failed/i.test(message.messageContent.trim());
+  }
+
+  function canRestoreToMessage(message: ChatMessage): boolean {
+    if (message.isWidgetHead) return false;
+    if (!message.messageContent) return false;
+    const trimmed = message.messageContent.trim();
+    if (/^Preview recompiled/i.test(trimmed)) return false;
+    return message.messageType === 'assistant'
+      && messageHasCommit(message)
+      && !isCodegenFailure(message);
+  }
+
+  function canUndoHeadUserTurn(turn: ChatTurn): boolean {
+    return !!turn.user && !!getTurnCommitReply(turn)?.isWidgetHead;
+  }
+
+  function extractCommitShort(content: string): string | null {
+    const match = content.match(/(?:Commit:?\s+|\n\nCommit\s+)([a-f0-9]{7,40})/i);
+    return match?.[1]?.slice(0, 8) ?? null;
+  }
+
+  /** Assistant reply that carries the widget commit for this turn (if any). */
+  function getTurnCommitReply(turn: ChatTurn): ChatMessage | null {
+    for (const reply of turn.replies) {
+      if (messageHasCommit(reply) && !isCodegenFailure(reply)) return reply;
+    }
+    return null;
+  }
+
+  /** Whether this assistant bubble should render (vs commit chrome on user message). */
+  function shouldShowAssistantReply(reply: ChatMessage): boolean {
+    if (isCodegenFailure(reply)) return true;
+    if (showGenerationSummaries) return true;
+    if (/^Code generation completed/i.test((reply.messageContent || '').trim())) return true;
+    if (/^Preview recompiled/i.test((reply.messageContent || '').trim())) return true;
+    return !isGenerationSummary(reply);
+  }
+
+  /** Show commit chrome + Restore Checkpoint on the user prompt whenever the turn has a commit.
+   *  Previously this was gated to Debug OFF (assistant reply hidden), but we want the buttons
+   *  available in Debug ON too — the assistant summary and the user-side chrome coexist. */
+  function showCommitMetaOnUserTurn(turn: ChatTurn): boolean {
+    return !!getTurnCommitReply(turn);
+  }
+
+  type ChatTurn = {
+    user: ChatMessage | null;
+    replies: ChatMessage[];
+  };
+
+  let chatTurns = $state<ChatTurn[]>([]);
+
+  $effect(() => {
+    const turns: ChatTurn[] = [];
+    let current: ChatTurn | null = null;
+
+    for (const message of messages) {
+      if (message.messageType === 'user') {
+        if (current) turns.push(current);
+        current = { user: message, replies: [] };
+      } else if (current) {
+        current.replies.push(message);
+      } else {
+        turns.push({ user: null, replies: [message] });
+      }
+    }
+    if (current) turns.push(current);
+    chatTurns = rebalancePreviewRecompiledAcrossTurns(turns);
+  });
+
+  function messageImpliesAttachedImage(content: string): boolean {
+    return /imaginea ata[sș]at[ăa]|attached (reference )?image|reference image|din imagine/i.test(content);
+  }
+
+  let visibleMessages = $derived(
+    showGenerationSummaries
+      ? messages
+      : messages.filter((m) => !isGenerationSummary(m))
+  );
+
+  // Local state
+  let messagesContainer: HTMLElement;
+  let showSuggestionsInline = $state(false);
+  let autoScroll = $state(true);
+
+  // Auto-scroll to bottom when new messages arrive or on initial load.
+  // Also scrolls on mount/conversation-switch when messages are already present.
+  $effect(() => {
+    if (messagesContainer && autoScroll && visibleMessages.length > 0) {
+      tick().then(() => {
+        messagesContainer?.scrollTo({
+          top: messagesContainer.scrollHeight,
+          behavior: 'smooth'
+        });
+      });
+    }
+  });
+
+  // Force scroll to bottom on initial mount (messages may already be loaded)
+  onMount(() => {
+    tick().then(() => {
+      if (messagesContainer && visibleMessages.length > 0) {
+        messagesContainer.scrollTo({ top: messagesContainer.scrollHeight, behavior: 'instant' });
+      }
+    });
+  });
+
+  function handleScroll() {
+    if (!messagesContainer) return;
+    const { scrollTop, scrollHeight, clientHeight } = messagesContainer;
+    autoScroll = scrollTop + clientHeight >= scrollHeight - 10;
+  }
+
+  function formatTime(date: Date): string {
+    return new Intl.DateTimeFormat('en-US', {
+      hour: '2-digit',
+      minute: '2-digit'
+    }).format(date);
+  }
+
+  function handleSuggestionSelected(event: CustomEvent) {
+    const suggestion = event.detail.suggestion;
+    chatLog('Suggestion selected in ChatMessages:', suggestion);
+    if (onSuggestionSelected) {
+      onSuggestionSelected(suggestion);
+    }
+  }
+
+  function handleRestoreCheckpoint(messageId: string) {
+    // Find the index of the message to restore to
+    const messageIndex = messages.findIndex(msg => msg.id === messageId);
+    if (messageIndex === -1) return;
+
+    chatLog('Restore checkpoint requested for message:', messageId, 'at index:', messageIndex);
+    
+    // Call the prop callback if provided
+    if (onRestoreCheckpoint) {
+      onRestoreCheckpoint(messageId, messageIndex);
+    }
+  }
+
+  function handleRetryTurn(turn: ChatTurn) {
+    const retryContent = turn.user?.messageContent?.trim();
+    if (!retryContent) return;
+    chatLog('Retry requested for failed codegen turn:', retryContent.slice(0, 80));
+    onRetryMessage?.(retryContent);
+  }
+
+  // Auto-detect programming language based on content
+  function detectLanguage(content: string): string {
+    const trimmed = content.trim().toLowerCase();
+    
+    // HTML detection
+    if (/<(!doctype|html|head|body|div|span|p|h[1-6]|img|a|ul|ol|li|table|tr|td|th|form|input|button|script|style)/i.test(content)) {
+      return 'html';
+    }
+    
+    // CSS detection
+    if (/[.#][\w-]+\s*\{[^}]*\}|@(media|import|font-face|keyframes)|[\w-]+\s*:\s*[^;]+;/.test(content)) {
+      return 'css';
+    }
+    
+    // JSON detection
+    if (/^\s*[\{\[]/.test(trimmed) && /[\}\]]\s*$/.test(trimmed)) {
+      try {
+        JSON.parse(content);
+        return 'json';
+      } catch (e) {
+        // Not valid JSON, continue checking
+      }
+    }
+    
+    // SQL detection
+    if (/\b(select|insert|update|delete|create|drop|alter|from|where|join|group by|order by|having)\b/i.test(content)) {
+      return 'sql';
+    }
+    
+    // JavaScript/TypeScript detection
+    if (/\b(function|const|let|var|class|import|export|if|else|for|while|return|async|await|=>|console\.log)\b/.test(content) ||
+        /\.(js|ts|jsx|tsx|vue|svelte)/.test(content) ||
+        /[\{\}]\s*$/.test(content) ||
+        /;\s*$/.test(content.split('\n').slice(-1)[0] || '')) {
+      return 'javascript';
+    }
+    
+    // Python detection
+    if (/\b(def|class|import|from|if __name__|print|return)\b/.test(content) ||
+        /\.py\b/.test(content) ||
+        /:\s*$/.test(content.split('\n')[0] || '')) {
+      return 'python';
+    }
+    
+    // XML detection
+    if (/<\?xml|<\/[\w:]+>/.test(content)) {
+      return 'xml';
+    }
+    
+    return 'plaintext';
+  }
+
+  // Enhanced code detection patterns
+  function detectCodePatterns(content: string): { isCode: boolean; confidence: number; language: string } {
+    const lines = content.split('\n');
+    let codeScore = 0;
+    let totalChecks = 0;
+    
+    // Check for code-like patterns
+    const patterns = [
+      // Function definitions
+      { regex: /\b(function|const|let|var|def|class)\s+\w+/, weight: 3 },
+      // Brackets and braces
+      { regex: /[{\[\(][^}\]\)]*[}\]\)]/, weight: 1 },
+      // Semicolons at end of lines
+      { regex: /;\s*$/, weight: 2 },
+      // Assignment operators
+      { regex: /\w+\s*[=:]\s*/, weight: 1 },
+      // Common keywords
+      { regex: /\b(if|else|for|while|return|import|export|from|class|function)\b/, weight: 2 },
+      // HTML tags
+      { regex: /<\/?[\w:]+[^>]*>/, weight: 3 },
+      // CSS properties
+      { regex: /[\w-]+\s*:\s*[^;]+;/, weight: 3 },
+      // Console/print statements
+      { regex: /\b(console\.log|print|echo)\s*\(/, weight: 2 },
+      // Method calls
+      { regex: /\w+\.\w+\s*\(/, weight: 1 },
+      // Array/object literals
+      { regex: /[\[\{].*[\]\}]/, weight: 1 },
+      // Comments
+      { regex: /^\s*(\/\/|#|<!--|\/\*|\*)/, weight: 1 },
+      // Indentation patterns (2+ spaces or tabs)
+      { regex: /^(\s{2,}|\t+)\S/, weight: 1 }
+    ];
+    
+    // Analyze each line
+    for (const line of lines) {
+      for (const pattern of patterns) {
+        totalChecks++;
+        if (pattern.regex.test(line)) {
+          codeScore += pattern.weight;
+        }
+      }
+    }
+    
+    // Additional checks for multi-line patterns
+    const fullContent = content.trim();
+    
+    // File extension mentions
+    if (/\.(js|ts|jsx|tsx|py|html|css|json|sql|xml|vue|svelte|php|java|cpp|c|go|rs|rb)\b/i.test(fullContent)) {
+      codeScore += 3;
+      totalChecks += 3;
+    }
+    
+    // Code block structure (multiple lines with consistent indentation)
+    if (lines.length > 2) {
+      const indentedLines = lines.filter(line => /^\s{2,}/.test(line));
+      if (indentedLines.length > lines.length * 0.3) {
+        codeScore += 2;
+        totalChecks += 2;
+      }
+    }
+    
+    // JSON-like structure
+    if (/^\s*[\{\[]/.test(fullContent) && /[\}\]]\s*$/.test(fullContent)) {
+      codeScore += 3;
+      totalChecks += 3;
+    }
+    
+    const confidence = totalChecks > 0 ? (codeScore / totalChecks) : 0;
+    const language = detectLanguage(fullContent);
+    
+    // Consider it code if confidence is high enough
+    const isCode = confidence > 0.3 || codeScore >= 3;
+
+    return { isCode, confidence, language };
+  }
+
+  // Code block detection and parsing
+  function parseMessageContent(content: string) {
+    const parts = [];
+    
+    // First, try to match explicit code blocks (``` syntax)
+    const completeCodeBlockRegex = /```(\w+)?\s*([\s\S]*?)```/g;
+    let lastIndex = 0;
+    let match;
+    let foundExplicitBlocks = false;
+
+    while ((match = completeCodeBlockRegex.exec(content)) !== null) {
+      foundExplicitBlocks = true;
+      
+      // Add text before code block
+      if (match.index > lastIndex) {
+        const textContent = content.slice(lastIndex, match.index).trim();
+        if (textContent) {
+          parts.push({
+            type: 'text',
+            content: textContent
+          });
+        }
+      }
+
+      // Add explicit code block
+      parts.push({
+        type: 'code',
+        language: match[1] || 'plaintext',
+        content: match[2].trim()
+      });
+
+      lastIndex = match.index + match[0].length;
+    }
+
+    // If no explicit code blocks found, check for incomplete ones
+    if (!foundExplicitBlocks) {
+      const incompleteCodeBlockMatch = content.match(/```(\w+)?\s*([\s\S]*)$/);
+      if (incompleteCodeBlockMatch) {
+        const beforeCodeBlock = content.slice(0, incompleteCodeBlockMatch.index).trim();
+        if (beforeCodeBlock) {
+          parts.push({
+            type: 'text',
+            content: beforeCodeBlock
+          });
+        }
+        
+        parts.push({
+          type: 'code',
+          language: incompleteCodeBlockMatch[1] || 'plaintext',
+          content: incompleteCodeBlockMatch[2].trim()
+        });
+        
+        return parts;
+      }
+    }
+
+    // Add remaining text after explicit code blocks
+    if (lastIndex < content.length) {
+      const remainingContent = content.slice(lastIndex).trim();
+      if (remainingContent) {
+        // If we found explicit blocks, just add remaining as text
+        if (foundExplicitBlocks) {
+          parts.push({
+            type: 'text',
+            content: remainingContent
+          });
+        } else if (enableAutoCodeDetection) {
+          // No explicit blocks found — fall back to auto-detection only when debug flag is on
+          const detection = detectCodePatterns(remainingContent);
+          if (detection.isCode) {
+            parts.push({
+              type: 'code',
+              language: detection.language,
+              content: remainingContent
+            });
+          } else {
+            parts.push({
+              type: 'text',
+              content: remainingContent
+            });
+          }
+        } else {
+          parts.push({
+            type: 'text',
+            content: remainingContent
+          });
+        }
+      }
+    }
+
+    // If no explicit blocks found and no remaining content processed, check entire content
+    if (parts.length === 0) {
+      if (enableAutoCodeDetection) {
+        const detection = detectCodePatterns(content);
+        if (detection.isCode) {
+          parts.push({
+            type: 'code',
+            language: detection.language,
+            content: content
+          });
+        } else {
+          parts.push({
+            type: 'text',
+            content: content
+          });
+        }
+      } else {
+        parts.push({
+          type: 'text',
+          content: content
+        });
+      }
+    }
+
+    return parts;
+  }
+
+  function hasCodeBlocks(content: string): boolean {
+    const hasExplicitCodeBlocks = /```[\s\S]*?```/.test(content);
+    const hasIncompleteCodeBlocks = /```\w*\s*[\s\S]*$/.test(content) && !/```[\s\S]*?```/.test(content);
+
+    if (hasExplicitCodeBlocks || hasIncompleteCodeBlocks) return true;
+
+    // Heuristic auto-detection is opt-in (debug only). Without it, plain prose
+    // that happens to mention file paths or extensions stays as text.
+    if (!enableAutoCodeDetection) return false;
+
+    return detectCodePatterns(content).isCode;
+  }
+</script>
+
+<div 
+  bind:this={messagesContainer}
+  class="{className ? className : 'prompt-chat-messages-ct'} flex flex-col overflow-y-auto p-4 space-y-4"
+  class:loading={isLoading}
+  style="scroll-behavior: smooth;"
+  onscroll={handleScroll}
+>
+  <!-- Empty state with suggestions -->
+  {#if messages.length === 0 && !isLoading}
+    <div class="empty-state-ct flex flex-col items-center justify-center text-center max-w-sm w-full" style="align-self: center;">
+      <h3 class="text-lg font-semibold text-gray-900 dark:text-gray-100 mb-2">💬 Start a conversation</h3>
+      <p class="text-gray-600 dark:text-gray-400 text-sm mb-6">
+        Ask a question about your {context.type} to get started.
+      </p>
+      
+      <!-- Contextual suggestions -->
+      <SuggestionButtons 
+        {context} 
+        visible={true}
+        on:suggestionSelected={handleSuggestionSelected}
+      />
+    </div>
+  {/if}
+
+  <!-- Conversation turns (user prompt + assistant status grouped) -->
+  {#each chatTurns as turn, turnIndex (turn.user?.id ?? turn.replies[0]?.id ?? turnIndex)}
+    <div class="wc-chat-turn-ct space-y-1.5 rounded-lg border border-transparent p-0.5"
+         class:wc-chat-turn-head={turnHasWidgetHead(turn)}
+    >
+      {#if turn.user}
+        {@const userMsg = turn.user}
+        <div class="message-container wc-chat-turn-user ml-auto"
+             class:opacity-75={userMsg._optimistic}
+             class:opacity-50={userMsg._failed}
+             class:has-attachments={userMsg.attachments && userMsg.attachments.length > 0}
+        >
+          <div 
+            class="message-bubble relative rounded-lg shadow-sm text-sm leading-relaxed select-text"
+            class:p-3={!userMsg.attachments || userMsg.attachments.length === 0}
+            class:p-0={userMsg.attachments && userMsg.attachments.length > 0}
+            class:bubble-fit-content={userMsg.attachments && userMsg.attachments.length > 0}
+            style="background-color: {isCompleted ? MESSAGE_COLORS.USER_COMPLETED : MESSAGE_COLORS.USER_ACTIVE}; color: {MESSAGE_COLORS.USER_TEXT};"
+          >
+            {#if userMsg.messageContent}
+              <p class="whitespace-pre-wrap p-3">{userMsg.messageContent}</p>
+            {/if}
+            {#if userMsg.attachments && userMsg.attachments.length > 0}
+              <div class="message-attachments-container">
+                <AttachmentDisplay attachments={userMsg.attachments} variant="message" showPreview={true} />
+              </div>
+            {:else if userMsg.messageContent && messageImpliesAttachedImage(userMsg.messageContent)}
+              <div class="wc-missing-attachment-placeholder message-attachments-container flex flex-col items-center gap-1 px-3 pb-3">
+                <div class="wc-missing-attachment-thumb flex h-20 w-28 flex-col items-center justify-center gap-1 rounded-md border border-dashed border-gray-300 bg-gray-100 text-gray-400" title="Image not available" aria-label="Missing image attachment">
+                  <svg class="h-8 w-8 shrink-0" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="1.5" aria-hidden="true">
+                    <rect x="3" y="5" width="18" height="14" rx="2" />
+                    <path d="M4 4l16 16" stroke-linecap="round" />
+                  </svg>
+                </div>
+                <p class="wc-missing-attachment-caption max-w-[12rem] text-center text-xs text-gray-500">Image not saved in chat history. Re-attach with Upload before Send.</p>
+              </div>
+            {/if}
+          </div>
+        </div>
+
+        {@const commitReply = getTurnCommitReply(turn)}
+        {#if commitReply && showCommitMetaOnUserTurn(turn)}
+          <div class="wc-user-commit-meta-ct ml-auto flex max-w-[85%] flex-col items-end gap-1">
+            {#if commitReply.isWidgetHead}
+              <div class="checkpoint-label wc-head-commit-label">
+                <svg width="12" height="12" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.5">
+                  <path d="M20 6L9 17l-5-5"/>
+                </svg>
+                Head at commit
+              </div>
+            {/if}
+            {#if extractCommitShort(commitReply.messageContent || '')}
+              <span class="wc-user-commit-sha font-mono text-xs text-gray-500">
+                Commit {extractCommitShort(commitReply.messageContent || '')}
+              </span>
+            {/if}
+            {#if turn.user && canUndoHeadUserTurn(turn) && !isLoading}
+              <div class="restore-checkpoint-row flex justify-end">
+                <button
+                  class="restore-checkpoint-btn wc-undo-head-btn"
+                  onclick={() => handleRestoreCheckpoint(turn.user!.id)}
+                  title="Undo this prompt and revert code to the previous commit"
+                >
+                  <svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                    <path d="M3 12a9 9 0 1 0 9-9 9.75 9.75 0 0 0-6.74 2.74L3 8"/>
+                    <path d="M3 3v5h5"/>
+                  </svg>
+                  Undo last change
+                </button>
+              </div>
+            {:else if canRestoreToMessage(commitReply) && !isLoading}
+              <div class="restore-checkpoint-row flex justify-end">
+                <button
+                  class="restore-checkpoint-btn"
+                  onclick={() => handleRestoreCheckpoint(commitReply.id)}
+                  title="Restore widget code to this prompt's commit"
+                >
+                  <svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                    <path d="M3 12a9 9 0 1 0 9-9 9.75 9.75 0 0 0-6.74 2.74L3 8"/>
+                    <path d="M3 3v5h5"/>
+                  </svg>
+                  Restore Checkpoint
+                </button>
+              </div>
+            {/if}
+          </div>
+        {/if}
+      {/if}
+
+      {#each turn.replies as reply (reply.id)}
+        {#if shouldShowAssistantReply(reply)}
+        <div class="message-container wc-chat-turn-reply mr-auto"
+             class:opacity-75={reply._optimistic}
+             class:opacity-50={reply._failed}
+             class:current-checkpoint={reply.isWidgetHead && messageHasCommit(reply)}
+        >
+          <div
+            class="message-bubble relative rounded-lg shadow-sm text-sm leading-relaxed select-text p-3"
+            class:wc-codegen-failure={isCodegenFailure(reply)}
+            style="background-color: {reply.isWidgetHead && messageHasCommit(reply) ? '#e0f2fe' : MESSAGE_COLORS.ASSISTANT}; color: {MESSAGE_COLORS.ASSISTANT_TEXT};{reply.isWidgetHead && messageHasCommit(reply) ? ' border: 1.5px solid #7dd3fc;' : ''}"
+          >
+            {#if reply.messageContent}
+              <p class="whitespace-pre-wrap">{reply.messageContent}</p>
+            {/if}
+          </div>
+
+          {#if reply.isWidgetHead && messageHasCommit(reply)}
+            <div class="wc-head-commit-meta-ct flex flex-col items-start gap-0.5 mt-1">
+              <div class="checkpoint-label wc-head-commit-label">
+                <svg width="12" height="12" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2.5">
+                  <path d="M20 6L9 17l-5-5"/>
+                </svg>
+                Head at commit
+              </div>
+              {#if extractCommitShort(reply.messageContent || '')}
+                <span class="wc-assistant-commit-sha font-mono text-xs text-gray-500">
+                  Commit {extractCommitShort(reply.messageContent || '')}
+                </span>
+              {/if}
+            </div>
+          {/if}
+
+          {#if isCodegenFailure(reply) && turn.user && !isLoading}
+            <div class="retry-codegen-row flex justify-start mt-1">
+              <button
+                class="retry-codegen-btn"
+                onclick={() => handleRetryTurn(turn)}
+                title="Retry this code generation prompt"
+              >
+                <svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                  <path d="M21 12a9 9 0 1 1-2.64-6.36"/>
+                  <path d="M21 3v6h-6"/>
+                </svg>
+                Retry
+              </button>
+            </div>
+          {/if}
+
+          {#if reply.isWidgetHead && messageHasCommit(reply) && !isLoading}
+            <div class="restore-checkpoint-row flex justify-start mt-1">
+              <button
+                class="restore-checkpoint-btn wc-undo-head-btn"
+                onclick={() => handleRestoreCheckpoint(turn.user?.id ?? reply.id)}
+                title="Undo this prompt and revert code to the previous commit"
+              >
+                <svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                  <path d="M3 12a9 9 0 1 0 9-9 9.75 9.75 0 0 0-6.74 2.74L3 8"/>
+                  <path d="M3 3v5h5"/>
+                </svg>
+                Undo last change
+              </button>
+            </div>
+          {:else if canRestoreToMessage(reply) && !isLoading}
+            <div class="restore-checkpoint-row flex justify-start mt-1">
+              <button
+                class="restore-checkpoint-btn"
+                onclick={() => handleRestoreCheckpoint(reply.id)}
+                title="Restore widget code to this prompt's commit"
+              >
+                <svg width="14" height="14" viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+                  <path d="M3 12a9 9 0 1 0 9-9 9.75 9.75 0 0 0-6.74 2.74L3 8"/>
+                  <path d="M3 3v5h5"/>
+                </svg>
+                Restore Checkpoint
+              </button>
+            </div>
+          {/if}
+        </div>
+        {/if}
+      {/each}
+    </div>
+  {/each}
+
+  <!-- Show suggestions toggle (when there are messages) -->
+  {#if messages.length > 0 && !isLoading}
+    <div class="suggestions-toggle-ct flex justify-center py-2">
+      {#if showSuggestionsInline}
+        <SuggestionButtons 
+          {context} 
+          visible={true}
+          onHide={() => { showSuggestionsInline = false; }}
+          on:suggestionSelected={(e) => { showSuggestionsInline = false; handleSuggestionSelected(e); }}
+        />
+      {:else}
+        <button
+          class="suggestions-toggle-btn text-xs text-gray-400 hover:text-gray-600 px-3 py-1.5 rounded-full border border-gray-200 hover:border-gray-300 bg-white hover:bg-gray-50 transition-colors cursor-pointer"
+          onclick={() => { showSuggestionsInline = true; }}
+        >
+          Show suggestions
+        </button>
+      {/if}
+    </div>
+  {/if}
+
+  <!-- Loading indicator -->
+  {#if isLoading}
+    <div class="loading-indicator-ct flex items-center justify-start p-3">
+      <div class="rounded-lg shadow-sm text-sm leading-relaxed p-3"
+           style="background-color: {MESSAGE_COLORS.ASSISTANT}; color: {MESSAGE_COLORS.ASSISTANT_TEXT}"
+      >
+        <LoadingIndicator 
+          type="thinking" 
+          text="Assistant is thinking..." 
+          color="secondary"
+          size="md"
+        />
+      </div>
+    </div>
+  {/if}
+</div>
+
+<style>
+  .loading {
+    opacity: 0.75;
+  }
+  
+  /* Remove any height constraints from messages */
+  .space-y-2 > * {
+    max-height: none !important;
+  }
+  
+  .message-container {
+    max-width: 80%;
+  }
+  
+  /* Messages with attachments: text full-width on top, images on a row below */
+  .message-container.has-attachments {
+    max-width: 90%;
+    max-height: none !important;
+  }
+  
+  .message-attachments-container {
+    margin: 0;
+    padding: 8px 12px 12px;
+    width: 100%;
+    max-height: none !important;
+    overflow: visible !important;
+    display: flex;
+    flex-wrap: wrap;
+    gap: 8px;
+  }
+  
+  /* Ensure message bubbles don't constrain attachment size */
+  .message-container.has-attachments > div {
+    max-height: none !important;
+    overflow: visible !important;
+  }
+  
+  /* Message bubble with attachments — full width for text, images below */
+  .bubble-fit-content {
+    width: 100% !important;
+  }
+  
+  /* Ensure proper alignment for fit-content bubbles */
+  .message-container.has-attachments {
+    display: flex;
+    flex-direction: column;
+  }
+  
+  .message-container.has-attachments.ml-auto {
+    align-items: flex-end; /* Right align for user messages */
+  }
+  
+  .message-container.has-attachments.mr-auto {
+    align-items: flex-start; /* Left align for assistant messages */
+  }
+  
+  /* Restore checkpoint — always visible below user messages */
+  .wc-chat-turn-head {
+    border-color: #bae6fd;
+    background-color: #f8fcff;
+    padding: 6px;
+  }
+
+  .wc-codegen-failure {
+    border: 1px solid #fecaca;
+    background-color: #fef2f2 !important;
+  }
+
+  .wc-head-commit-label {
+    margin-bottom: 0;
+  }
+
+  .wc-head-commit-meta-ct .wc-head-commit-label {
+    margin-top: 0;
+  }
+
+  .restore-checkpoint-row {
+    margin-top: 4px;
+  }
+
+  .retry-codegen-row {
+    margin-top: 4px;
+  }
+  
+  .restore-checkpoint-btn,
+  .retry-codegen-btn {
+    display: inline-flex;
+    align-items: center;
+    gap: 4px;
+    padding: 3px 10px;
+    background: transparent;
+    color: #888;
+    border: 1px solid #ddd;
+    border-radius: 12px;
+    font-size: 11px;
+    font-weight: 500;
+    cursor: pointer;
+    transition: background-color 0.15s ease, color 0.15s ease, border-color 0.15s ease;
+    white-space: nowrap;
+  }
+  
+  .restore-checkpoint-btn:hover,
+  .retry-codegen-btn:hover {
+    background: rgba(0, 0, 0, 0.06);
+    color: #555;
+    border-color: #bbb;
+  }
+  
+  .retry-codegen-btn {
+    color: #b91c1c;
+    border-color: #fecaca;
+    background: #fff7f7;
+  }
+
+  .retry-codegen-btn:hover {
+    background: #fee2e2;
+    color: #991b1b;
+    border-color: #fca5a5;
+  }
+
+  .restore-checkpoint-btn svg,
+  .retry-codegen-btn svg {
+    flex-shrink: 0;
+  }
+
+  /* Copy message button — top-right, visible on hover */
+  .copy-msg-btn {
+    position: absolute;
+    top: 4px;
+    right: 4px;
+    display: flex;
+    align-items: center;
+    justify-content: center;
+    width: 22px;
+    height: 22px;
+    border-radius: 4px;
+    border: none;
+    background: rgba(0, 0, 0, 0.06);
+    color: #888;
+    cursor: pointer;
+    opacity: 0;
+    transition: opacity 0.15s ease;
+  }
+  .message-bubble:hover .copy-msg-btn {
+    opacity: 1;
+  }
+  .copy-msg-btn:hover {
+    background: rgba(0, 0, 0, 0.12);
+    color: #555;
+  }
+
+  /* Checkpoint label — shown above the restored message */
+  .checkpoint-label {
+    display: inline-flex;
+    align-items: center;
+    gap: 4px;
+    padding: 2px 8px;
+    background: #0ea5e9;
+    color: white;
+    border-radius: 8px;
+    font-size: 10px;
+    font-weight: 600;
+    letter-spacing: 0.02em;
+    margin-bottom: 4px;
+    align-self: flex-end;
+  }
+
+  /* Restored button state */
+  .restore-checkpoint-btn.restored {
+    opacity: 0.5;
+    cursor: default;
+    border-color: #7dd3fc;
+    color: #0ea5e9;
+  }
+  .restore-checkpoint-btn.restored:hover {
+    background: transparent;
+    color: #0ea5e9;
+    border-color: #7dd3fc;
+  }
+
+  /* Cap huge messages (legacy full-file dumps) so bubbles stay readable */
+  .message-bubble {
+    max-width: 100%;
+  }
+  .message-bubble p,
+  .message-content-mixed {
+    max-height: 280px;
+    overflow-y: auto;
+    overflow-x: auto;
+    word-break: break-word;
+  }
+  .message-content-mixed :global(pre) {
+    max-height: 240px;
+    overflow: auto;
+  }
+</style>