/* ═══════════════════════════════════════════════════════════════
   Computer — Retro-themed workspace chat UI
   Connects to Captain Claw's existing WebSocket protocol.
   ═══════════════════════════════════════════════════════════════ */

const $ = (sel) => document.querySelector(sel);
const $$ = (sel) => document.querySelectorAll(sel);

/* ── State ───────────────────────────────────────────────────── */

let ws = null;
let connected = false;
let currentTheme = "amiga";
let isProcessing = false;
let ledTimer = null;

// Store last prompt/result for Visual tab re-generation.
let lastPrompt = "";
let lastResult = "";
let lastAnswerTimestamp = "";
let visualGenerating = false;

// Track HTML files created by agent during the current turn.
let agentCreatedHtmlFiles = [];

// Visual generation settings (persisted in localStorage).
let selectedTokenTier = localStorage.getItem("computer-token-tier") || "standard";
let selectedModel = localStorage.getItem("computer-model") || "";
let availableModels = [];

// Persona state.
let availablePersonas = [];
let selectedPersona = localStorage.getItem("computer-persona") || "";

// Playbook state.
let availablePlaybooks = [];
let selectedPlaybookId = localStorage.getItem("computer-playbook") || ""; // '' = auto, '__none__' = disabled

// Session settings (persisted in session metadata).
let _sessionSettings = { name: "", description: "", instructions: "" };
let _publicCode = "";

function _updateSessionBadge(displayName, fallback) {
  const name = (displayName || "").trim();
  const code = _publicCode;
  if (name && code) {
    $("#session-badge").textContent = name + " (" + code + ")";
  } else if (name) {
    $("#session-badge").textContent = name;
  } else if (code) {
    $("#session-badge").textContent = code;
  } else if (fallback) {
    $("#session-badge").textContent = fallback;
  }
}

// BYOK (Bring Your Own Key) state.
let isPublicSession = false;
let byokProvider = localStorage.getItem("computer-byok-provider") || "";
let byokModel = localStorage.getItem("computer-byok-model") || "";
let byokApiKey = localStorage.getItem("computer-byok-apikey") || "";
let byokEnabled = localStorage.getItem("computer-byok-enabled") !== "false"; // default true if credentials exist
let byokActive = false; // server-confirmed active state

// Attachment state — arrays for multi-file support.
let pendingImages = [];   // [{path, name}, ...]
let pendingFiles = [];    // [{path, name}, ...]
const _IMAGE_EXTS = ["png", "jpg", "jpeg", "webp", "gif", "bmp"];
const _DATA_EXTS = ["csv", "xlsx", "xls", "pdf", "docx", "doc", "pptx", "ppt", "md", "txt", "zip"];

// Folder browser state.
let folderCurrentPath = null;
let _gwsAvailable = false;
let _gdriveCurrent = { id: "root", name: "My Drive" };
let _gdriveBcStack = [{ id: "root", name: "My Drive" }];

/* ── Exploration state ───────────────────────────────────────── */

const MAX_EXPLORATION_NODES = 200;

let explorationNodes = new Map();  // id -> ExplorationNode
let currentNodeId = null;
let pendingExploration = null;     // { parentId, edgeLabel, source }
let sessionId = null;              // set on welcome
let exploreDebounceTimer = null;
let historyBranchNodeId = null;    // set when user navigates to an old node

// postMessage bridge script injected into visual iframe HTML.
const EXPLORE_BRIDGE = `<script>
document.addEventListener('DOMContentLoaded', function() {
  // Wire up explore-link elements.
  document.querySelectorAll('.explore-link').forEach(function(el) {
    el.style.cursor = 'pointer';
    el.addEventListener('click', function(e) {
      e.preventDefault();
      window.parent.postMessage({
        type: 'explore-click',
        topic: el.dataset.topic || el.textContent.trim(),
        context: el.dataset.context || ''
      }, '*');
    });
  });

  // Selection-to-explore: show a floating button when user selects text.
  var exploreFab = document.createElement('div');
  exploreFab.id = 'explore-fab';
  exploreFab.innerHTML = '\\u{1F50D} Explore this';
  exploreFab.style.cssText = 'display:none;position:fixed;z-index:99999;' +
    'padding:5px 12px;border-radius:14px;font-size:12px;font-weight:700;' +
    'cursor:pointer;pointer-events:auto;user-select:none;' +
    'box-shadow:0 2px 8px rgba(0,0,0,0.3);transition:opacity 0.15s,transform 0.15s;' +
    'opacity:0;transform:translateY(4px);';
  document.body.appendChild(exploreFab);

  // High-contrast styling that works across all themes.
  // Use a dark background with white text — guaranteed readable.
  exploreFab.style.background = '#1a1a1a';
  exploreFab.style.color = '#f0f0f0';
  exploreFab.style.border = '1px solid #555';

  // Try to pick up the page accent for a subtle highlight.
  var cs = getComputedStyle(document.body);
  var accent = (cs.getPropertyValue('--accent') || '').trim();
  if (accent) {
    exploreFab.style.borderColor = accent;
    exploreFab.style.boxShadow = '0 2px 10px rgba(0,0,0,0.4), 0 0 0 1px ' + accent;
  }

  var fabTimeout = null;
  var selectedText = '';

  function showFab(x, y, text) {
    selectedText = text;
    exploreFab.style.left = Math.min(x, window.innerWidth - 140) + 'px';
    exploreFab.style.top = Math.max(y - 36, 4) + 'px';
    exploreFab.style.display = 'block';
    requestAnimationFrame(function() {
      exploreFab.style.opacity = '1';
      exploreFab.style.transform = 'translateY(0)';
    });
    clearTimeout(fabTimeout);
    fabTimeout = setTimeout(hideFab, 6000);
  }

  function hideFab() {
    exploreFab.style.opacity = '0';
    exploreFab.style.transform = 'translateY(4px)';
    setTimeout(function() { exploreFab.style.display = 'none'; }, 150);
    clearTimeout(fabTimeout);
  }

  document.addEventListener('mouseup', function(e) {
    if (e.target === exploreFab || exploreFab.contains(e.target)) return;
    var sel = window.getSelection();
    var text = (sel ? sel.toString() : '').trim();
    if (text.length >= 3 && text.length <= 300) {
      var range = sel.getRangeAt(0);
      var rect = range.getBoundingClientRect();
      showFab(rect.left + rect.width / 2 - 50, rect.top, text);
    } else {
      setTimeout(hideFab, 100);
    }
  });

  exploreFab.addEventListener('click', function(e) {
    e.preventDefault();
    e.stopPropagation();
    if (selectedText) {
      window.parent.postMessage({
        type: 'explore-click',
        topic: selectedText,
        context: ''
      }, '*');
      hideFab();
      window.getSelection().removeAllRanges();
    }
  });

  // Hide on scroll.
  document.addEventListener('scroll', hideFab, true);
});
<\/script>`;

function uuid4() {
  return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function(c) {
    const r = Math.random() * 16 | 0;
    return (c === 'x' ? r : (r & 0x3 | 0x8)).toString(16);
  });
}

function createExplorationNode(prompt, answer) {
  const node = {
    id: uuid4(),
    session_id: sessionId || '',
    parent_id: pendingExploration ? pendingExploration.parentId : null,
    edge_label: pendingExploration ? pendingExploration.edgeLabel : null,
    prompt: prompt,
    answer: answer,
    visual_html: null,
    theme: currentTheme,
    source: pendingExploration ? 'click' : 'manual',
    created_at: new Date().toISOString(),
    metadata: JSON.stringify(pendingExploration || {}),
  };

  explorationNodes.set(node.id, node);

  // Cap exploration tree.
  if (explorationNodes.size > MAX_EXPLORATION_NODES) {
    const oldest = explorationNodes.keys().next().value;
    explorationNodes.delete(oldest);
    logEntry("system", `Exploration tree capped at ${MAX_EXPLORATION_NODES} nodes`);
  }

  return node;
}

function injectExploreBridge(html) {
  // Inject the postMessage bridge before </body> or at end.
  if (html.includes('</body>')) {
    return html.replace('</body>', EXPLORE_BRIDGE + '</body>');
  }
  return html + EXPLORE_BRIDGE;
}

/* ── Exploration: save/load from backend ─────────────────────── */

async function saveExplorationNode(node) {
  try {
    await fetch('/api/computer/exploration', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify(node),
    });
  } catch (e) {
    console.warn('Failed to save exploration node:', e);
  }
}

async function updateExplorationNodeVisual(nodeId, visualHtml) {
  try {
    await fetch(`/api/computer/exploration/${nodeId}/visual`, {
      method: 'PUT',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ visual_html: visualHtml }),
    });
  } catch (e) {
    console.warn('Failed to update exploration node visual:', e);
  }
}

async function loadExplorationHistory(sid) {
  try {
    const res = await fetch(`/api/computer/exploration?session_id=${encodeURIComponent(sid)}`);
    if (!res.ok) return;
    const data = await res.json();
    const allNodes = data.nodes || [];
    // Cap to last 200 nodes to keep the UI responsive.
    const nodes = allNodes.length > 200 ? allNodes.slice(-200) : allNodes;
    explorationNodes.clear();
    for (const n of nodes) {
      explorationNodes.set(n.id, n);
    }
    // Set current to most recent.
    if (nodes.length > 0) {
      currentNodeId = nodes[nodes.length - 1].id;
    }
    console.log(`[Computer] Loaded ${nodes.length} exploration nodes (of ${allNodes.length} total)`);
    renderHistoryDrawer();
  } catch (e) {
    console.warn('Failed to load exploration history:', e);
  }
}

/* ── History drawer ──────────────────────────────────────────── */

function renderHistoryDrawer() {
  const nodes = Array.from(explorationNodes.values())
    .sort((a, b) => (b.created_at || '').localeCompare(a.created_at || ''));

  const isBranching = !!historyBranchNodeId;

  for (const tab of ['answer', 'blueprint']) {
    const drawer = $(`#${tab}-history`);
    if (!drawer) continue;
    const countEl = drawer.querySelector('.history-bar-count');
    const listEl = drawer.querySelector('.history-list');
    const labelEl = drawer.querySelector('.history-bar-label');
    countEl.textContent = nodes.length;

    // Show branch indicator in the bar.
    if (isBranching) {
      drawer.classList.add('branching');
      labelEl.innerHTML = `<span class="history-branch-badge">&#9095; fork</span> History (<span class="history-bar-count">${nodes.length}</span>)`;
    } else {
      drawer.classList.remove('branching');
      labelEl.innerHTML = `History (<span class="history-bar-count">${nodes.length}</span>)`;
    }

    if (nodes.length === 0) {
      drawer.style.display = 'none';
      continue;
    }
    drawer.style.display = '';

    listEl.innerHTML = nodes.map(n => {
      const t = n.created_at ? new Date(n.created_at) : null;
      const time = t ? t.toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' }) : '';
      const prompt = (n.prompt || '').slice(0, 55) + ((n.prompt || '').length > 55 ? '\u2026' : '');
      const active = n.id === currentNodeId ? ' active' : '';
      // Dim entries that will be pruned on next send (newer than branch point).
      let pruned = '';
      if (isBranching && historyBranchNodeId) {
        const branchNode = explorationNodes.get(historyBranchNodeId);
        if (branchNode && (n.created_at || '') > (branchNode.created_at || '')) {
          pruned = ' pruned';
        }
      }
      return `<div class="history-entry${active}${pruned}" data-node-id="${n.id}" onclick="historySelect('${n.id}')">
        <span class="history-entry-bullet">${n.id === currentNodeId ? '\u25CF' : '\u25CB'}</span>
        <span class="history-entry-time">${esc(time)}</span>
        <span class="history-entry-prompt">${esc(prompt)}</span>
      </div>`;
    }).join('');
  }
}

function toggleHistoryDrawer(tab) {
  const drawer = $(`#${tab}-history`);
  if (!drawer) return;
  drawer.classList.toggle('open');
}

function historySelect(nodeId) {
  navigateToNode(nodeId);
  renderHistoryDrawer();
}

function historyPrev(tab) {
  const nodes = Array.from(explorationNodes.values())
    .sort((a, b) => (a.created_at || '').localeCompare(b.created_at || ''));
  if (nodes.length === 0) return;
  const idx = nodes.findIndex(n => n.id === currentNodeId);
  if (idx > 0) {
    historySelect(nodes[idx - 1].id);
  }
}

function historyNext(tab) {
  const nodes = Array.from(explorationNodes.values())
    .sort((a, b) => (a.created_at || '').localeCompare(b.created_at || ''));
  if (nodes.length === 0) return;
  const idx = nodes.findIndex(n => n.id === currentNodeId);
  if (idx < nodes.length - 1) {
    historySelect(nodes[idx + 1].id);
  }
}

/* ── Exploration: click-to-prompt ────────────────────────────── */

let _exploreCountdownTimer = null;
let _exploreCountdownSecs = 0;

function handleIframeMessage(event) {
  if (!event.data || event.data.type !== 'explore-click') return;
  const { topic, context } = event.data;
  if (!topic) return;

  // Debounce rapid clicks.
  if (exploreDebounceTimer) return;
  exploreDebounceTimer = setTimeout(() => { exploreDebounceTimer = null; }, 500);

  console.log('%c[Computer] Explore click', 'color:#ff8800;font-weight:bold', { topic, context });
  showExploreConfirm(topic, context);
}

/** Play a tick/click sound for the countdown timer (themed) */
function _playTickSound() {
  if (soundMode === "off") return;
  try {
    const p = _snd().tick;
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const osc = ctx.createOscillator();
    const gain = ctx.createGain();
    osc.connect(gain);
    gain.connect(ctx.destination);
    osc.type = p.wave;
    osc.frequency.setValueAtTime(p.freq, now);
    osc.frequency.exponentialRampToValueAtTime(p.freqEnd, now + p.dur / 2);
    gain.gain.setValueAtTime(p.vol, now);
    gain.gain.exponentialRampToValueAtTime(0.001, now + p.dur);
    osc.start(now);
    osc.stop(now + p.dur);
  } catch (_) {}
}

function _startExploreCountdown() {
  _stopExploreCountdown();
  _exploreCountdownSecs = 4;
  _updateCountdownDisplay();
  _playTickSound();

  _exploreCountdownTimer = setInterval(() => {
    _exploreCountdownSecs--;
    if (_exploreCountdownSecs <= 0) {
      _stopExploreCountdown();
      confirmExploration();
    } else {
      _updateCountdownDisplay();
      _playTickSound();
    }
  }, 1000);
}

function _stopExploreCountdown() {
  if (_exploreCountdownTimer) {
    clearInterval(_exploreCountdownTimer);
    _exploreCountdownTimer = null;
  }
  _exploreCountdownSecs = 0;
}

function _updateCountdownDisplay() {
  const btn = $("#explore-send-btn");
  if (!btn || !btn._exploreTopic) return;
  const shortTopic = btn._exploreTopic.length > 20
    ? btn._exploreTopic.slice(0, 20) + "…"
    : btn._exploreTopic;
  btn.innerHTML = `▶ <span class="explore-btn-label">${esc(shortTopic)}</span> <span class="explore-countdown">${_exploreCountdownSecs}</span>`;
}

function showExploreConfirm(topic, context) {
  if (!connected) {
    logEntry("error", "Not connected — cannot explore");
    return;
  }

  const basePrompt = context
    ? `Explore: ${topic} — ${context}`
    : `Tell me more about: ${topic}`;

  // Put the prepared prompt in the input box for the user to see/edit.
  const input = $("#input-box");
  input.value = basePrompt;
  input.style.height = "auto";
  input.style.height = Math.min(input.scrollHeight, 120) + "px";

  // Show explore / cancel buttons inline in the input-actions row.
  let exploreBtn = $("#explore-send-btn");
  let cancelBtn = $("#explore-cancel-btn");
  if (!exploreBtn) {
    const inputActions = $("#input-actions");
    exploreBtn = document.createElement("button");
    exploreBtn.id = "explore-send-btn";
    exploreBtn.className = "explore-inline-btn";
    exploreBtn.title = "Send exploration prompt";
    exploreBtn.addEventListener("click", () => confirmExploration());
    inputActions.appendChild(exploreBtn);

    cancelBtn = document.createElement("button");
    cancelBtn.id = "explore-cancel-btn";
    cancelBtn.className = "explore-inline-btn explore-cancel";
    cancelBtn.textContent = "✕";
    cancelBtn.title = "Cancel exploration";
    cancelBtn.addEventListener("click", () => cancelExploration());
    inputActions.appendChild(cancelBtn);
  }

  // Store exploration metadata on the buttons for later use.
  exploreBtn._exploreTopic = topic;
  exploreBtn._exploreContext = context;

  // Truncate long topic labels.
  const shortTopic = topic.length > 20 ? topic.slice(0, 20) + "…" : topic;
  exploreBtn.innerHTML = `▶ <span class="explore-btn-label">${esc(shortTopic)}</span>`;

  exploreBtn.style.display = "inline-flex";
  cancelBtn.style.display = "inline-flex";

  // Hide the Send button — Explore replaces it.
  $("#send-btn").style.display = "none";

  // Focus the input so the user can immediately type additions.
  input.focus();
  // Place cursor at end.
  input.setSelectionRange(input.value.length, input.value.length);

  logEntry("system", `Explore: "${topic}" — edit prompt or press Explore to send`);

  // Start auto-exploration countdown (4 seconds).
  _startExploreCountdown();

  // Cancel countdown if user starts editing the input.
  input._exploreInputHandler = () => {
    if (_exploreCountdownTimer) {
      _stopExploreCountdown();
      // Restore button text without countdown.
      const st = exploreBtn._exploreTopic.length > 20
        ? exploreBtn._exploreTopic.slice(0, 20) + "…"
        : exploreBtn._exploreTopic;
      exploreBtn.innerHTML = `▶ <span class="explore-btn-label">${esc(st)}</span>`;
      logEntry("system", "Auto-explore paused — input edited");
    }
  };
  input.addEventListener("input", input._exploreInputHandler);
}

function confirmExploration() {
  _stopExploreCountdown();
  const btn = $("#explore-send-btn");
  if (!btn) return;

  const topic = btn._exploreTopic || "";

  // Store the parent linkage before sending.
  pendingExploration = {
    parentId: currentNodeId,
    edgeLabel: topic,
    source: 'click',
  };

  // Hide explore buttons and restore Send button.
  _hideExploreButtons();

  // Send whatever is in the input box (user may have edited it).
  handleSend();
}

function cancelExploration() {
  _stopExploreCountdown();
  _hideExploreButtons();
  $("#input-box").value = "";
  logEntry("system", "Exploration cancelled");
}

function _hideExploreButtons() {
  _stopExploreCountdown();
  const eb = $("#explore-send-btn");
  const cb = $("#explore-cancel-btn");
  if (eb) eb.style.display = "none";
  if (cb) cb.style.display = "none";
  $("#send-btn").style.display = "";
  // Remove input listener.
  const input = $("#input-box");
  if (input && input._exploreInputHandler) {
    input.removeEventListener("input", input._exploreInputHandler);
    input._exploreInputHandler = null;
  }
}

/* ── Exploration: navigate to historical node ────────────────── */

function navigateToNode(nodeId) {
  const node = explorationNodes.get(nodeId);
  if (!node) return;

  currentNodeId = nodeId;
  lastPrompt = node.prompt;
  lastResult = node.answer;

  // Detect if this is a branch point (navigating to a non-latest node).
  const sorted = Array.from(explorationNodes.values())
    .sort((a, b) => (a.created_at || '').localeCompare(b.created_at || ''));
  const latestId = sorted.length > 0 ? sorted[sorted.length - 1].id : null;
  historyBranchNodeId = (nodeId !== latestId) ? nodeId : null;

  // Render answer without creating a new node.
  const el = $("#answer-content");
  el.innerHTML = markdownToHtml(node.answer);
  el.classList.remove("output-empty");
  _renderAnswerActions(el, node.answer);

  // Restore visual if available.
  const frame = $("#visual-frame");
  const visualEl = $("#visual-content");
  if (node.visual_html) {
    visualEl.innerHTML = "";
    visualEl.classList.remove("output-empty");
    frame.srcdoc = injectExploreBridge(node.visual_html);
    frame.style.display = "block";
  } else {
    clearVisual();
  }

  // Rebuild blueprint.
  generateBlueprint(node.answer);

  // Update map highlighting & history drawer.
  renderMap();
  renderHistoryDrawer();

  if (historyBranchNodeId) {
    logEntry("system", `Branched to: ${node.prompt.slice(0, 50)}${node.prompt.length > 50 ? '…' : ''} — next message will fork from here`);
  } else {
    logEntry("system", `Navigated to: ${node.prompt.slice(0, 60)}${node.prompt.length > 60 ? '…' : ''}`);
  }
}

/* ── Boot sequences per theme ────────────────────────────────── */

const BOOT_SEQUENCES = {
  amiga: [
    { html: '<div class="boot-amiga-hand">💾</div><div class="boot-text">Insert Workbench disk</div>', delay: 1500 },
    { html: '<div class="boot-amiga-hand">💾</div><div class="boot-text">Reading disk...</div>', delay: 1200 },
    { html: '<div class="boot-amiga-hand">💾</div><div class="boot-text">Loading Kickstart...</div>', delay: 1400 },
    { html: '<div class="boot-amiga-check">✓</div><div class="boot-text">Kickstart 1.3 loaded</div>', delay: 1000 },
    { html: '<div class="boot-amiga-check">✓</div><div class="boot-text">Amiga Workbench 1.3</div>', delay: 800 },
  ],
  atarist: [
    { html: '<div style="font-size:32px">⚡</div><div class="boot-text">Atari ST — TOS 1.04</div>', delay: 1200 },
    { html: '<div style="font-size:32px">⚡</div><div class="boot-text">Memory test... 512KB OK</div>', delay: 1000 },
    { html: '<div style="font-size:32px">⚡</div><div class="boot-text">GEM Desktop loading...</div>', delay: 1200 },
  ],
  c64: [
    { html: '<div style="font-size:13px;text-align:left;font-family:Silkscreen,monospace;line-height:1.8">**** COMMODORE 64 GEOS V2.0 ****</div>', delay: 1200 },
    { html: '<div style="font-size:13px;text-align:left;font-family:Silkscreen,monospace;line-height:1.8">**** COMMODORE 64 GEOS V2.0 ****<br><br>64K RAM SYSTEM  38911 BASIC BYTES FREE</div>', delay: 1000 },
    { html: '<div style="font-size:13px;text-align:left;font-family:Silkscreen,monospace;line-height:1.8">**** COMMODORE 64 GEOS V2.0 ****<br><br>64K RAM SYSTEM  38911 BASIC BYTES FREE<br><br>LOADING DESKTOP...</div>', delay: 1400 },
  ],
  mac: [
    { html: '<div style="font-size:48px">🙂</div><div class="boot-text">Welcome to Macintosh</div>', delay: 1500 },
    { html: '<div style="font-size:48px">🙂</div><div class="boot-text">Loading System 7...</div>', delay: 1200 },
    { html: '<div style="font-size:48px">🙂</div><div class="boot-text">Loading extensions...</div>', delay: 1000 },
  ],
  win31: [
    { html: '<div style="font-size:13px;text-align:left;font-family:monospace;line-height:1.8">Microsoft(R) Windows(TM)<br>Version 3.1<br><br>Copyright (C) Microsoft Corp. 1985-1992</div>', delay: 1400 },
    { html: '<div style="font-size:13px;text-align:left;font-family:monospace;line-height:1.8">Microsoft(R) Windows(TM)<br>Version 3.1<br><br>Copyright (C) Microsoft Corp. 1985-1992<br><br>Loading drivers...</div>', delay: 1200 },
    { html: '<div style="font-size:13px;text-align:left;font-family:monospace;line-height:1.8">Microsoft(R) Windows(TM)<br>Version 3.1<br><br>Copyright (C) Microsoft Corp. 1985-1992<br><br>Loading Program Manager...</div>', delay: 1000 },
  ],
  hacker: [
    { html: '<div style="font-size:12px;text-align:left;font-family:monospace;line-height:1.6;text-shadow:0 0 8px #00ff41">Wake up, Neo...</div>', delay: 1800 },
    { html: '<div style="font-size:12px;text-align:left;font-family:monospace;line-height:1.6;text-shadow:0 0 8px #00ff41">Wake up, Neo...<br><br>The Matrix has you...</div>', delay: 1600 },
    { html: '<div style="font-size:12px;text-align:left;font-family:monospace;line-height:1.6;text-shadow:0 0 8px #00ff41">Wake up, Neo...<br><br>The Matrix has you...<br><br>Follow the white rabbit.</div>', delay: 1400 },
    { html: '<div style="font-size:12px;text-align:left;font-family:monospace;line-height:1.6;text-shadow:0 0 8px #00ff41">Wake up, Neo...<br><br>The Matrix has you...<br><br>Follow the white rabbit.<br><br>Knock, knock, Neo.</div>', delay: 1200 },
    { html: '<div style="font-size:14px;font-family:monospace;text-shadow:0 0 10px #00ff41">root@computer:~# <span style="animation:boot-blink 0.7s step-end infinite">_</span></div>', delay: 800 },
  ],
  modern: [
    { html: '<div style="font-size:28px;font-weight:300;letter-spacing:2px;opacity:0.3">Computer</div>', delay: 800 },
    { html: '<div style="font-size:28px;font-weight:300;letter-spacing:2px;opacity:0.7">Computer</div>', delay: 600 },
    { html: '<div style="font-size:28px;font-weight:300;letter-spacing:2px">Computer</div><div style="margin-top:16px;width:120px;height:3px;border-radius:2px;background:#2a2e3d;overflow:hidden"><div style="width:0%;height:100%;background:#6c8cff;border-radius:2px;animation:modern-load 1.2s ease-out forwards"></div></div>', delay: 1400 },
  ],
  win11: [
    { html: '<div style="font-size:32px">⊞</div>', delay: 800 },
    { html: '<div style="font-size:32px">⊞</div><div style="margin-top:12px;font-family:Segoe UI,sans-serif;font-size:14px;color:#1a1a1a">Windows 11</div>', delay: 1000 },
    { html: '<div style="font-size:32px">⊞</div><div style="margin-top:12px;font-family:Segoe UI,sans-serif;font-size:14px;color:#1a1a1a">Windows 11</div><div style="margin-top:16px;width:100px;height:3px;border-radius:2px;background:#e5e5e5;overflow:hidden"><div style="width:0%;height:100%;background:#0078d4;border-radius:2px;animation:modern-load 1.2s ease-out forwards"></div></div>', delay: 1400 },
  ],
  macos: [
    { html: '<div style="font-size:48px"></div>', delay: 800 },
    { html: '<div style="font-size:48px"></div><div style="margin-top:12px;font-family:-apple-system,sans-serif;font-size:14px;color:#e0e0e0">macOS</div>', delay: 1000 },
    { html: '<div style="font-size:48px"></div><div style="margin-top:12px;font-family:-apple-system,sans-serif;font-size:14px;color:#e0e0e0">macOS</div><div style="margin-top:16px;width:120px;height:4px;border-radius:4px;background:#3d3d3d;overflow:hidden"><div style="width:0%;height:100%;background:#fff;border-radius:4px;animation:modern-load 1.2s ease-out forwards"></div></div>', delay: 1400 },
  ],
  iphone: [
    { html: '<div style="font-size:48px;opacity:0.4"></div>', delay: 800 },
    { html: '<div style="font-size:48px"></div>', delay: 700 },
    { html: '<div style="font-size:48px"></div><div style="margin-top:16px;font-family:-apple-system,sans-serif;font-size:24px;font-weight:200;color:#fff;letter-spacing:2px">iPhone</div>', delay: 1000 },
  ],
  android: [
    { html: '<div style="font-size:48px;color:#3ddc84">▲</div>', delay: 800 },
    { html: '<div style="font-size:48px;color:#3ddc84">▲</div><div style="margin-top:12px;font-family:Roboto,sans-serif;font-size:16px;color:#e8eaed;letter-spacing:1px">android</div>', delay: 1000 },
    { html: '<div style="font-size:48px;color:#3ddc84">▲</div><div style="margin-top:12px;font-family:Roboto,sans-serif;font-size:16px;color:#e8eaed;letter-spacing:1px">android</div><div style="margin-top:12px;width:80px;height:3px;border-radius:2px;background:#333;overflow:hidden"><div style="width:0%;height:100%;background:#8ab4f8;border-radius:2px;animation:modern-load 1.2s ease-out forwards"></div></div>', delay: 1200 },
  ],
  nokia7110: [
    { html: '<div style="font-size:12px;font-family:Share Tech Mono,monospace;font-weight:700;color:#1a2a0a;line-height:1.6;text-align:center">NOKIA</div>', delay: 1200 },
    { html: '<div style="font-size:12px;font-family:Share Tech Mono,monospace;font-weight:700;color:#1a2a0a;line-height:1.6;text-align:center">NOKIA<br><br>CONNECTING PEOPLE</div>', delay: 1400 },
    { html: '<div style="font-size:11px;font-family:Share Tech Mono,monospace;font-weight:700;color:#1a2a0a;line-height:1.5;text-align:left">Menu<br>──────────<br>1 Messages<br>2 Contacts<br>3 Computer<br>──────────<br>Select ▸</div>', delay: 1000 },
  ],
  nokiacomm: [
    { html: '<div style="font-size:13px;font-family:Space Mono,monospace;color:#b0c0a0;text-align:left;line-height:1.8">Nokia 9110 Communicator</div>', delay: 1000 },
    { html: '<div style="font-size:13px;font-family:Space Mono,monospace;color:#b0c0a0;text-align:left;line-height:1.8">Nokia 9110 Communicator<br><br>ROM Version 5.02<br><br>Initializing applications...</div>', delay: 1400 },
    { html: '<div style="font-size:13px;font-family:Space Mono,monospace;color:#b0c0a0;text-align:left;line-height:1.8">Nokia 9110 Communicator<br><br>ROM Version 5.02<br><br>Loading Computer...</div>', delay: 1000 },
  ],
};

/* ── Theme descriptions for Visual tab LLM prompt ────────────── */

const THEME_INSTRUCTIONS = {
  amiga: "Style it like an Amiga Workbench application: use colors #0055aa (blue), #ff8800 (orange), #aaaaaa (grey), #ffffff (white). Monospaced font. Beveled 3D borders, no rounded corners. Think 1987 Commodore Amiga.",
  atarist: "Style it like an Atari ST GEM application: clean black-and-white UI with green (#00aa00) accents. Sharp lines, minimal decoration, monospaced font. Think 1985 Atari ST desktop.",
  c64: "Style it like a C64 GEOS application: purple/lavender palette (#6c6c9e, #b4b4dc, #d4d4ee, #3c3c6c). Chunky pixel aesthetic, monospaced font. Think 1986 Commodore 64 GEOS desktop.",
  mac: "Style it like a classic Macintosh System 7 application: black-and-white with grey (#cccccc) chrome, 1-bit iconography feel. Chicago-like sans-serif font. Rounded corners. Think 1991 Macintosh.",
  win31: "Style it like a Windows 3.1 Program Manager application: teal (#008080) and navy (#000080) accents, silver (#c0c0c0) surfaces, beveled borders. MS Sans Serif style font. Think 1992 Windows 3.1.",
  hacker: "Style it like a hacker terminal: pure black background, electric green (#00ff41) text and borders. Monospaced font. Text should glow with text-shadow. Add subtle scanline effect. Think The Matrix.",
  modern: "Style it as a modern dark web application: dark slate background (#0f1117, #1a1d27), blue accent (#6c8cff), clean sans-serif typography, rounded corners (8px), subtle shadows. Polished SaaS aesthetic.",
  win11: "Style it like a Windows 11 application: light theme with white (#fff) surfaces, subtle grey (#e5e5e5) borders, blue (#0078d4) accent. Segoe UI font. Rounded corners (8px), Mica-like frosted feel. Clean modern Microsoft design.",
  macos: "Style it like a modern macOS dark mode application: dark (#2d2d2d) surfaces, blue (#0a84ff) accent, SF Pro-like sans-serif font. Rounded corners (10px), subtle vibrancy. Think modern Apple desktop app.",
  iphone: "Style it like an iOS dark mode app: pure black (#000) background, dark grey (#1c1c1e) surfaces, blue (#0a84ff) accent. SF Pro-like sans-serif, large rounded corners (12px). Think iOS 17 native app.",
  android: "Style it like a Material You Android app: dark (#121212) background, pastel blue (#8ab4f8) and purple (#bb86fc) accents, Roboto font. Large rounded corners (12px), subtle elevation shadows. Think Material Design 3.",
  nokia7110: "Style it like a Nokia 7110 phone screen: olive-green LCD background (#8b9f6b, #9aaf7a), dark green (#1a2a0a) text, very simple chunky layout. Monospaced font, no decoration, minimalist. Think 1999 WAP phone.",
  nokiacomm: "Style it like a Nokia Communicator (9110) app: grey-green (#d0d8c0, #dce4cc) background, dark green (#202820) text, navy blue (#004488) accent. Monospaced font, simple beveled borders. Think 1998 PDA phone.",
};

/* ── Boot ────────────────────────────────────────────────────── */

async function runBoot() {
  const screen = $("#boot-screen");
  const content = $("#boot-content");
  const steps = BOOT_SEQUENCES[currentTheme] || BOOT_SEQUENCES.amiga;

  for (const step of steps) {
    content.innerHTML = step.html;
    await sleep(step.delay);
  }

  // Fade out.
  await sleep(300);
  screen.classList.add("hiding");
  await sleep(500);
  screen.classList.add("hidden");
}

function sleep(ms) {
  return new Promise((r) => setTimeout(r, ms));
}

/* ── Theme switching ─────────────────────────────────────────── */

function setTheme(theme) {
  document.body.classList.remove(`theme-${currentTheme}`);
  document.body.dataset.theme = theme;
  document.body.classList.add(`theme-${theme}`);
  currentTheme = theme;
  localStorage.setItem("computer-theme", theme);
}

/* ── Theme modal ─────────────────────────────────────────────── */

const THEME_LIST = [
  { id: "amiga",      name: "Amiga Workbench", bg: "#0055aa", fg: "#ff8800", pattern: "repeating-conic-gradient(#0055aa 0% 25%, #004488 0% 50%) 0 0 / 4px 4px" },
  { id: "atarist",    name: "Atari ST GEM",    bg: "#00aa00", fg: "#000000", pattern: null },
  { id: "c64",        name: "C64 GEOS",        bg: "#6c6c9e", fg: "#d4d4ee", pattern: "repeating-conic-gradient(#6c6c9e 0% 25%, #5a5a8a 0% 50%) 0 0 / 2px 2px" },
  { id: "mac",        name: "Classic Mac",      bg: "#666699", fg: "#000000", pattern: "repeating-conic-gradient(#666699 0% 25%, #555588 0% 50%) 0 0 / 2px 2px" },
  { id: "win31",      name: "Windows 3.1",      bg: "#008080", fg: "#ffffff", pattern: null },
  { id: "hacker",     name: "Hacker",           bg: "#000000", fg: "#00ff41", pattern: null },
  { id: "modern",     name: "Modern",           bg: "#0f1117", fg: "#6c8cff", pattern: null },
  { id: "win11",      name: "Windows 11",       bg: "#f3f3f3", fg: "#0078d4", pattern: null },
  { id: "macos",      name: "macOS",            bg: "#1e1e1e", fg: "#0a84ff", pattern: null },
  { id: "iphone",     name: "iPhone",           bg: "#000000", fg: "#0a84ff", pattern: null },
  { id: "android",    name: "Android",          bg: "#121212", fg: "#8ab4f8", pattern: null },
  { id: "nokia7110",  name: "Nokia 7110",       bg: "#8b9f6b", fg: "#1a2a0a", pattern: null },
  { id: "nokiacomm",  name: "Nokia Communicator", bg: "#d0d8c0", fg: "#004488", pattern: null },
];

/* ── Custom themes (user-uploaded, stored in localStorage) ──── */

// All CSS variable keys that define a theme.
const THEME_VARS = [
  "bg", "bg-pattern", "surface", "surface-alt",
  "chrome", "chrome-hi", "chrome-lo", "chrome-dark",
  "titlebar-bg", "titlebar-fg", "titlebar-stripe",
  "text", "text-dim", "text-inv",
  "accent", "accent-alt",
  "input-bg", "input-border",
  "btn-bg", "btn-fg", "btn-active",
  "log-bg", "log-text", "log-dim",
  "led-on", "led-off",
  "scrollbar-track", "scrollbar-thumb",
  "font", "font-size", "font-weight", "line-height",
  "font-title", "font-title-size",
  "bevel", "radius",
  "win-close-w", "win-depth-w", "win-gadget-h",
];

// Template based on the Modern theme — a clean starting point.
const THEME_TEMPLATE = {
  name: "My Custom Theme",
  description: "Describe your theme style for visual generation prompts.",
  variables: {
    "bg":             "#0f1117",
    "bg-pattern":     "none",
    "surface":        "#1a1d27",
    "surface-alt":    "#222633",
    "chrome":         "#1a1d27",
    "chrome-hi":      "#2a2e3d",
    "chrome-lo":      "#13151c",
    "chrome-dark":    "#2a2e3d",
    "titlebar-bg":    "#1a1d27",
    "titlebar-fg":    "#e0e0e6",
    "titlebar-stripe": "none",
    "text":           "#e0e0e6",
    "text-dim":       "#8888a0",
    "text-inv":       "#0f1117",
    "accent":         "#6c8cff",
    "accent-alt":     "#a78bfa",
    "input-bg":       "#13151c",
    "input-border":   "#2a2e3d",
    "btn-bg":         "#6c8cff",
    "btn-fg":         "#ffffff",
    "btn-active":     "#5070dd",
    "log-bg":         "#0a0d14",
    "log-text":       "#7ee787",
    "log-dim":        "#4a6a52",
    "led-on":         "#6c8cff",
    "led-off":        "#2a2e3d",
    "scrollbar-track": "#1a1d27",
    "scrollbar-thumb": "#2a2e3d",
    "font":           "-apple-system, BlinkMacSystemFont, \"Segoe UI\", Roboto, sans-serif",
    "font-size":      "14px",
    "font-weight":    "400",
    "line-height":    "1.55",
    "font-title":     "var(--font)",
    "font-title-size": "13px",
    "bevel":          "1px",
    "radius":         "8px",
    "win-close-w":    "0px",
    "win-depth-w":    "0px",
    "win-gadget-h":   "0px",
  },
  boot: [
    { text: "Loading custom theme...", delay: 600 },
    { text: "Ready.", delay: 400 }
  ]
};

let customThemes = []; // Array of { id, name, description, variables, boot }

function loadCustomThemes() {
  try {
    const raw = localStorage.getItem("computer-custom-themes");
    if (raw) customThemes = JSON.parse(raw);
  } catch (e) {
    console.warn("Failed to load custom themes:", e);
    customThemes = [];
  }
}

function saveCustomThemes() {
  localStorage.setItem("computer-custom-themes", JSON.stringify(customThemes));
}

function customThemeId(name) {
  // Generate a stable CSS-safe id from the name.
  return "custom-" + name.toLowerCase().replace(/[^a-z0-9]+/g, "-").replace(/^-|-$/g, "").slice(0, 30);
}

function injectCustomThemeCSS(theme) {
  // Remove old style element for this theme if present.
  const existingEl = document.getElementById(`custom-theme-style-${theme.id}`);
  if (existingEl) existingEl.remove();

  // Build CSS from variables.
  let css = `.theme-${theme.id} {\n`;
  for (const [key, val] of Object.entries(theme.variables)) {
    if (THEME_VARS.includes(key)) {
      css += `  --${key}: ${val};\n`;
    }
  }
  css += `}\n`;

  // Boot screen.
  const bg = theme.variables["bg"] || "#000";
  const fg = theme.variables["text"] || "#fff";
  css += `.theme-${theme.id} #boot-screen { background: ${bg}; }\n`;
  css += `.theme-${theme.id} #boot-content { color: ${fg}; }\n`;

  // Modern-style window overrides (hide retro gadgets, rounded corners).
  const radius = theme.variables["radius"] || "8px";
  const radiusPx = parseInt(radius);
  if (radiusPx >= 6) {
    css += `.theme-${theme.id} .window { border-radius: ${radius}; box-shadow: 0 2px 12px rgba(0,0,0,0.3); }\n`;
    css += `.theme-${theme.id} .win-btn { display: none; }\n`;
    css += `.theme-${theme.id} ::-webkit-scrollbar { width: 6px; }\n`;
    css += `.theme-${theme.id} ::-webkit-scrollbar-button { display: none; }\n`;
    css += `.theme-${theme.id} ::-webkit-scrollbar-thumb { border-radius: 3px; border: none; }\n`;
  }

  const style = document.createElement("style");
  style.id = `custom-theme-style-${theme.id}`;
  style.textContent = css;
  document.head.appendChild(style);
}

function registerCustomThemeBoot(theme) {
  // Register boot sequence from the theme definition.
  const bootSteps = (theme.boot || []).map((step) => ({
    html: `<div style="font-size:14px">${esc(step.text || "")}</div>`,
    delay: step.delay || 500,
  }));
  if (bootSteps.length === 0) {
    bootSteps.push({ html: `<div style="font-size:14px">${esc(theme.name)}</div>`, delay: 600 });
  }
  BOOT_SEQUENCES[theme.id] = bootSteps;

  // Register theme instructions for visual generation.
  if (theme.description) {
    THEME_INSTRUCTIONS[theme.id] = theme.description;
  } else {
    const accent = theme.variables["accent"] || "#6c8cff";
    const bg = theme.variables["bg"] || "#000";
    THEME_INSTRUCTIONS[theme.id] = `Style it to match this custom theme: background ${bg}, accent color ${accent}. Use the aesthetic described by the theme name "${theme.name}".`;
  }
}

function downloadThemeTemplate() {
  const blob = new Blob([JSON.stringify(THEME_TEMPLATE, null, 2)], { type: "application/json" });
  const url = URL.createObjectURL(blob);
  const a = document.createElement("a");
  a.href = url;
  a.download = "computer-theme-template.json";
  a.click();
  URL.revokeObjectURL(url);
}

function validateCustomTheme(obj) {
  if (!obj || typeof obj !== "object") return "Invalid JSON — expected an object.";
  if (!obj.name || typeof obj.name !== "string" || obj.name.trim().length < 2) {
    return 'Missing or invalid "name" (must be at least 2 characters).';
  }
  if (!obj.variables || typeof obj.variables !== "object") {
    return 'Missing "variables" object with CSS custom properties.';
  }
  // Check that at least some key variables are present.
  const required = ["bg", "text", "accent", "surface"];
  const missing = required.filter((k) => !obj.variables[k]);
  if (missing.length > 0) {
    return `Missing required variables: ${missing.join(", ")}`;
  }
  return null; // valid
}

function uploadCustomTheme(file) {
  const reader = new FileReader();
  reader.onload = (e) => {
    try {
      const obj = JSON.parse(e.target.result);
      const err = validateCustomTheme(obj);
      if (err) {
        alert("Invalid theme file:\n\n" + err);
        return;
      }

      const id = customThemeId(obj.name);

      // Check for duplicates — overwrite if same name.
      const existingIdx = customThemes.findIndex((t) => t.id === id);

      const theme = {
        id,
        name: obj.name.trim(),
        description: (obj.description || "").trim(),
        variables: {},
        boot: obj.boot || [],
      };

      // Copy only valid variable keys.
      for (const key of THEME_VARS) {
        if (obj.variables[key] !== undefined) {
          theme.variables[key] = String(obj.variables[key]);
        }
      }

      if (existingIdx >= 0) {
        customThemes[existingIdx] = theme;
      } else {
        customThemes.push(theme);
      }

      saveCustomThemes();
      injectCustomThemeCSS(theme);
      registerCustomThemeBoot(theme);
      renderCustomThemeGrid();

      logEntry("system", `Custom theme "${theme.name}" ${existingIdx >= 0 ? "updated" : "uploaded"}`);

      // Auto-select the new theme.
      setTheme(theme.id);
      updateThemeModalSelection();
      setTimeout(() => closeThemeModal(), 300);

    } catch (parseErr) {
      alert("Failed to parse theme file:\n\n" + parseErr.message);
    }
  };
  reader.readAsText(file);
}

function deleteCustomTheme(id) {
  const idx = customThemes.findIndex((t) => t.id === id);
  if (idx < 0) return;

  const theme = customThemes[idx];
  customThemes.splice(idx, 1);
  saveCustomThemes();

  // Remove injected CSS.
  const styleEl = document.getElementById(`custom-theme-style-${id}`);
  if (styleEl) styleEl.remove();

  // Remove boot/instructions.
  delete BOOT_SEQUENCES[id];
  delete THEME_INSTRUCTIONS[id];

  // If this theme was active, switch to Modern.
  if (currentTheme === id) {
    setTheme("modern");
  }

  renderCustomThemeGrid();
  updateThemeModalSelection();
  logEntry("system", `Custom theme "${theme.name}" deleted`);
}

function renderCustomThemeGrid() {
  const grid = $("#custom-theme-grid");
  const empty = $("#custom-theme-empty");

  // Remove old custom cards (but keep empty placeholder).
  grid.querySelectorAll(".theme-card.custom-theme").forEach((c) => c.remove());

  if (customThemes.length === 0) {
    if (empty) empty.style.display = "";
    return;
  }

  if (empty) empty.style.display = "none";

  for (const t of customThemes) {
    const card = document.createElement("div");
    card.className = "theme-card custom-theme" + (t.id === currentTheme ? " selected" : "");
    card.dataset.theme = t.id;

    const preview = document.createElement("div");
    preview.className = "theme-card-preview";
    preview.style.background = t.variables["bg-pattern"] && t.variables["bg-pattern"] !== "none"
      ? t.variables["bg-pattern"]
      : (t.variables["bg"] || "#333");
    preview.style.color = t.variables["accent"] || "#fff";
    preview.textContent = "Aa";

    const nameRow = document.createElement("div");
    nameRow.className = "theme-card-name";

    const nameSpan = document.createElement("span");
    nameSpan.textContent = t.name;
    nameSpan.style.flex = "1";
    nameSpan.style.overflow = "hidden";
    nameSpan.style.textOverflow = "ellipsis";

    const delBtn = document.createElement("button");
    delBtn.className = "theme-card-delete";
    delBtn.title = "Delete this theme";
    delBtn.textContent = "✕";
    delBtn.addEventListener("click", (e) => {
      e.stopPropagation();
      if (confirm(`Delete custom theme "${t.name}"?`)) {
        deleteCustomTheme(t.id);
      }
    });

    nameRow.appendChild(nameSpan);
    nameRow.appendChild(delBtn);

    card.appendChild(preview);
    card.appendChild(nameRow);
    card.addEventListener("click", () => {
      setTheme(t.id);
      updateThemeModalSelection();
      setTimeout(() => closeThemeModal(), 200);
    });

    grid.appendChild(card);
  }
}

function updateThemeModalSelection() {
  const allCards = document.querySelectorAll("#theme-grid .theme-card, #custom-theme-grid .theme-card");
  allCards.forEach((c) => {
    c.classList.toggle("selected", c.dataset.theme === currentTheme);
  });
}

function initThemeModal() {
  const modal = $("#theme-modal");
  const grid = $("#theme-grid");
  const btn = $("#theme-btn");

  // Load custom themes from localStorage and inject their CSS.
  loadCustomThemes();
  for (const t of customThemes) {
    injectCustomThemeCSS(t);
    registerCustomThemeBoot(t);
  }

  // Build built-in theme cards.
  for (const t of THEME_LIST) {
    const card = document.createElement("div");
    card.className = "theme-card" + (t.id === currentTheme ? " selected" : "");
    card.dataset.theme = t.id;

    const preview = document.createElement("div");
    preview.className = "theme-card-preview";
    preview.style.background = t.pattern || t.bg;
    preview.style.color = t.fg;
    preview.textContent = "Aa";

    const name = document.createElement("div");
    name.className = "theme-card-name";
    name.textContent = t.name;

    card.appendChild(preview);
    card.appendChild(name);
    card.addEventListener("click", () => {
      setTheme(t.id);
      updateThemeModalSelection();
      // Close modal after a short delay so user sees the selection.
      setTimeout(() => closeThemeModal(), 200);
    });
    grid.appendChild(card);
  }

  // Render custom themes.
  renderCustomThemeGrid();

  // Download template button.
  const dlBtn = $("#theme-download-tpl");
  if (dlBtn) dlBtn.addEventListener("click", (e) => {
    e.stopPropagation();
    downloadThemeTemplate();
  });

  // Upload button.
  const uploadInput = $("#theme-upload-input");
  if (uploadInput) uploadInput.addEventListener("change", (e) => {
    const file = e.target.files[0];
    if (file) {
      uploadCustomTheme(file);
      e.target.value = ""; // reset so same file can be re-uploaded
    }
  });

  // Open modal.
  btn.addEventListener("click", () => {
    modal.classList.add("active");
    updateThemeModalSelection();
  });

  // Close on backdrop click.
  modal.querySelector(".modal-backdrop").addEventListener("click", closeThemeModal);
  modal.querySelector(".modal-close").addEventListener("click", closeThemeModal);

  // Close on Escape.
  document.addEventListener("keydown", (e) => {
    if (e.key === "Escape" && modal.classList.contains("active")) {
      closeThemeModal();
    }
  });
}

function closeThemeModal() {
  $("#theme-modal").classList.remove("active");
}

/* ── WebSocket connection ────────────────────────────────────── */

function connect() {
  const protocol = location.protocol === "https:" ? "wss:" : "ws:";
  ws = new WebSocket(`${protocol}//${location.host}/ws`);

  ws.onopen = () => {
    connected = true;
    _errorSoundPlayed = false; // reset so next error can sound again
    $(".status-dot").classList.add("connected");
    logEntry("system", "Connected to Captain Claw");
  };

  ws.onclose = () => {
    connected = false;
    $(".status-dot").classList.remove("connected");
    logEntry("error", "Connection lost — reconnecting...");
    setTimeout(connect, 3000);
  };

  ws.onerror = () => {
    logEntry("error", "WebSocket error");
  };

  ws.onmessage = (event) => {
    try {
      const data = JSON.parse(event.data);
      handleMessage(data);
    } catch (e) {
      console.warn("Bad message:", e);
    }
  };
}

function send(data) {
  if (ws && ws.readyState === WebSocket.OPEN) {
    ws.send(JSON.stringify(data));
  }
}

/* ── Message handling ────────────────────────────────────────── */

function handleMessage(data) {
  switch (data.type) {
    case "welcome":
      handleWelcome(data);
      break;

    case "chat_message":
      handleChatMessage(data);
      break;

    case "monitor":
      handleMonitor(data);
      break;

    case "status":
      handleStatus(data);
      break;

    case "thinking":
      handleThinking(data);
      break;

    case "tool_output_inline":
    case "tool_stream":
      handleToolStream(data);
      break;

    case "response_stream":
      handleResponseStream(data);
      break;

    case "error":
      logEntry("error", data.message || data.error || "Unknown error");
      if (streamPanelActive && streamPanelSource === "answer") {
        appendStreamText("\n⚠ " + (data.message || data.error || "Error") + "\n");
        finishStreamPanel("Error");
      }
      break;

    case "session_info":
      if (data.name) {
        $("#session-badge").textContent = data.name;
      }
      // Update session id and reload exploration history when session changes.
      var newSid = data.id || data.session_id;
      if (newSid && newSid !== sessionId) {
        sessionId = newSid;
        explorationNodes.clear();
        currentNodeId = null;
        historyBranchNodeId = null;
        pendingExploration = null;
        lastPrompt = "";
        lastResult = "";
        agentCreatedHtmlFiles = [];
        clearAnswer();
        clearVisual();
        renderHistoryDrawer();
        $("#log-content").innerHTML = "";
        logEntry("system", "Session switched — " + (data.name || newSid.slice(0, 8)));
        loadExplorationHistory(sessionId);
      }
      break;

    case "session_switched":
      // Backend signals a full session switch (e.g. /nuke).
      // session_info broadcast follows, which handles the reload.
      break;

    case "command_result":
      if (data.command === "/btw") {
        logEntry("btw", data.content || "Noted.");
      } else {
        logEntry("system", data.content || "");
      }
      break;

    case "next_steps":
      renderNextSteps(data.options);
      break;

    case "approval_request":
      showApprovalModal(data.id, data.message, data.category || "action");
      break;

    case "byok_status":
      handleByokStatus(data);
      break;

    case "session_settings_saved":
      // Update local state after server confirms save.
      _sessionSettings.name = data.session_name || "";
      _sessionSettings.description = data.session_description || "";
      _sessionSettings.instructions = data.session_instructions || "";
      _updateSessionBadge(_sessionSettings.name);
      logEntry("system", "Session settings saved");
      break;
  }
}

function handleWelcome(data) {
  // Session info is nested under data.session (from ws_handler).
  const sess = data.session || {};
  _publicCode = data.public_code || "";
  const _ssName = (data.session_settings && data.session_settings.session_name) || "";
  _updateSessionBadge(_ssName, sess.name || data.session_name || "");
  // Capture session id for exploration persistence and file scoping.
  sessionId = sess.id || data.session_id || data.id || null;
  logEntry("system", `Session ready${sessionId ? ' (' + sessionId.slice(0, 8) + '\u2026)' : ''}`);

  // Restore session settings from welcome payload.
  if (data.session_settings) {
    _sessionSettings.name = data.session_settings.session_name || "";
    _sessionSettings.description = data.session_settings.session_description || "";
    _sessionSettings.instructions = data.session_settings.session_instructions || "";
    _sessionSettings.locked = !!data.session_settings.locked;
  }
  // Hide session settings button when locked for public users.
  const _ssBtn = $("#session-settings-btn");
  if (_ssBtn) {
    _ssBtn.style.display = (_sessionSettings.locked && data.is_public) ? "none" : "";
  }

  // BYOK: detect public mode and show/hide BYOK button.
  isPublicSession = !!data.is_public;
  if (isPublicSession) {
    $("#byok-btn").style.display = "";
    // In public mode, apply BYOK credentials if saved and enabled.
    if (byokEnabled && byokProvider && byokModel && byokApiKey && connected) {
      send({ type: "set_byok", provider: byokProvider, model: byokModel, api_key: byokApiKey });
      logEntry("system", `Applying saved BYOK: ${byokProvider}/${byokModel}`);
    }
  } else {
    $("#byok-btn").style.display = "none";
    // Apply the saved model selection to the session (non-public only).
    if (selectedModel && connected) {
      send({ type: "set_model", selector: selectedModel });
      logEntry("system", `Applying saved model: ${selectedModel}`);
    }
  }

  // Apply the saved persona selection to the session.
  if (selectedPersona && connected) {
    send({ type: "set_personality", personality_id: selectedPersona });
    logEntry("system", `Applying saved persona: ${selectedPersona}`);
  }

  // Apply the saved playbook selection to the session.
  if (selectedPlaybookId && connected) {
    send({ type: "set_playbook", playbook_id: selectedPlaybookId });
    const pbLabel = selectedPlaybookId === "__none__" ? "None"
      : (availablePlaybooks.find(p => p.id === selectedPlaybookId)?.name || selectedPlaybookId.slice(0, 8));
    logEntry("system", `Applying saved playbook: ${pbLabel}`);
  }

  // Load exploration history.
  if (sessionId) {
    loadExplorationHistory(sessionId);
  }

  // Replay history into the answer panel.
  if (data.history && data.history.length > 0) {
    for (const msg of data.history) {
      if (msg.role === "assistant" && msg.content) {
        renderAnswer(msg.content, true, msg.timestamp || "");
      }
      // Replay media messages.
      if (msg.content) {
        if (msg.role === "html_file" || msg.role === "image" || msg.role === "audio") {
          handleChatMessage(msg);
        }
      }
    }
  }
}

function handleChatMessage(data) {
  if (data.role === "assistant") {
    setProcessing(false);
    // Finish the stream panel for the answer flow.
    if (streamPanelActive && streamPanelSource === "answer") {
      finishStreamPanel("Agent response complete");
      // Auto-close stream panel after answer arrives — unless Visual tab
      // is active, because visual generation will start next and reuse it.
      const visualTabActive = $(".tab-btn[data-tab='visual']").classList.contains("active");
      if (visualTabActive) {
        // Keep panel open — visual generation will take over.
      } else {
        setTimeout(() => {
          closeStreamPanel();
          if (_isMobile()) {
            document.body.classList.remove("mobile-streaming");
          }
        }, 600);
      }
    }
    renderAnswer(data.content || "", false, data.timestamp || "");
  }
  // Track HTML files created by the agent (broadcast by backend).
  if (data.role === "html_file" && data.content) {
    agentCreatedHtmlFiles.push(data.content);
    console.log('[Computer] Agent created HTML file:', data.content);
    // Append a view card to the answer area.
    const el = $("#answer-content");
    const filePath = data.content;
    const fileName = filePath.split('/').pop();
    const viewUrl = '/api/files/view?path=' + encodeURIComponent(filePath);
    const card = document.createElement('div');
    card.className = 'html-view-card';
    card.style.cssText = 'margin:8px 0;padding:8px 12px;border:1px solid var(--border);border-radius:8px;display:inline-flex;align-items:center;gap:8px;';
    card.innerHTML = '<span style="opacity:.7;">HTML</span> <span>' + esc(fileName) + '</span> ' +
      '<a href="' + viewUrl + '" target="_blank" style="padding:3px 10px;border-radius:4px;background:var(--accent);color:#fff;text-decoration:none;font-size:.85em;">View</a>';
    el.appendChild(card);
  }
  // Show image files inline in the answer area.
  if (data.role === "image" && data.content) {
    const el = $("#answer-content");
    const imgUrl = '/api/media?path=' + encodeURIComponent(data.content);
    const img = document.createElement('img');
    img.src = imgUrl;
    img.alt = 'Generated image';
    img.style.cssText = 'max-width:100%;border-radius:8px;cursor:pointer;display:block;margin:8px 0;';
    img.onclick = function() { window.open(this.src, '_blank'); };
    el.appendChild(img);
  }
  // Show audio files inline in the answer area.
  if (data.role === "audio" && data.content) {
    const el = $("#answer-content");
    const audioUrl = '/api/media?path=' + encodeURIComponent(data.content);
    const wrapper = document.createElement('div');
    wrapper.style.cssText = 'margin:8px 0;';
    wrapper.innerHTML = '<audio controls style="display:block;width:100%;border-radius:8px;">' +
      '<source src="' + audioUrl + '">Your browser does not support audio.</audio>';
    el.appendChild(wrapper);
  }
}

function handleMonitor(data) {
  const toolName = data.tool_name || "unknown";
  const output = data.output || "";
  const truncated = output.length > 200 ? output.slice(0, 200) + "..." : output;
  logEntry("tool", `${toolName}: ${truncated}`, toolName);
  blinkLed();
}

function handleStatus(data) {
  const status = data.status || "";
  logEntry("status", status);

  if (status.toLowerCase().includes("thinking") || status.toLowerCase().includes("running")) {
    setProcessing(true);
  }
}

function handleThinking(data) {
  const text = data.text || "";
  const tool = data.tool || "";
  const phase = data.phase || "";

  if (phase === "done") {
    return;
  }

  if (tool) {
    logEntry("thinking", `[${tool}] ${text}`);
  } else if (text) {
    logEntry("thinking", text);
  }
  blinkLed();
}

function handleToolStream(data) {
  const chunk = data.chunk || data.text || "";
  if (chunk.trim()) {
    logEntry("stream", chunk);
  }
}

function handleResponseStream(data) {
  const text = data.text || "";
  if (text && streamPanelActive && streamPanelSource === "answer") {
    appendStreamText(text);
  }
}

/* ── Input handling ──────────────────────────────────────────── */

function handleStop() {
  if (!connected) return;
  send({ type: "cancel" });
  logEntry("system", "Stop requested");
  // Brief visual feedback on the button.
  var btn = $("#stop-btn");
  if (btn) {
    btn.classList.add("stop-active");
    setTimeout(function() { btn.classList.remove("stop-active"); }, 500);
  }
}

function handleSend() {
  const input = $("#input-box");
  const text = input.value.trim();
  if (!text && !pendingImages.length && !pendingFiles.length) return;
  if (!connected) return;

  // ── /btw command: inject additional instructions while task is running ──
  const btwMatch = text.match(/^\/btw\s+([\s\S]+)/i) || text.match(/^btw\s+([\s\S]+)/i);
  if (btwMatch && isProcessing) {
    const btwText = btwMatch[1].trim();
    if (btwText) {
      send({ type: "btw", content: btwText });
      logEntry("btw", btwText);
      input.value = "";
      input.style.height = "auto";
    }
    return;
  }

  // If the explore inline button is visible, treat this as an exploration send.
  const exploreBtn = $("#explore-send-btn");
  if (exploreBtn && exploreBtn.style.display !== "none") {
    const topic = exploreBtn._exploreTopic || "";
    pendingExploration = {
      parentId: currentNodeId,
      edgeLabel: topic,
      source: 'click',
    };
    _hideExploreButtons();
  }

  input.value = "";
  input.style.height = "auto";

  // Store prompt for Visual tab.
  lastPrompt = text;
  lastResult = "";

  // Clear tracked HTML files from previous turn.
  agentCreatedHtmlFiles = [];

  // Show user input in log.
  logEntry("user", text || "(attachment)");

  // Clear previous answer for fresh response.
  clearAnswer();
  clearVisual();

  // Render input decomposition.
  if (text) renderInputDecomposition(text);

  // ── History branching: rewind session context if forking from an old node ──
  let rewindTimestamp = null;
  if (historyBranchNodeId) {
    const branchNode = explorationNodes.get(historyBranchNodeId);
    if (branchNode) {
      rewindTimestamp = branchNode.created_at;
      // Prune exploration nodes newer than the branch point.
      const branchTime = branchNode.created_at || '';
      const toDelete = [];
      for (const [id, n] of explorationNodes) {
        if ((n.created_at || '') > branchTime) toDelete.push(id);
      }
      for (const id of toDelete) {
        explorationNodes.delete(id);
        // Fire-and-forget backend cleanup.
        fetch(`/api/computer/exploration/${id}`, { method: 'DELETE' }).catch(() => {});
      }
      currentNodeId = historyBranchNodeId;
      logEntry("system", `Forking from: ${branchNode.prompt.slice(0, 50)}${branchNode.prompt.length > 50 ? '…' : ''}`);
    }
    historyBranchNodeId = null;
  }

  // Send to agent.
  if (text.startsWith("/")) {
    send({ type: "command", command: text });
  } else {
    const msg = { type: "chat", content: text || "" };
    // Multi-file: send arrays; single-file: send scalar for backward compat.
    if (pendingImages.length === 1) msg.image_path = pendingImages[0].path;
    else if (pendingImages.length > 1) msg.image_paths = pendingImages.map(i => i.path);
    if (pendingFiles.length === 1) msg.file_path = pendingFiles[0].path;
    else if (pendingFiles.length > 1) msg.file_paths = pendingFiles.map(f => f.path);
    if (rewindTimestamp) msg.rewind_to = rewindTimestamp;
    send(msg);
  }

  clearAttachment();
  setProcessing(true);

  // Open stream panel to show agent thinking during answer generation.
  openStreamPanel("answer");

  // On mobile, flag that we're streaming so CSS can compact the layout.
  if (_isMobile()) {
    document.body.classList.add("mobile-streaming");
  }
}

/* ── Input decomposition ─────────────────────────────────────── */

function renderInputDecomposition(text) {
  const container = $("#decomposition-content");

  const words = text.split(/\s+/);
  const isQuestion = text.includes("?");
  const hasAction = /\b(create|build|write|fix|update|add|remove|delete|analyze|find|search|list|show|explain|summarize|review|refactor)\b/i.test(text);

  let intent = "query";
  if (hasAction) intent = "action";
  if (text.startsWith("/")) intent = "command";

  const actionPattern = /\b(create|build|write|fix|update|add|remove|delete|analyze|find|search|list|show|explain|summarize|review|refactor)\s+(.+?)(?:\s+(?:and|then|,)\s+|$)/gi;
  const steps = [];
  let match;
  while ((match = actionPattern.exec(text)) !== null) {
    steps.push({ action: match[1], target: match[2].trim() });
  }

  if (steps.length === 0) {
    steps.push({ action: intent, target: text });
  }

  const stepsHtml = steps.map((s) => `
    <div class="decomp-step">
      <span class="decomp-bullet">◆</span>
      <span class="decomp-action">${esc(s.action)}</span>
      <span class="decomp-target">→ ${esc(s.target)}</span>
    </div>
  `).join("");

  const complexity = words.length > 20 ? "complex" : words.length > 8 ? "moderate" : "simple";

  container.innerHTML = `
    <div class="decomp-node">
      <div class="decomp-intent">${esc(intent)}</div>
      ${stepsHtml}
      <div class="decomp-meta">
        Complexity: ${complexity} · Words: ${words.length}${isQuestion ? " · Question" : ""}
      </div>
    </div>
  `;
  container.classList.remove("decomp-empty");
}

/* ── Answer rendering ────────────────────────────────────────── */

function clearAnswer() {
  removeNextSteps();
  const el = $("#answer-content");
  el.innerHTML = '<div class="output-processing"><span class="processing-dots">⣾</span> Processing...</div>';
  el.classList.remove("output-empty");
}

function clearVisual() {
  const el = $("#visual-content");
  el.innerHTML = '<span class="output-placeholder">Visual rendering will appear here...</span>';
  el.classList.add("output-empty");
  const frame = $("#visual-frame");
  if (frame) {
    frame.style.display = "none";
    frame.srcdoc = "";
  }
  // Hide visual choice bar if present.
  const bar = $("#visual-choice-bar");
  if (bar) bar.style.display = "none";
}

function renderAnswer(content, isReplay, timestamp) {
  const el = $("#answer-content");
  el.innerHTML = markdownToHtml(content);
  el.classList.remove("output-empty");

  // Store for Visual tab.
  lastResult = content;
  if (timestamp) lastAnswerTimestamp = timestamp;

  // Add answer action bar (copy + feedback).
  _renderAnswerActions(el, content);

  // Auto-generate blueprint from the answer.
  generateBlueprint(content);

  // Create exploration node (skip for replayed history).
  if (!isReplay && lastPrompt) {
    const node = createExplorationNode(lastPrompt, content);
    currentNodeId = node.id;
    saveExplorationNode(node);
    pendingExploration = null;

    // Update map & history drawer.
    renderMap();
    renderHistoryDrawer();
  }

  // Auto-trigger Visual generation if the Visual tab is active.
  const visualTabActive = $(".tab-btn[data-tab='visual']").classList.contains("active");
  if (visualTabActive) {
    if (agentCreatedHtmlFiles.length > 0) {
      // Agent already created an HTML file — ask user what to do.
      showVisualChoiceBar(agentCreatedHtmlFiles);
    } else {
      generateVisual();
    }
  }
}

function _renderAnswerActions(container, content) {
  // Remove existing bar if any.
  const old = container.parentElement.querySelector(".answer-actions");
  if (old) old.remove();

  const bar = document.createElement("div");
  bar.className = "answer-actions";

  // Copy button.
  const copyBtn = document.createElement("button");
  copyBtn.className = "answer-action-btn";
  copyBtn.title = "Copy to clipboard";
  copyBtn.innerHTML = "&#x1F4CB;";
  copyBtn.addEventListener("click", function () {
    const text = container.innerText || container.textContent || "";
    navigator.clipboard.writeText(text).then(function () {
      copyBtn.innerHTML = "&#x2705;";
      setTimeout(function () { copyBtn.innerHTML = "&#x1F4CB;"; }, 1500);
    });
  });
  bar.appendChild(copyBtn);

  // Like button.
  const likeBtn = document.createElement("button");
  likeBtn.className = "answer-action-btn";
  likeBtn.title = "Good response";
  likeBtn.innerHTML = "&#x1F44D;";

  // Dislike button.
  const dislikeBtn = document.createElement("button");
  dislikeBtn.className = "answer-action-btn";
  dislikeBtn.title = "Bad response";
  dislikeBtn.innerHTML = "&#x1F44E;";

  // Saved flash.
  const saved = document.createElement("span");
  saved.className = "answer-fb-saved";
  saved.textContent = "Saved";

  let currentFb = "";

  function sendFb(value) {
    const newVal = currentFb === value ? "" : value;
    currentFb = newVal;
    likeBtn.classList.toggle("active", newVal === "good");
    dislikeBtn.classList.toggle("active", newVal === "bad");
    if (ws && ws.readyState === WebSocket.OPEN && lastAnswerTimestamp) {
      ws.send(JSON.stringify({
        type: "message_feedback",
        timestamp: lastAnswerTimestamp,
        feedback: newVal || null,
      }));
    }
    saved.classList.remove("show");
    void saved.offsetWidth;
    saved.classList.add("show");
  }

  likeBtn.addEventListener("click", function () { sendFb("good"); });
  dislikeBtn.addEventListener("click", function () { sendFb("bad"); });

  // Brain Graph button.
  const graphBtn = document.createElement("button");
  graphBtn.className = "answer-action-btn";
  graphBtn.title = "View in Brain Graph";
  graphBtn.innerHTML = "&#x1F9E0;";
  graphBtn.addEventListener("click", function () {
    const params = "?focus_ts=" + encodeURIComponent(lastAnswerTimestamp || "");
    window.open("/brain-graph" + params, "_blank");
  });

  bar.appendChild(likeBtn);
  bar.appendChild(dislikeBtn);
  bar.appendChild(graphBtn);
  bar.appendChild(saved);

  container.parentElement.appendChild(bar);
}

/* ── Next-step suggestion buttons ────────────────────────────── */

function removeNextSteps() {
  document.querySelectorAll(".next-steps-bar").forEach(el => el.remove());
}

function renderNextSteps(options) {
  if (!options || !options.length) return;
  removeNextSteps();

  const wrapper = document.getElementById("output-answer");
  if (!wrapper) return;

  const bar = document.createElement("div");
  bar.className = "next-steps-bar";

  const label = document.createElement("div");
  label.className = "next-steps-label";
  label.textContent = "Suggested next steps";
  bar.appendChild(label);

  const row = document.createElement("div");
  row.className = "next-steps-row";

  options.forEach(function (opt) {
    const btn = document.createElement("button");
    btn.className = "next-step-btn";
    btn.title = opt.description || opt.action;
    btn.textContent = opt.label;
    btn.addEventListener("click", function () {
      removeNextSteps();
      // Put the action into the input and trigger send so the full
      // handleSend flow runs (stream panel, exploration node, etc.).
      const input = $("#input-box");
      input.value = opt.action;
      handleSend();
    });
    row.appendChild(btn);
  });

  bar.appendChild(row);
  wrapper.appendChild(bar);
}

/* ── Blueprint tab (structural decomposition) ────────────────── */

function generateBlueprint(content) {
  const el = $("#blueprint-content");

  const sections = [];
  const lines = content.split("\n");
  let currentSection = { type: "text", title: "Response", lines: [] };

  for (const line of lines) {
    const trimmed = line.trim();
    if (/^#{1,3}\s+/.test(trimmed)) {
      if (currentSection.lines.length > 0) {
        sections.push(currentSection);
      }
      currentSection = {
        type: "section",
        title: trimmed.replace(/^#+\s*/, ""),
        lines: [],
      };
    } else if (/^```/.test(trimmed)) {
      if (currentSection.lines.length > 0) {
        sections.push(currentSection);
      }
      currentSection = {
        type: "code",
        title: "Code",
        lines: [],
      };
    } else if (/^[-*]\s+/.test(trimmed)) {
      currentSection.type = "list";
      currentSection.lines.push(trimmed.replace(/^[-*]\s+/, ""));
    } else if (trimmed) {
      currentSection.lines.push(trimmed);
    }
  }
  if (currentSection.lines.length > 0) {
    sections.push(currentSection);
  }

  if (sections.length === 0) {
    el.innerHTML = '<span class="output-placeholder">No structure to decompose</span>';
    el.classList.add("output-empty");
    return;
  }

  const typeIcons = { text: "📄", section: "📑", code: "💻", list: "📋" };

  el.innerHTML = sections.map((s, i) => {
    const icon = typeIcons[s.type] || "📄";
    const bodyHtml = s.lines.map((l) => `<div>${esc(l)}</div>`).join("");
    return `
      <div class="blueprint-section">
        <div class="blueprint-header" onclick="this.parentElement.classList.toggle('collapsed')">
          ${icon} ${esc(s.title || `Section ${i + 1}`)}
          <span class="blueprint-label">${s.type}</span>
        </div>
        <div class="blueprint-body">${bodyHtml}</div>
      </div>
    `;
  }).join(`
    <div class="blueprint-step">
      <span class="blueprint-arrow">↓</span>
    </div>
  `);

  el.classList.remove("output-empty");
}

/* ── Visual tab (LLM-generated HTML) ─────────────────────────── */

/* Strip <script>…</script> tags from a chunk so partial JS never executes
   during the progressive render.  Handles partial tags across chunks via
   the _scriptStrip* state variables. */
let _scriptStripInTag = false;  // true when we're inside a <script> block

function _stripScriptsProgressive(chunk) {
  let out = "";
  let i = 0;
  while (i < chunk.length) {
    if (_scriptStripInTag) {
      // Look for </script>
      const closeIdx = chunk.toLowerCase().indexOf("</script", i);
      if (closeIdx === -1) {
        // Entire remainder is inside <script> — discard.
        break;
      }
      // Skip past the closing tag.
      const gtIdx = chunk.indexOf(">", closeIdx + 8);
      if (gtIdx === -1) break; // incomplete closing tag — wait for next chunk
      i = gtIdx + 1;
      _scriptStripInTag = false;
    } else {
      const openIdx = chunk.toLowerCase().indexOf("<script", i);
      if (openIdx === -1) {
        out += chunk.slice(i);
        break;
      }
      out += chunk.slice(i, openIdx);
      // Check if tag is self-closing or has a closing >
      const gtIdx = chunk.indexOf(">", openIdx + 7);
      if (gtIdx === -1) {
        // Incomplete open tag — assume script starts, wait for next chunk.
        _scriptStripInTag = true;
        break;
      }
      _scriptStripInTag = true;
      i = gtIdx + 1;
    }
  }
  return out;
}

async function generateVisual() {
  if (!lastPrompt || !lastResult) return;
  if (visualGenerating) return;

  visualGenerating = true;
  const el = $("#visual-content");
  const frame = $("#visual-frame");

  el.innerHTML = "";
  el.classList.remove("output-empty");

  // Show iframe immediately for progressive rendering.
  frame.style.display = "block";
  // Reset to about:blank so contentDocument is immediately writable.
  frame.removeAttribute("srcdoc");
  frame.src = "about:blank";

  logEntry("system", "Generating visual rendering...");
  blinkLed();

  // Open stream panel to show live LLM output.
  openStreamPanel("visual");

  const payload = {
    prompt: lastPrompt,
    result: lastResult,
    theme: currentTheme,
    theme_instructions: THEME_INSTRUCTIONS[currentTheme] || THEME_INSTRUCTIONS.modern,
    token_tier: selectedTokenTier,
    model: selectedModel || "",
  };

  const t0 = performance.now();
  console.group("%c[Computer] Visual Generation (streaming)", "color:#6c8cff;font-weight:bold");
  console.log("Theme:", currentTheme);
  console.log("Tier:", selectedTokenTier, "| Model:", selectedModel || "(default)");
  console.log("Prompt:", lastPrompt.length > 120 ? lastPrompt.slice(0, 120) + "…" : lastPrompt);
  console.log("Result length:", lastResult.length, "chars");

  // Reset progressive script-strip state.
  _scriptStripInTag = false;
  let liveDocReady = false;
  let liveDoc = null;

  function _initLiveDoc() {
    if (liveDocReady) return;
    try {
      liveDoc = frame.contentDocument || frame.contentWindow.document;
      liveDoc.open();
      liveDocReady = true;
    } catch (e) {
      console.warn("[Computer] Cannot access iframe document for progressive render:", e);
    }
  }

  function _writeLiveChunk(chunk) {
    const safe = _stripScriptsProgressive(chunk);
    if (!safe) return;
    _initLiveDoc();
    if (liveDoc) {
      try {
        liveDoc.write(safe);
        // Scroll iframe to bottom so newly rendered content is visible.
        const win = frame.contentWindow;
        if (win) win.scrollTo(0, liveDoc.body ? liveDoc.body.scrollHeight : 0);
      } catch (_) { /* ignore write errors */ }
    }
  }

  try {
    const res = await fetch("/api/computer/visualize/stream", {
      method: "POST",
      headers: { "Content-Type": "application/json" },
      body: JSON.stringify(payload),
    });

    if (!res.ok) {
      const errBody = await res.text();
      console.error("Response:", res.status, errBody);
      console.groupEnd();
      throw new Error(`HTTP ${res.status}`);
    }

    // Read SSE stream.
    const reader = res.body.getReader();
    const decoder = new TextDecoder();
    let html = "";
    let sseBuffer = "";

    while (true) {
      const { done, value } = await reader.read();
      if (done) break;

      sseBuffer += decoder.decode(value, { stream: true });

      // Parse SSE lines.
      const lines = sseBuffer.split("\n");
      sseBuffer = lines.pop(); // keep incomplete line in buffer

      for (const line of lines) {
        if (!line.startsWith("data: ")) continue;
        try {
          const evt = JSON.parse(line.slice(6));
          if (evt.type === "start") {
            appendStreamText("● Streaming from " + (evt.model || "LLM") + "...\n\n");
          } else if (evt.type === "chunk") {
            appendStreamText(evt.text);
            _writeLiveChunk(evt.text);
            blinkLed();
          } else if (evt.type === "done") {
            html = evt.html || "";
          } else if (evt.type === "error") {
            throw new Error(evt.error || "Stream error");
          }
        } catch (parseErr) {
          if (parseErr.message && !parseErr.message.includes("JSON")) throw parseErr;
        }
      }
    }

    // Close the progressive document.
    if (liveDoc) {
      try { liveDoc.close(); } catch (_) {}
    }

    const elapsed = Math.round(performance.now() - t0);
    console.log("HTML size:", html.length, "chars");
    console.log("Elapsed:", elapsed, "ms");
    console.groupEnd();

    finishStreamPanel(`Visual generation complete (${humanTime(elapsed)}, ${humanSize(html.length)})`);
    // Auto-close stream panel after visual generation completes.
    setTimeout(() => closeStreamPanel(), 600);

    if (!html) {
      el.innerHTML = '<span class="output-placeholder">No visual generated</span>';
      el.classList.add("output-empty");
      frame.style.display = "none";
      return;
    }

    // Count explore links for logging.
    const linkCount = (html.match(/class="explore-link"/g) || []).length;
    if (linkCount > 0) {
      console.log(`[Computer] Found ${linkCount} explore-link elements`);
    }

    // Final render: inject explore bridge and scripts, replace progressive content.
    const enrichedHtml = injectExploreBridge(html);
    frame.srcdoc = enrichedHtml;

    // Store visual HTML on the current exploration node.
    if (currentNodeId && explorationNodes.has(currentNodeId)) {
      const node = explorationNodes.get(currentNodeId);
      node.visual_html = html;
      updateExplorationNodeVisual(currentNodeId, html);
    }

    logEntry("system", `Visual rendering complete (${humanTime(elapsed)}, ${humanSize(html.length)}${linkCount > 0 ? `, ${linkCount} explore links` : ''})`);

  } catch (err) {
    console.error("Visual generation failed:", err);
    console.groupEnd();
    if (liveDoc) { try { liveDoc.close(); } catch (_) {} }
    el.innerHTML = `<span class="output-placeholder">Visual generation failed: ${esc(err.message)}</span>`;
    el.classList.add("output-empty");
    frame.style.display = "none";
    logEntry("error", `Visual generation failed: ${err.message}`);
    finishStreamPanel(`Failed: ${err.message}`);
    setTimeout(() => closeStreamPanel(), 600);
  } finally {
    visualGenerating = false;
  }
}

/* ── Visual choice bar (agent HTML vs generate new) ──────────── */

function showVisualChoiceBar(htmlFiles) {
  const el = $("#visual-content");
  const frame = $("#visual-frame");

  // Build file list display.
  const fileNames = htmlFiles.map(f => f.split('/').pop()).join(', ');

  let bar = $("#visual-choice-bar");
  if (!bar) {
    bar = document.createElement("div");
    bar.id = "visual-choice-bar";
    el.parentNode.insertBefore(bar, el);
  }

  bar.innerHTML = `
    <div class="visual-choice-info">
      <span class="visual-choice-icon">🌐</span>
      <span class="visual-choice-text">Agent created <strong>${esc(fileNames)}</strong> — use it as visual or generate a new themed rendering?</span>
    </div>
    <div class="visual-choice-actions">
      <button id="visual-use-file-btn" class="tab-btn" title="Use agent's HTML file">📄 Use File</button>
      <button id="visual-generate-btn" class="tab-btn" title="Generate new themed visual">🎨 Generate New</button>
    </div>
  `;
  bar.style.display = "flex";

  // Clear the visual area placeholder.
  el.innerHTML = '';
  el.classList.remove("output-empty");
  frame.style.display = "none";

  // Wire buttons.
  bar.querySelector("#visual-use-file-btn").addEventListener("click", async () => {
    bar.style.display = "none";
    await useAgentHtmlFile(htmlFiles[htmlFiles.length - 1]);
  });

  bar.querySelector("#visual-generate-btn").addEventListener("click", () => {
    bar.style.display = "none";
    generateVisual();
  });

  logEntry("system", `Agent created HTML file — choose: use it or generate new visual`);
}

async function useAgentHtmlFile(filePath) {
  const el = $("#visual-content");
  const frame = $("#visual-frame");
  const fileName = filePath.split('/').pop();

  el.innerHTML = '<div class="output-processing"><span class="processing-dots">⣾</span> Loading agent HTML...</div>';
  el.classList.remove("output-empty");

  try {
    // The path from the backend broadcast is the physical path — use it directly.
    const contentRes = await fetch(`/api/files/content?path=${encodeURIComponent(filePath)}`);
    if (!contentRes.ok) {
      // Fall back to view endpoint in iframe.
      el.innerHTML = "";
      frame.src = `/api/files/view?path=${encodeURIComponent(filePath)}`;
      frame.style.display = "block";
      logEntry("system", `Loaded agent HTML file: ${fileName}`);
      return;
    }

    const data = await contentRes.json();
    const html = data.content || "";

    if (!html) {
      throw new Error("Empty HTML file");
    }

    // Inject the explore bridge and render in the iframe.
    const enrichedHtml = injectExploreBridge(html);
    el.innerHTML = "";
    el.classList.remove("output-empty");
    frame.srcdoc = enrichedHtml;
    frame.style.display = "block";

    // Store visual HTML on the current exploration node.
    if (currentNodeId && explorationNodes.has(currentNodeId)) {
      const node = explorationNodes.get(currentNodeId);
      node.visual_html = html;
      updateExplorationNodeVisual(currentNodeId, html);
    }

    logEntry("system", `Using agent HTML file: ${fileName} (${humanSize(html.length)}) — no extra LLM call`);

  } catch (err) {
    console.warn("Failed to load agent HTML, falling back to generate:", err);
    logEntry("system", `Failed to load agent HTML (${err.message}), generating new visual...`);
    generateVisual();
  }
}

/* ── Fullscreen toggle ───────────────────────────────────────── */

let isFullscreen = false;

function toggleFullscreen() {
  const panel = $("#output-panel");
  isFullscreen = !isFullscreen;
  panel.classList.toggle("fullscreen", isFullscreen);
  document.body.classList.toggle("visual-fullscreen", isFullscreen);
  $("#fullscreen-btn").textContent = isFullscreen ? "⛶" : "⛶";
  $("#fullscreen-btn").title = isFullscreen ? "Exit fullscreen" : "Fullscreen";
}

async function exportVisualPdf() {
  const frame = $("#visual-frame");
  const html = frame.srcdoc || "";
  if (!html) {
    logEntry("error", "No visual to export — generate one first");
    return;
  }

  const pdfBtn = $("#export-pdf-btn");
  pdfBtn.disabled = true;
  pdfBtn.textContent = "...";
  logEntry("system", "Exporting visual to PDF...");
  blinkLed();

  try {
    const res = await fetch("/api/computer/export-pdf", {
      method: "POST",
      headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ html, prompt: lastPrompt || "" }),
    });

    if (!res.ok) {
      const err = await res.json().catch(() => ({}));
      throw new Error(err.error || `HTTP ${res.status}`);
    }

    // Extract filename from Content-Disposition header.
    const cd = res.headers.get("Content-Disposition") || "";
    const fnMatch = cd.match(/filename="([^"]+)"/);
    const filename = fnMatch ? fnMatch[1] : "visual.pdf";

    const blob = await res.blob();
    const url = URL.createObjectURL(blob);
    const a = document.createElement("a");
    a.href = url;
    a.download = filename;
    document.body.appendChild(a);
    a.click();
    document.body.removeChild(a);
    URL.revokeObjectURL(url);

    logEntry("system", `PDF exported (${humanSize(blob.size)})`);
  } catch (err) {
    logEntry("error", `PDF export failed: ${err.message}`);
  } finally {
    pdfBtn.disabled = false;
    pdfBtn.textContent = "PDF";
  }
}

function initFullscreen() {
  const btn = $("#fullscreen-btn");
  btn.addEventListener("click", toggleFullscreen);

  // Re-run visual button.
  const rerunBtn = $("#rerun-visual-btn");
  rerunBtn.addEventListener("click", () => {
    if (!lastPrompt || !lastResult) {
      logEntry("system", "Nothing to re-run — no prompt/result yet");
      return;
    }
    // Force re-generation by clearing the iframe first.
    const frame = $("#visual-frame");
    frame.srcdoc = "";
    frame.style.display = "none";
    visualGenerating = false;
    generateVisual();
  });

  // Export PDF button.
  const pdfBtn = $("#export-pdf-btn");
  pdfBtn.addEventListener("click", exportVisualPdf);

  // Escape key exits fullscreen.
  document.addEventListener("keydown", (e) => {
    if (e.key === "Escape" && isFullscreen) {
      toggleFullscreen();
    }
  });
}

/* ── Activity log ────────────────────────────────────────────── */

const MAX_LOG_ENTRIES = 200;
let _errorSoundPlayed = false; // true = suppress until reset on reconnect

function logEntry(type, text, toolName) {
  const container = $("#log-content");
  const now = new Date();
  const ts = now.toLocaleTimeString("en-GB", { hour: "2-digit", minute: "2-digit", second: "2-digit" });

  const icons = {
    system: "●",
    user: "▶",
    btw: "💡",
    status: "◎",
    tool: "⚙",
    thinking: "◇",
    stream: "│",
    error: "✖",
  };

  const entry = document.createElement("div");
  entry.className = `log-entry`;

  const icon = icons[type] || "·";
  const textClass = `log-${type}`;

  entry.innerHTML = `
    <span class="log-ts">${ts}</span>
    <span class="log-icon">${icon}</span>
    <span class="log-text ${textClass}">${esc(text)}</span>
  `;

  container.appendChild(entry);
  // Play notification sounds: warning for blocked/denied, error for other errors, activity for the rest.
  // Error/warning sounds play once, then suppress until reconnect resets the flag.
  if (type === "error") {
    if (!_errorSoundPlayed) {
      _errorSoundPlayed = true;
      const lower = (text || "").toLowerCase();
      if (lower.includes("blocked") || lower.includes("forbidden") || lower.includes("denied")
          || lower.includes("not allowed") || lower.includes("duplicate")) {
        playWarningSound();
      } else {
        playErrorSound();
      }
    }
  } else {
    playActivitySound(type);
  }

  while (container.children.length > MAX_LOG_ENTRIES) {
    container.removeChild(container.firstChild);
  }

  container.scrollTop = container.scrollHeight;
}

/* ── Drive LED ───────────────────────────────────────────────── */

function blinkLed() {
  const led = $("#drive-led");
  led.classList.add("on");
  clearTimeout(ledTimer);
  ledTimer = setTimeout(() => led.classList.remove("on"), 150);
}

function setProcessing(active) {
  const wasProcessing = isProcessing;
  isProcessing = active;
  const led = $("#drive-led");
  const sendBtn = $("#send-btn");
  const stopBtn = $("#stop-btn");
  if (active) {
    led.classList.add("on");
    if (sendBtn) sendBtn.style.display = "none";
    if (stopBtn) stopBtn.style.display = "";
    _acquireWakeLock();
  } else {
    led.classList.remove("on");
    if (sendBtn) sendBtn.style.display = "";
    if (stopBtn) stopBtn.style.display = "none";
    _releaseWakeLock();
    // Play completion sound when a task finishes.
    if (wasProcessing) playCompletionSound();
  }
}

/* ── Screen Wake Lock (keep mobile screen on during tasks) ──── */

let _wakeLock = null;

async function _acquireWakeLock() {
  if (_wakeLock) return; // already held
  if (!("wakeLock" in navigator)) return; // not supported
  try {
    _wakeLock = await navigator.wakeLock.request("screen");
    _wakeLock.addEventListener("release", () => { _wakeLock = null; });
  } catch (_) {}
}

function _releaseWakeLock() {
  if (_wakeLock) {
    _wakeLock.release().catch(() => {});
    _wakeLock = null;
  }
}

// Re-acquire wake lock when page becomes visible again (e.g. tab switch back)
// while a task is still processing.
document.addEventListener("visibilitychange", () => {
  if (document.visibilityState === "visible" && isProcessing) {
    _acquireWakeLock();
  }
});

/* ── Panel resizing ──────────────────────────────────────────── */

function initResize() {
  const handle = $("#h-resize");
  const left = $("#left-column");
  const workspace = $("#workspace");
  let dragging = false;

  // Restore persisted panel width.
  const saved = localStorage.getItem("panel-width");
  if (saved) {
    const pct = parseFloat(saved);
    if (pct >= 25 && pct <= 70) left.style.width = pct + "%";
  }

  handle.addEventListener("mousedown", (e) => {
    e.preventDefault();
    dragging = true;
    document.body.style.cursor = "col-resize";
    document.body.style.userSelect = "none";
  });

  document.addEventListener("mousemove", (e) => {
    if (!dragging) return;
    const rect = workspace.getBoundingClientRect();
    const pct = ((e.clientX - rect.left) / rect.width) * 100;
    const clamped = Math.max(25, Math.min(70, pct));
    left.style.width = clamped + "%";
  });

  document.addEventListener("mouseup", () => {
    if (dragging) {
      dragging = false;
      document.body.style.cursor = "";
      document.body.style.userSelect = "";
      // Persist the panel width.
      const pct = (left.getBoundingClientRect().width / workspace.getBoundingClientRect().width) * 100;
      localStorage.setItem("panel-width", pct.toFixed(2));
      // Re-align the stream panel after resize.
      if (!$("#stream-panel").classList.contains("hidden")) _alignStreamPanel();
    }
  });

  window.addEventListener("resize", () => {
    if (!$("#stream-panel").classList.contains("hidden")) _alignStreamPanel();
  });
}

/* ── Tab switching ───────────────────────────────────────────── */

function initTabs() {
  for (const btn of $$("#output-tabs .tab-btn")) {
    btn.addEventListener("click", () => {
      const tab = btn.dataset.tab;

      // Update buttons.
      for (const b of $$("#output-tabs .tab-btn")) {
        b.classList.toggle("active", b === btn);
      }

      // Update panels.
      for (const panel of $$(".output-tab")) {
        panel.classList.toggle("active", panel.id === `output-${tab}`);
      }

      // Show/hide Visual-only buttons.
      const fsBtn = $("#fullscreen-btn");
      const rerunBtn = $("#rerun-visual-btn");
      const pdfBtn = $("#export-pdf-btn");
      if (tab === "visual") {
        fsBtn.style.display = "inline-block";
        rerunBtn.style.display = "inline-block";
        pdfBtn.style.display = "inline-block";
        // Auto-generate visual if we have content and haven't generated yet.
        const frame = $("#visual-frame");
        if (lastResult && (!frame.srcdoc || frame.srcdoc === "")) {
          if (agentCreatedHtmlFiles.length > 0) {
            showVisualChoiceBar(agentCreatedHtmlFiles);
          } else {
            generateVisual();
          }
        }
      } else {
        fsBtn.style.display = "none";
        rerunBtn.style.display = "none";
        pdfBtn.style.display = "none";
        // Exit fullscreen if leaving visual tab.
        if (isFullscreen) {
          toggleFullscreen();
        }
      }

      // Render map when switching to Map tab.
      if (tab === "map") {
        renderMap();
      }

      // Load files when switching to Files tab.
      if (tab === "files") {
        loadFiles();
      }

      // Load todos when switching to Todos tab.
      if (tab === "todos") {
        loadTodos();
      }

      // Load data tables when switching to Data tab.
      if (tab === "data") {
        loadDataTables();
      }

      // Load insights when switching to Insights tab.
      if (tab === "insights") {
        loadInsights();
      }
    });
  }
}

/* ═══════════════════════════════════════════════════════════════
   TODOS TAB
   ═══════════════════════════════════════════════════════════════ */

let _todosCache = [];
let _todosFilter = "active"; // "active" | "all" | "done"
let _todosQuery = "";

function _todoRelTime(iso) {
  if (!iso) return "";
  const d = new Date(iso);
  const now = Date.now();
  const diff = now - d.getTime();
  if (diff < 60000) return "just now";
  if (diff < 3600000) return `${Math.floor(diff / 60000)}m ago`;
  if (diff < 86400000) return `${Math.floor(diff / 3600000)}h ago`;
  if (diff < 604800000) return `${Math.floor(diff / 86400000)}d ago`;
  return d.toLocaleDateString();
}

function _renderTodoCard(t) {
  const isDone = t.status === "done" || t.status === "cancelled";
  const checked = isDone ? "checked" : "";
  const doneClass = isDone ? " todo-done" : "";
  const priorityCls = (t.priority === "urgent" || t.priority === "high") ? ` ${t.priority}` : "";
  const tags = t.tags ? `<span class="todo-badge todo-badge-tags">${esc(t.tags)}</span>` : "";
  const time = _todoRelTime(t.updated_at || t.created_at);

  return `
    <div class="todo-card${doneClass}" data-id="${esc(t.id)}">
      <input type="checkbox" class="todo-checkbox" ${checked} title="${isDone ? 'Reopen' : 'Mark done'}">
      <div class="todo-body">
        <div class="todo-text">${esc(t.content)}</div>
        <div class="todo-meta">
          <span class="todo-badge todo-badge-priority${priorityCls}">${esc(t.priority || 'normal')}</span>
          <span class="todo-badge todo-badge-responsible">${esc(t.responsible || 'bot')}</span>
          ${tags}
          <span class="todo-time">${esc(time)}</span>
        </div>
      </div>
      <div class="todo-actions">
        <button class="todo-action-btn todo-delete-btn" title="Delete">✕</button>
      </div>
    </div>
  `;
}

function _renderTodos(todos) {
  const el = $("#todos-content");
  const countEl = $("#todos-count");

  if (!todos || todos.length === 0) {
    el.innerHTML = '<span class="output-placeholder">No to-do items.</span>';
    el.classList.add("output-empty");
    countEl.textContent = "";
    return;
  }

  el.classList.remove("output-empty");
  countEl.textContent = `${todos.length} item${todos.length !== 1 ? "s" : ""}`;
  el.innerHTML = todos.map(_renderTodoCard).join("");

  // Wire up checkbox toggles.
  for (const cb of el.querySelectorAll(".todo-checkbox")) {
    cb.addEventListener("change", async () => {
      const card = cb.closest(".todo-card");
      const id = card.dataset.id;
      const newStatus = cb.checked ? "done" : "pending";
      try {
        await fetch(`/api/todos/${encodeURIComponent(id)}`, {
          method: "PATCH",
          headers: { "Content-Type": "application/json" },
          body: JSON.stringify({ status: newStatus }),
        });
        loadTodos();
      } catch (e) {
        console.error("Todo toggle failed", e);
      }
    });
  }

  // Wire up delete buttons.
  for (const btn of el.querySelectorAll(".todo-delete-btn")) {
    btn.addEventListener("click", async () => {
      const card = btn.closest(".todo-card");
      const id = card.dataset.id;
      try {
        await fetch(`/api/todos/${encodeURIComponent(id)}`, { method: "DELETE" });
        loadTodos();
      } catch (e) {
        console.error("Todo delete failed", e);
      }
    });
  }
}

function _applyTodosFilters() {
  let filtered = _todosCache;
  if (_todosFilter === "active") {
    filtered = filtered.filter(t => t.status === "pending" || t.status === "in_progress");
  } else if (_todosFilter === "done") {
    filtered = filtered.filter(t => t.status === "done" || t.status === "cancelled");
  }
  if (_todosQuery) {
    const q = _todosQuery.toLowerCase();
    filtered = filtered.filter(t =>
      (t.content || "").toLowerCase().includes(q) ||
      (t.tags || "").toLowerCase().includes(q)
    );
  }
  _renderTodos(filtered);
}

async function loadTodos() {
  const el = $("#todos-content");
  try {
    const params = new URLSearchParams();
    if (sessionId) params.set("session_id", sessionId);
    const res = await fetch(`/api/todos?${params}`);
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    _todosCache = await res.json();

    _applyTodosFilters();
  } catch (e) {
    console.error("Failed to load todos", e);
    el.innerHTML = '<span class="output-placeholder">Failed to load todos.</span>';
    el.classList.add("output-empty");
  }
}

async function addTodo(content) {
  if (!content.trim()) return;
  try {
    const body = { content: content.trim(), responsible: "human", priority: "normal" };
    if (sessionId) body.source_session = sessionId;
    await fetch("/api/todos", {
      method: "POST",
      headers: { "Content-Type": "application/json" },
      body: JSON.stringify(body),
    });
    loadTodos();
  } catch (e) {
    console.error("Failed to add todo", e);
  }
}

function initTodoTab() {
  // Filter buttons.
  for (const btn of $$("#todos-filters .todos-filter-btn")) {
    btn.addEventListener("click", () => {
      _todosFilter = btn.dataset.filter;
      for (const b of $$("#todos-filters .todos-filter-btn")) {
        b.classList.toggle("active", b === btn);
      }
      _applyTodosFilters();
    });
  }

  // Search input.
  const searchInput = $("#todos-search");
  searchInput.addEventListener("input", () => {
    _todosQuery = searchInput.value.trim();
    _applyTodosFilters();
  });

  // Refresh button.
  $("#todos-refresh-btn").addEventListener("click", loadTodos);

  // Add input.
  const addInput = $("#todos-add-input");
  const addBtn = $("#todos-add-btn");
  addBtn.addEventListener("click", () => {
    addTodo(addInput.value);
    addInput.value = "";
  });
  addInput.addEventListener("keydown", (e) => {
    if (e.key === "Enter") {
      e.preventDefault();
      addTodo(addInput.value);
      addInput.value = "";
    }
  });
}


/* ═══════════════════════════════════════════════════════════════
   DATA TAB — datastore tables with upload
   ═══════════════════════════════════════════════════════════════ */

let _dsTables = [];
let _dsSelectedTable = null;
let _dsPage = 0;
const _dsPageSize = 50;

function _renderTableList() {
  const el = $("#data-table-list");
  const countEl = $("#data-table-count");
  if (!_dsTables.length) {
    el.innerHTML = '<div style="padding:12px;color:var(--text-dim);font-size:12px;">No tables yet.</div>';
    countEl.textContent = "";
    return;
  }
  countEl.textContent = `${_dsTables.length} table${_dsTables.length !== 1 ? "s" : ""}`;
  el.innerHTML = _dsTables.map(t => {
    const sel = _dsSelectedTable === t.name ? " selected" : "";
    return `<div class="ds-table-item${sel}" data-name="${esc(t.name)}">
      <span class="ds-table-name">${esc(t.name)}</span>
      <span class="ds-table-rows">${t.row_count}</span>
    </div>`;
  }).join("");

  for (const item of el.querySelectorAll(".ds-table-item")) {
    item.addEventListener("click", () => {
      _dsSelectedTable = item.dataset.name;
      _dsPage = 0;
      _renderTableList();
      _loadTableData(_dsSelectedTable);
    });
  }
}

function _showTableView(show) {
  $("#data-view-empty").style.display = show ? "none" : "flex";
  $("#data-view-header").style.display = show ? "flex" : "none";
  $("#data-grid-wrap").style.display = show ? "block" : "none";
  $("#data-paging").style.display = show ? "flex" : "none";
}

async function _loadTableData(name) {
  _showTableView(true);
  const nameEl = $("#data-view-name");
  const metaEl = $("#data-view-meta");
  const thead = $("#data-grid thead");
  const tbody = $("#data-grid tbody");

  nameEl.textContent = name;
  metaEl.textContent = "Loading...";
  thead.innerHTML = "";
  tbody.innerHTML = "";

  try {
    // Fetch rows with pagination.
    const params = new URLSearchParams({
      limit: String(_dsPageSize),
      offset: String(_dsPage * _dsPageSize),
    });
    const res = await fetch(`/api/datastore/tables/${encodeURIComponent(name)}/rows?${params}`);
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    const data = await res.json();

    const columns = data.columns || [];
    const rows = data.rows || [];
    const total = data.total != null ? data.total : rows.length;

    // Find the table info for column count.
    const tInfo = _dsTables.find(t => t.name === name);
    const colCount = tInfo ? tInfo.columns.length : columns.length;
    metaEl.textContent = `${total} row${total !== 1 ? "s" : ""} · ${colCount} col${colCount !== 1 ? "s" : ""}`;

    // Export links.
    $("#data-export-csv").href = `/api/datastore/tables/${encodeURIComponent(name)}/export?format=csv`;
    $("#data-export-csv").download = `${name}.csv`;
    $("#data-export-xlsx").href = `/api/datastore/tables/${encodeURIComponent(name)}/export?format=xlsx`;
    $("#data-export-xlsx").download = `${name}.xlsx`;

    // Render header.
    thead.innerHTML = "<tr>" + columns.map(c => `<th>${esc(c)}</th>`).join("") + "</tr>";

    // Render rows (API returns array of objects keyed by column name).
    if (rows.length === 0) {
      tbody.innerHTML = `<tr><td colspan="${columns.length || 1}" style="text-align:center;color:var(--text-dim);padding:20px;">No rows.</td></tr>`;
    } else {
      tbody.innerHTML = rows.map(row => {
        const cells = columns.map(c => {
          const v = row[c];
          const s = v != null ? String(v) : "NULL";
          return `<td title="${esc(s)}">${esc(s)}</td>`;
        });
        return "<tr>" + cells.join("") + "</tr>";
      }).join("");
    }

    // Paging.
    const start = _dsPage * _dsPageSize + 1;
    const end = Math.min(start + rows.length - 1, total);
    $("#data-paging-info").textContent = rows.length ? `${start}–${end} of ${total}` : `0 of ${total}`;
    $("#data-prev-btn").disabled = _dsPage === 0;
    $("#data-next-btn").disabled = end >= total;

  } catch (e) {
    console.error("Failed to load table data", e);
    metaEl.textContent = "Error loading data";
  }
}

async function loadDataTables() {
  try {
    const res = await fetch("/api/datastore/tables");
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    _dsTables = await res.json();
    _renderTableList();

    // If selected table no longer exists, deselect.
    if (_dsSelectedTable && !_dsTables.find(t => t.name === _dsSelectedTable)) {
      _dsSelectedTable = null;
      _showTableView(false);
    }
    // Refresh data if a table is selected.
    if (_dsSelectedTable) {
      _loadTableData(_dsSelectedTable);
    }
  } catch (e) {
    console.error("Failed to load datastore tables", e);
  }
}

async function _uploadDataFile(file) {
  if (!file) return;
  const ext = file.name.split(".").pop().toLowerCase();
  if (!["csv", "xlsx", "xls"].includes(ext)) {
    alert("Only CSV and XLSX files are supported.");
    return;
  }

  const formData = new FormData();
  formData.append("file", file);
  formData.append("force_new", "true");

  try {
    const res = await fetch("/api/datastore/upload", { method: "POST", body: formData });
    if (!res.ok) {
      const err = await res.json().catch(() => ({}));
      alert(err.error || `Upload failed: HTTP ${res.status}`);
      return;
    }
    const result = await res.json();
    _dsSelectedTable = result.table;
    _dsPage = 0;
    await loadDataTables();
  } catch (e) {
    console.error("Data upload failed", e);
    alert("Upload failed: " + e.message);
  }
}

async function _dropTable(name) {
  if (!confirm(`Drop table "${name}"? This cannot be undone.`)) return;
  try {
    const res = await fetch(`/api/datastore/tables/${encodeURIComponent(name)}`, { method: "DELETE" });
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    _dsSelectedTable = null;
    _showTableView(false);
    await loadDataTables();
  } catch (e) {
    console.error("Drop table failed", e);
  }
}

function initDataTab() {
  // Refresh.
  $("#data-refresh-btn").addEventListener("click", loadDataTables);

  // File upload via button.
  const uploadInput = $("#data-upload-input");
  uploadInput.addEventListener("change", () => {
    if (uploadInput.files.length > 0) {
      _uploadDataFile(uploadInput.files[0]);
      uploadInput.value = "";
    }
  });

  // Drag & drop onto the data panel.
  const panel = $("#output-data");
  panel.addEventListener("dragover", (e) => { e.preventDefault(); panel.classList.add("drag-over"); });
  panel.addEventListener("dragleave", () => panel.classList.remove("drag-over"));
  panel.addEventListener("drop", (e) => {
    e.preventDefault();
    panel.classList.remove("drag-over");
    if (e.dataTransfer.files.length > 0) {
      _uploadDataFile(e.dataTransfer.files[0]);
    }
  });

  // Drop table button.
  $("#data-drop-btn").addEventListener("click", () => {
    if (_dsSelectedTable) _dropTable(_dsSelectedTable);
  });

  // Paging.
  $("#data-prev-btn").addEventListener("click", () => {
    if (_dsPage > 0) { _dsPage--; _loadTableData(_dsSelectedTable); }
  });
  $("#data-next-btn").addEventListener("click", () => {
    _dsPage++;
    _loadTableData(_dsSelectedTable);
  });
}


/* ═══════════════════════════════════════════════════════════════
   INSIGHTS TAB
   ═══════════════════════════════════════════════════════════════ */

let _insightsCache = [];
let _insightsQuery = "";
let _insightsCategory = "";

function _insightRelTime(iso) {
  if (!iso) return "";
  const d = new Date(iso);
  const diff = Date.now() - d.getTime();
  if (diff < 60000) return "just now";
  if (diff < 3600000) return `${Math.floor(diff / 60000)}m ago`;
  if (diff < 86400000) return `${Math.floor(diff / 3600000)}h ago`;
  if (diff < 604800000) return `${Math.floor(diff / 86400000)}d ago`;
  return d.toLocaleDateString();
}

function _renderInsightCard(item) {
  const cat = esc(item.category || "fact");
  const time = _insightRelTime(item.updated_at || item.created_at);
  const tags = item.tags ? `<span class="insight-tags">${esc(item.tags)}</span>` : "";
  const imp = item.importance || 5;
  return `
    <div class="insight-card" data-id="${esc(item.id)}">
      <div class="insight-body">
        <div class="insight-text">${esc(item.content)}</div>
        <div class="insight-meta">
          <span class="insight-badge insight-cat-${cat}">${cat}</span>
          ${imp >= 7 ? '<span class="insight-badge insight-important">★ ' + imp + '</span>' : ''}
          ${tags}
          <span class="insight-time">${time}</span>
        </div>
      </div>
      <div class="insight-actions">
        <button class="todo-action-btn todo-delete-btn insight-delete-btn" title="Delete">✕</button>
      </div>
    </div>`;
}

function _renderInsightsList() {
  const el = $("#insights-content");
  const countEl = $("#insights-count");
  let items = _insightsCache;

  if (_insightsQuery) {
    const q = _insightsQuery.toLowerCase();
    items = items.filter(i =>
      (i.content || "").toLowerCase().includes(q) ||
      (i.tags || "").toLowerCase().includes(q) ||
      (i.category || "").toLowerCase().includes(q)
    );
  }
  if (_insightsCategory) {
    items = items.filter(i => i.category === _insightsCategory);
  }

  countEl.textContent = items.length ? `${items.length} insight${items.length !== 1 ? "s" : ""}` : "";

  if (!items.length) {
    el.innerHTML = '<span class="output-placeholder">No insights found.</span>';
    el.classList.add("output-empty");
    return;
  }

  el.classList.remove("output-empty");
  el.innerHTML = items.map(_renderInsightCard).join("");

  // Wire delete buttons.
  for (const btn of el.querySelectorAll(".insight-delete-btn")) {
    btn.addEventListener("click", async (e) => {
      e.stopPropagation();
      const card = btn.closest(".insight-card");
      const id = card.dataset.id;
      if (!confirm("Delete this insight?")) return;
      try {
        await fetch(`/api/insights/${encodeURIComponent(id)}`, { method: "DELETE" });
        await loadInsights();
      } catch (err) {
        console.error("Failed to delete insight", err);
      }
    });
  }
}

async function loadInsights() {
  const el = $("#insights-content");
  try {
    const params = new URLSearchParams();
    if (_insightsQuery) params.set("q", _insightsQuery);
    if (_insightsCategory) params.set("category", _insightsCategory);
    params.set("limit", "100");
    const res = await fetch(`/api/insights?${params}`);
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    const data = await res.json();
    _insightsCache = data.items || [];
    _renderInsightsList();
  } catch (e) {
    console.error("Failed to load insights", e);
    el.innerHTML = '<span class="output-placeholder">Failed to load insights.</span>';
    el.classList.add("output-empty");
  }
}

function initInsightsTab() {
  let _searchTimer = null;
  const searchInput = $("#insights-search");
  searchInput.addEventListener("input", () => {
    clearTimeout(_searchTimer);
    _searchTimer = setTimeout(() => {
      _insightsQuery = searchInput.value.trim();
      loadInsights();
    }, 300);
  });

  const catFilter = $("#insights-category-filter");
  catFilter.addEventListener("change", () => {
    _insightsCategory = catFilter.value;
    loadInsights();
  });

  $("#insights-refresh-btn").addEventListener("click", loadInsights);
}


/* ═══════════════════════════════════════════════════════════════
   MAP TAB — SVG mind-map of the exploration tree
   ═══════════════════════════════════════════════════════════════ */

const MAP_MAIN_W = 200;
const MAP_MAIN_H = 60;
const MAP_SUB_W = 140;
const MAP_SUB_H = 36;
const MAP_H_GAP = 40;
const MAP_V_GAP = 80;
const MAP_SUB_V_GAP = 20;  // tighter gap for sub-steps

let mapTransform = { x: 0, y: 0, scale: 1 };
let mapDragging = false;
let mapDragStart = { x: 0, y: 0 };

function renderMap() {
  const svg = $("#map-svg");
  if (!svg) return;

  // Show all exploration nodes including text-selection-initiated ones.
  const allNodes = Array.from(explorationNodes.values());
  const nodes = allNodes;
  if (nodes.length === 0) {
    svg.innerHTML = `<text x="50%" y="50%" text-anchor="middle" fill="var(--text-dim)" font-size="13" font-family="var(--font)">No exploration history yet. Ask a question to begin.</text>`;
    return;
  }

  // Sort chronologically.
  const sorted = [...nodes].sort((a, b) => (a.created_at || '').localeCompare(b.created_at || ''));

  // Build parent→children map (only among main nodes).
  const mainIds = new Set(sorted.map(n => n.id));
  const childrenMap = new Map();
  const rootIds = [];

  for (const n of sorted) {
    // Walk up parent chain to find nearest main-node ancestor.
    let pid = n.parent_id;
    while (pid && !mainIds.has(pid)) {
      const pn = explorationNodes.get(pid);
      pid = pn ? pn.parent_id : null;
    }
    if (!pid || !mainIds.has(pid)) {
      rootIds.push(n.id);
    } else {
      if (!childrenMap.has(pid)) childrenMap.set(pid, []);
      childrenMap.get(pid).push(n.id);
    }
  }

  // Layout: scatter nodes across the stage for readability.
  // Uses a spiral/zigzag pattern so nodes spread in 2D instead of a single column.
  const padding = 40;
  const nodeW = MAP_MAIN_W;
  const nodeH = MAP_MAIN_H + 14;  // taller to fit answer snippet
  const pixelPositions = new Map();

  // Determine if the graph is mostly linear (no real branching).
  const totalBranches = [...childrenMap.values()].filter(c => c.length > 1).length;
  const isLinear = totalBranches === 0;

  if (isLinear && sorted.length > 1) {
    // Scatter layout: place nodes in a zigzag grid pattern.
    // Compute grid dimensions to be roughly square.
    const cols = Math.max(2, Math.ceil(Math.sqrt(sorted.length)));
    const cellW = nodeW + MAP_H_GAP + 20;
    const cellH = nodeH + MAP_V_GAP;

    // Deterministic jitter based on node index for organic feel.
    function jitter(i, range) {
      return ((i * 7 + 13) % 17) / 17 * range - range / 2;
    }

    sorted.forEach((n, i) => {
      const col = i % cols;
      const row = Math.floor(i / cols);
      // Alternate row direction for a snake/zigzag path.
      const effCol = row % 2 === 0 ? col : (cols - 1 - col);
      pixelPositions.set(n.id, {
        px: padding + effCol * cellW + jitter(i, 30),
        py: padding + row * cellH + jitter(i + 5, 20),
      });
    });
  } else {
    // Tree layout for branching graphs.
    const positions = new Map();
    let nextX = 0;

    function layoutNode(id, depth) {
      const children = childrenMap.get(id) || [];
      if (children.length === 0) {
        positions.set(id, { x: nextX, y: depth });
        nextX++;
        return;
      }
      for (const cid of children) {
        layoutNode(cid, depth + 1);
      }
      const firstChild = positions.get(children[0]);
      const lastChild = positions.get(children[children.length - 1]);
      positions.set(id, { x: (firstChild.x + lastChild.x) / 2, y: depth });
    }

    for (const rid of rootIds) {
      layoutNode(rid, 0);
    }

    let maxDepth = 0;
    for (const pos of positions.values()) maxDepth = Math.max(maxDepth, pos.y);

    const rowY = [padding];
    for (let d = 1; d <= maxDepth; d++) {
      rowY.push(rowY[d - 1] + nodeH + MAP_V_GAP);
    }

    for (const [id, pos] of positions) {
      pixelPositions.set(id, {
        px: padding + pos.x * (nodeW + MAP_H_GAP),
        py: rowY[pos.y] || padding,
      });
    }
  }

  let maxPx = 0, maxPy = 0;
  for (const p of pixelPositions.values()) {
    maxPx = Math.max(maxPx, p.px + nodeW);
    maxPy = Math.max(maxPy, p.py + nodeH);
  }
  const svgW = maxPx + padding;
  const svgH = maxPy + padding;

  let svgContent = '';

  // Edges.
  for (const n of sorted) {
    // Find effective parent among main nodes.
    let pid = n.parent_id;
    while (pid && !mainIds.has(pid)) {
      const pn = explorationNodes.get(pid);
      pid = pn ? pn.parent_id : null;
    }
    if (pid && pixelPositions.has(pid) && pixelPositions.has(n.id)) {
      const ppos = pixelPositions.get(pid);
      const cpos = pixelPositions.get(n.id);
      const x1 = ppos.px + nodeW / 2;
      const y1 = ppos.py + nodeH;
      const x2 = cpos.px + nodeW / 2;
      const y2 = cpos.py;
      const midY = (y1 + y2) / 2;

      svgContent += `<path d="M${x1},${y1} C${x1},${midY} ${x2},${midY} ${x2},${y2}"
        fill="none" stroke="var(--accent)" stroke-width="2" opacity="0.5"/>`;
    }
  }

  // Nodes — show user prompt + answer snippet.
  for (const n of sorted) {
    if (!pixelPositions.has(n.id)) continue;
    const pos = pixelPositions.get(n.id);
    const isCurrent = n.id === currentNodeId;

    const strokeColor = isCurrent ? 'var(--accent)' : 'var(--chrome-lo)';
    const strokeWidth = isCurrent ? 3 : 1;
    const fillColor = isCurrent ? 'var(--surface-alt)' : 'var(--surface)';

    svgContent += `<g class="map-node" data-id="${n.id}" style="cursor:pointer">`;
    svgContent += `<rect x="${pos.px}" y="${pos.py}" width="${nodeW}" height="${nodeH}"
      rx="4" ry="4" fill="${fillColor}" stroke="${strokeColor}" stroke-width="${strokeWidth}"/>`;

    // User prompt (bold).
    const prompt = n.prompt.length > 28 ? n.prompt.slice(0, 28) + '…' : n.prompt;
    svgContent += `<text x="${pos.px + 8}" y="${pos.py + 16}" font-size="11" font-weight="700"
      fill="var(--text)" font-family="var(--font)">▸ ${escSvg(prompt)}</text>`;

    // Answer snippet (dimmed, italic — first meaningful line).
    const answerSnippet = _mapAnswerSnippet(n.answer, 30);
    svgContent += `<text x="${pos.px + 8}" y="${pos.py + 34}" font-size="10" font-style="italic"
      fill="var(--text-dim)" font-family="var(--font)">${escSvg(answerSnippet)}</text>`;

    // Metadata line: time + indicators.
    const ts = new Date(n.created_at).toLocaleTimeString("en-GB", { hour: "2-digit", minute: "2-digit" });
    const indicators = [ts];
    if (n.answer) indicators.push(humanSize(n.answer.length));
    if (n.visual_html) indicators.push('🖼');
    svgContent += `<text x="${pos.px + 8}" y="${pos.py + 50}" font-size="9"
      fill="var(--text-dim)" font-family="var(--font)">${escSvg(indicators.join(' · '))}</text>`;

    svgContent += `</g>`;
  }

  // Apply transform.
  const t = mapTransform;
  svg.innerHTML = `<g transform="translate(${t.x},${t.y}) scale(${t.scale})">${svgContent}</g>`;
  svg.setAttribute('viewBox', `0 0 ${Math.max(svgW, 400)} ${Math.max(svgH, 300)}`);
  svg.style.width = '100%';
  svg.style.height = '100%';

  // Attach click handlers.
  for (const g of svg.querySelectorAll('.map-node')) {
    g.addEventListener('click', () => {
      const nodeId = g.dataset.id;
      if (nodeId) {
        navigateToNode(nodeId);
        const answerBtn = $(".tab-btn[data-tab='answer']");
        if (answerBtn) answerBtn.click();
      }
    });
  }
}

/** Extract a short snippet from the answer for the map node. */
function _mapAnswerSnippet(answer, maxLen) {
  if (!answer) return '—';
  // Strip markdown heading markers, bullet points, code fences.
  const lines = answer.split('\n').map(l => l.trim()).filter(l =>
    l && !l.startsWith('#') && !l.startsWith('```') && !l.startsWith('---')
  );
  const text = (lines[0] || '').replace(/^\*+\s*/, '').replace(/\*+/g, '');
  if (text.length > maxLen) return text.slice(0, maxLen) + '…';
  return text || '—';
}

function escSvg(s) {
  return s.replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/"/g, '&quot;');
}

function initMap() {
  const container = $("#map-container");
  if (!container) return;

  // Pan via drag.
  container.addEventListener("mousedown", (e) => {
    if (e.target.closest('.map-node')) return;
    mapDragging = true;
    mapDragStart = { x: e.clientX - mapTransform.x, y: e.clientY - mapTransform.y };
    container.style.cursor = "grabbing";
  });

  container.addEventListener("mousemove", (e) => {
    if (!mapDragging) return;
    mapTransform.x = e.clientX - mapDragStart.x;
    mapTransform.y = e.clientY - mapDragStart.y;
    renderMap();
  });

  container.addEventListener("mouseup", () => {
    mapDragging = false;
    container.style.cursor = "grab";
  });

  container.addEventListener("mouseleave", () => {
    mapDragging = false;
    container.style.cursor = "grab";
  });

  // Zoom via wheel.
  container.addEventListener("wheel", (e) => {
    e.preventDefault();
    const delta = e.deltaY > 0 ? 0.9 : 1.1;
    mapTransform.scale = Math.max(0.2, Math.min(3, mapTransform.scale * delta));
    renderMap();
  }, { passive: false });

  // Zoom buttons.
  const zoomIn = $("#map-zoom-in");
  const zoomOut = $("#map-zoom-out");
  const fit = $("#map-fit");

  if (zoomIn) zoomIn.addEventListener("click", () => {
    mapTransform.scale = Math.min(3, mapTransform.scale * 1.2);
    renderMap();
  });
  if (zoomOut) zoomOut.addEventListener("click", () => {
    mapTransform.scale = Math.max(0.2, mapTransform.scale / 1.2);
    renderMap();
  });
  if (fit) fit.addEventListener("click", () => {
    mapTransform = { x: 0, y: 0, scale: 1 };
    renderMap();
  });
}

/* ═══════════════════════════════════════════════════════════════
   FILES TAB — themed file cards with preview
   ═══════════════════════════════════════════════════════════════ */

const FILE_ICONS = {
  '.html': '🌐', '.htm': '🌐',
  '.js': '📜', '.ts': '📜', '.mjs': '📜',
  '.py': '🐍',
  '.json': '📋', '.yaml': '📋', '.yml': '📋', '.toml': '📋',
  '.css': '🎨', '.scss': '🎨',
  '.md': '📝', '.txt': '📝', '.log': '📝',
  '.png': '🖼', '.jpg': '🖼', '.jpeg': '🖼', '.gif': '🖼', '.webp': '🖼', '.svg': '🖼', '.bmp': '🖼',
  '.mp3': '🎵', '.wav': '🎵', '.ogg': '🎵', '.flac': '🎵',
  '.mp4': '🎬', '.webm': '🎬', '.mov': '🎬',
  '.pdf': '📕',
  '.csv': '📊', '.xlsx': '📊', '.xls': '📊',
  '.zip': '📦', '.tar': '📦', '.gz': '📦',
  '.sh': '⚙', '.bash': '⚙',
};

const IMAGE_EXTS = new Set(['.png', '.jpg', '.jpeg', '.gif', '.webp', '.svg', '.bmp']);
const HTML_EXTS = new Set(['.html', '.htm']);

function getFileIcon(ext) {
  return FILE_ICONS[ext] || '📄';
}

function humanFileSize(bytes) {
  if (bytes < 1024) return bytes + ' B';
  if (bytes < 1024 * 1024) return (bytes / 1024).toFixed(1) + ' KB';
  return (bytes / (1024 * 1024)).toFixed(1) + ' MB';
}

const FOLDER_ICONS = {
  media: '🖼', downloads: '⬇', scripts: '📜', output: '📤',
  showcase: '🏆', skills: '🧩', summaries: '📝', tmp: '📁',
  tools: '🔧', other: '📂',
};

function extractFolder(file) {
  // Derive folder group from logical path (e.g. "media/photo.png" → "media").
  const logical = file.logical || '';
  if (logical) {
    const seg = logical.split('/')[0];
    if (seg && seg !== logical) return seg;
  }
  // Fall back: try to parse category from physical path segments.
  const phys = file.physical || '';
  const parts = phys.replace(/\\/g, '/').split('/');
  const savedIdx = parts.lastIndexOf('saved');
  if (savedIdx >= 0 && savedIdx + 1 < parts.length) {
    return parts[savedIdx + 1];
  }
  const outputIdx = parts.lastIndexOf('output');
  if (outputIdx >= 0) return 'output';
  return file.source || 'other';
}

const TWO_DAYS_MS = 2 * 24 * 60 * 60 * 1000;

function _fileTimeLabel(mtime) {
  if (!mtime) return '';
  const d = new Date(mtime * 1000);
  const now = Date.now();
  const diff = now - d.getTime();
  const pad = (n) => String(n).padStart(2, '0');
  // Today: just time
  const today = new Date();
  if (d.toDateString() === today.toDateString()) {
    return `Today ${pad(d.getHours())}:${pad(d.getMinutes())}`;
  }
  // Yesterday
  const yesterday = new Date(now - 86400000);
  if (d.toDateString() === yesterday.toDateString()) {
    return `Yesterday ${pad(d.getHours())}:${pad(d.getMinutes())}`;
  }
  // Older: date + time
  return `${d.getFullYear()}-${pad(d.getMonth()+1)}-${pad(d.getDate())} ${pad(d.getHours())}:${pad(d.getMinutes())}`;
}

function _isOldFile(mtime) {
  if (!mtime) return false;
  return (Date.now() - mtime * 1000) > TWO_DAYS_MS;
}

let _filesSortMode = 'date'; // 'date' | 'name' | 'type'

function renderFileCard(f) {
  const icon = getFileIcon(f.extension || '');
  const size = f.size ? humanFileSize(f.size) : '';
  const previewable = f.is_text || IMAGE_EXTS.has(f.extension) || HTML_EXTS.has(f.extension);
  const timeLabel = _fileTimeLabel(f.modified);
  const isOld = _isOldFile(f.modified);
  const oldBadge = isOld ? '<span class="file-old-badge">OLD</span>' : '';
  return `
    <div class="file-card${isOld ? ' file-old' : ''}" data-path="${esc(f.physical)}" data-ext="${esc(f.extension || '')}" data-text="${f.is_text ? '1' : '0'}">
      <span class="file-icon">${icon}</span>
      <div class="file-info">
        <div class="file-name">${esc(f.filename)}${oldBadge}</div>
        <div class="file-meta">${esc(size)}${timeLabel ? ' · ' + esc(timeLabel) : ''}${f.logical ? ' · ' + esc(f.logical) : ''}</div>
      </div>
      <div class="file-actions">
        ${previewable ? '<button class="file-action-btn file-preview-btn" title="Preview file"><span class="file-action-icon">👁</span><span class="file-action-label">Preview</span></button>' : ''}
        <a class="file-action-btn" href="/api/files/view?path=${encodeURIComponent(f.physical)}" target="_blank" title="Open in new tab"><span class="file-action-icon">↗</span><span class="file-action-label">Open</span></a>
        <a class="file-action-btn" href="/api/files/download?path=${encodeURIComponent(f.physical)}" title="Download file"><span class="file-action-icon">⬇</span><span class="file-action-label">Download</span></a>
        <button class="file-action-btn file-delete-btn" title="Delete file"><span class="file-action-icon">🗑</span><span class="file-action-label">Delete</span></button>
      </div>
      <label class="file-select-cb" style="display:none"><input type="checkbox" class="file-select-check"></label>
    </div>
  `;
}

function _sortFiles(files) {
  const sorted = [...files];
  switch (_filesSortMode) {
    case 'name':
      sorted.sort((a, b) => (a.filename || '').localeCompare(b.filename || ''));
      break;
    case 'type':
      sorted.sort((a, b) => {
        const ea = (a.extension || '').toLowerCase();
        const eb = (b.extension || '').toLowerCase();
        if (ea !== eb) return ea.localeCompare(eb);
        return (a.filename || '').localeCompare(b.filename || '');
      });
      break;
    case 'date':
    default:
      sorted.sort((a, b) => (b.modified || 0) - (a.modified || 0)); // newest first
      break;
  }
  return sorted;
}

async function loadFiles() {
  const el = $("#files-content");
  const countEl = $("#files-count");

  try {
    // Use session-scoped endpoint when session is known, fall back to global.
    const url = sessionId
      ? `/api/files/session/${encodeURIComponent(sessionId)}`
      : "/api/files";
    const res = await fetch(url);
    if (!res.ok) throw new Error(`HTTP ${res.status}`);
    const files = await res.json();

    if (!files || files.length === 0) {
      _filesCache = [];
      el.innerHTML = '<span class="output-placeholder">No files yet.</span>';
      el.classList.add("output-empty");
      countEl.textContent = '';
      return;
    }

    _filesCache = files;

    // Apply current search filter if any.
    const searchInput = $("#files-search");
    const q = searchInput ? searchInput.value.trim() : '';
    if (q) {
      filterFiles(q);
      return;
    }

    countEl.textContent = `${files.length} file${files.length !== 1 ? 's' : ''}`;
    el.classList.remove("output-empty");

    // Sort then group files by folder category.
    const sortedFiles = _sortFiles(files);
    const groups = new Map();
    for (const f of sortedFiles) {
      const folder = extractFolder(f);
      if (!groups.has(folder)) groups.set(folder, []);
      groups.get(folder).push(f);
    }

    // Sort groups: put common categories first, then alpha.
    const order = ['output', 'media', 'downloads', 'scripts', 'tools', 'skills', 'summaries', 'showcase', 'tmp'];
    const sortedKeys = Array.from(groups.keys()).sort((a, b) => {
      const ia = order.indexOf(a);
      const ib = order.indexOf(b);
      if (ia >= 0 && ib >= 0) return ia - ib;
      if (ia >= 0) return -1;
      if (ib >= 0) return 1;
      return a.localeCompare(b);
    });

    let html = '';
    for (const folder of sortedKeys) {
      const groupFiles = groups.get(folder);
      const folderIcon = FOLDER_ICONS[folder] || FOLDER_ICONS.other;
      const collapsed = '';  // start expanded
      html += `
        <div class="file-group${collapsed}">
          <div class="file-group-header" onclick="this.parentElement.classList.toggle('collapsed')">
            <span class="file-group-icon">${folderIcon}</span>
            <span class="file-group-name">${esc(folder)}</span>
            <span class="file-group-count">${groupFiles.length}</span>
            <span class="file-group-chevron">▾</span>
          </div>
          <div class="file-group-body">
            ${groupFiles.map(renderFileCard).join('')}
          </div>
        </div>
      `;
    }

    el.innerHTML = html;

    // Attach preview handlers.
    for (const btn of el.querySelectorAll('.file-preview-btn')) {
      btn.addEventListener('click', (e) => {
        e.stopPropagation();
        const card = btn.closest('.file-card');
        openFilePreview(card.dataset.path, card.dataset.ext, card.dataset.text === '1');
      });
    }

    // Attach single-file delete handlers.
    for (const btn of el.querySelectorAll('.file-delete-btn')) {
      btn.addEventListener('click', (e) => {
        e.stopPropagation();
        const card = btn.closest('.file-card');
        const fname = card.querySelector('.file-name')?.textContent || 'this file';
        deleteFileWithConfirm([card.dataset.path], fname);
      });
    }

  } catch (err) {
    console.warn("Failed to load files:", err);
    el.innerHTML = `<span class="output-placeholder">Failed to load files: ${esc(err.message)}</span>`;
    el.classList.add("output-empty");
  }
}

let _filesCache = [];   // cached file list for search filtering
let _filesSelectMode = false;

function filterFiles(query) {
  const el = $("#files-content");
  if (!_filesCache.length) return;

  const q = (query || '').trim().toLowerCase();
  const filtered = q
    ? _filesCache.filter(f =>
        (f.filename || '').toLowerCase().includes(q) ||
        (f.extension || '').toLowerCase().includes(q) ||
        (f.logical || '').toLowerCase().includes(q))
    : _filesCache;

  const countEl = $("#files-count");
  if (q) {
    countEl.textContent = `${filtered.length}/${_filesCache.length} files`;
  } else {
    countEl.textContent = `${_filesCache.length} file${_filesCache.length !== 1 ? 's' : ''}`;
  }

  if (filtered.length === 0) {
    el.innerHTML = `<span class="output-placeholder">No files matching "${esc(q)}"</span>`;
    el.classList.add("output-empty");
    return;
  }

  el.classList.remove("output-empty");

  // Sort then group filtered files by folder.
  const sortedFiltered = _sortFiles(filtered);
  const groups = new Map();
  for (const f of sortedFiltered) {
    const folder = extractFolder(f);
    if (!groups.has(folder)) groups.set(folder, []);
    groups.get(folder).push(f);
  }
  const order = ['output', 'media', 'downloads', 'scripts', 'tools', 'skills', 'summaries', 'showcase', 'tmp'];
  const sortedKeys = Array.from(groups.keys()).sort((a, b) => {
    const ia = order.indexOf(a); const ib = order.indexOf(b);
    if (ia >= 0 && ib >= 0) return ia - ib;
    if (ia >= 0) return -1;
    if (ib >= 0) return 1;
    return a.localeCompare(b);
  });

  let html = '';
  for (const folder of sortedKeys) {
    const groupFiles = groups.get(folder);
    const folderIcon = FOLDER_ICONS[folder] || FOLDER_ICONS.other;
    html += `
      <div class="file-group">
        <div class="file-group-header" onclick="this.parentElement.classList.toggle('collapsed')">
          <span class="file-group-icon">${folderIcon}</span>
          <span class="file-group-name">${esc(folder)}</span>
          <span class="file-group-count">${groupFiles.length}</span>
          <span class="file-group-chevron">▾</span>
        </div>
        <div class="file-group-body">
          ${groupFiles.map(renderFileCard).join('')}
        </div>
      </div>
    `;
  }
  el.innerHTML = html;

  // Re-attach preview handlers.
  for (const btn of el.querySelectorAll('.file-preview-btn')) {
    btn.addEventListener('click', (e) => {
      e.stopPropagation();
      const card = btn.closest('.file-card');
      openFilePreview(card.dataset.path, card.dataset.ext, card.dataset.text === '1');
    });
  }
  // Re-attach delete handlers.
  for (const btn of el.querySelectorAll('.file-delete-btn')) {
    btn.addEventListener('click', (e) => {
      e.stopPropagation();
      const card = btn.closest('.file-card');
      const fname = card.querySelector('.file-name')?.textContent || 'this file';
      deleteFileWithConfirm([card.dataset.path], fname);
    });
  }
  // Sync selection mode checkboxes.
  if (_filesSelectMode) {
    el.querySelectorAll('.file-select-cb').forEach(cb => cb.style.display = '');
  }
}

/* ── File deletion ───────────────────────────────────────────── */

async function deleteFileWithConfirm(paths, label) {
  if (!paths || !paths.length) return;
  const msg = paths.length === 1
    ? `Delete "${label}"?`
    : `Delete ${paths.length} selected files?`;
  if (!confirm(msg)) return;
  await deleteFiles(paths);
}

async function deleteFiles(paths) {
  try {
    const res = await fetch('/api/files/delete', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ paths }),
      credentials: 'same-origin',
    });
    const data = await res.json();
    if (data.deleted > 0) {
      logEntry('system', `Deleted ${data.deleted} file${data.deleted !== 1 ? 's' : ''}`);
    }
    if (data.errors && data.errors.length) {
      for (const err of data.errors) {
        logEntry('error', err);
      }
    }
    // Exit selection mode and refresh.
    exitFileSelectMode();
    loadFiles();
  } catch (err) {
    logEntry('error', 'Failed to delete files: ' + err.message);
  }
}

function enterFileSelectMode() {
  _filesSelectMode = true;
  // Show checkboxes on all file cards.
  document.querySelectorAll('#files-content .file-select-cb').forEach(cb => cb.style.display = '');
  // Toggle toolbar buttons.
  $("#files-select-btn").style.display = 'none';
  $("#files-delete-selected-btn").style.display = '';
  $("#files-cancel-select-btn").style.display = '';
  updateDeleteSelectedCount();
}

function exitFileSelectMode() {
  _filesSelectMode = false;
  // Hide and uncheck all checkboxes.
  document.querySelectorAll('#files-content .file-select-cb').forEach(cb => {
    cb.style.display = 'none';
    cb.querySelector('input').checked = false;
  });
  // Toggle toolbar buttons.
  $("#files-select-btn").style.display = '';
  $("#files-delete-selected-btn").style.display = 'none';
  $("#files-cancel-select-btn").style.display = 'none';
}

function getSelectedFilePaths() {
  const paths = [];
  document.querySelectorAll('#files-content .file-select-check:checked').forEach(cb => {
    const card = cb.closest('.file-card');
    if (card && card.dataset.path) paths.push(card.dataset.path);
  });
  return paths;
}

function updateDeleteSelectedCount() {
  const btn = $("#files-delete-selected-btn");
  const count = document.querySelectorAll('#files-content .file-select-check:checked').length;
  btn.textContent = count > 0 ? `🗑 Delete (${count})` : '🗑 Delete';
  btn.disabled = count === 0;
}

function initFiles() {
  const refreshBtn = $("#files-refresh-btn");
  if (refreshBtn) {
    refreshBtn.addEventListener("click", () => loadFiles());
  }
  const searchInput = $("#files-search");
  if (searchInput) {
    let debounceTimer;
    searchInput.addEventListener("input", () => {
      clearTimeout(debounceTimer);
      debounceTimer = setTimeout(() => filterFiles(searchInput.value), 150);
    });
  }
  // Sort buttons.
  const sortBtns = document.querySelectorAll(".files-sort-btn");
  sortBtns.forEach((btn) => {
    btn.addEventListener("click", () => {
      const mode = btn.dataset.sort;
      if (mode === _filesSortMode) return;
      _filesSortMode = mode;
      sortBtns.forEach((b) => b.classList.toggle("active", b.dataset.sort === mode));
      // Re-render with current search filter.
      const q = searchInput ? searchInput.value.trim() : '';
      if (q) {
        filterFiles(q);
      } else if (_filesCache.length) {
        // Re-trigger full render from cache.
        loadFiles();
      }
    });
  });

  // Select / bulk delete buttons.
  $("#files-select-btn").addEventListener("click", enterFileSelectMode);
  $("#files-cancel-select-btn").addEventListener("click", exitFileSelectMode);
  $("#files-delete-selected-btn").addEventListener("click", () => {
    const paths = getSelectedFilePaths();
    if (!paths.length) return;
    deleteFileWithConfirm(paths, paths.length + ' files');
  });
  // Update count when checkboxes change (event delegation).
  $("#files-content").addEventListener("change", (e) => {
    if (e.target.classList.contains("file-select-check")) {
      updateDeleteSelectedCount();
    }
  });
}

/* ── File preview modal ──────────────────────────────────────── */

async function openFilePreview(path, ext, isText) {
  const modal = $("#file-preview-modal");
  const title = $("#preview-title");
  const body = $("#preview-body");
  const pdfBtn = $("#preview-pdf-btn");
  const filename = path.split('/').pop();

  title.textContent = filename;
  body.innerHTML = '<div class="output-processing"><span class="processing-dots">⣾</span> Loading preview...</div>';
  modal.classList.remove("hidden");
  pdfBtn.style.display = "none";
  // Clear previous listener.
  pdfBtn.replaceWith(pdfBtn.cloneNode(true));
  const pdfBtnNew = $("#preview-pdf-btn");

  const MD_EXTS = new Set(['.md', '.markdown', '.mdown', '.mkd']);

  try {
    if (IMAGE_EXTS.has(ext)) {
      // Image: render inline.
      body.innerHTML = `<img src="/api/files/view?path=${encodeURIComponent(path)}" alt="${esc(filename)}">`;
    } else if (HTML_EXTS.has(ext)) {
      // HTML: render in sandboxed iframe + show PDF button.
      const contentRes = await fetch(`/api/files/content?path=${encodeURIComponent(path)}`);
      let htmlContent = "";
      if (contentRes.ok) {
        const data = await contentRes.json();
        htmlContent = data.content || "";
      }
      if (htmlContent) {
        body.innerHTML = `<iframe sandbox="allow-scripts"></iframe>`;
        body.querySelector("iframe").srcdoc = htmlContent;
      } else {
        body.innerHTML = `<iframe src="/api/files/view?path=${encodeURIComponent(path)}" sandbox="allow-scripts"></iframe>`;
      }
      // Show PDF export button.
      pdfBtnNew.style.display = "";
      pdfBtnNew.addEventListener("click", () => exportPreviewPdf(htmlContent || null, path, filename));
    } else if (MD_EXTS.has(ext)) {
      // Markdown: fetch content and render as formatted HTML.
      const res = await fetch(`/api/files/content?path=${encodeURIComponent(path)}`);
      if (!res.ok) throw new Error(`HTTP ${res.status}`);
      const data = await res.json();
      body.innerHTML = `<div class="md-preview">${markdownToHtml(data.content || '(empty)')}</div>`;
    } else if (isText) {
      // Text: fetch content and render in pre block.
      const res = await fetch(`/api/files/content?path=${encodeURIComponent(path)}`);
      if (!res.ok) throw new Error(`HTTP ${res.status}`);
      const data = await res.json();
      body.innerHTML = `<pre>${esc(data.content || '(empty)')}</pre>`;
    } else {
      body.innerHTML = '<span class="output-placeholder">Preview not available for this file type.</span>';
    }
  } catch (err) {
    body.innerHTML = `<span class="output-placeholder">Preview failed: ${esc(err.message)}</span>`;
  }
}

async function exportPreviewPdf(html, path, filename) {
  const pdfBtn = $("#preview-pdf-btn");
  pdfBtn.disabled = true;
  pdfBtn.textContent = "...";

  try {
    // If we don't have inline HTML, fetch it.
    if (!html) {
      const res = await fetch(`/api/files/content?path=${encodeURIComponent(path)}`);
      if (!res.ok) throw new Error(`HTTP ${res.status}`);
      const data = await res.json();
      html = data.content || "";
    }
    if (!html) throw new Error("No HTML content to export");

    const pdfName = filename.replace(/\.html?$/i, "") || "file";

    const res = await fetch("/api/computer/export-pdf", {
      method: "POST",
      headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ html, prompt: pdfName }),
    });

    if (!res.ok) {
      const err = await res.json().catch(() => ({}));
      throw new Error(err.error || `HTTP ${res.status}`);
    }

    const cd = res.headers.get("Content-Disposition") || "";
    const fnMatch = cd.match(/filename="([^"]+)"/);
    const outFilename = fnMatch ? fnMatch[1] : `${pdfName}.pdf`;

    const blob = await res.blob();
    const url = URL.createObjectURL(blob);
    const a = document.createElement("a");
    a.href = url;
    a.download = outFilename;
    document.body.appendChild(a);
    a.click();
    document.body.removeChild(a);
    URL.revokeObjectURL(url);

    logEntry("system", `PDF exported: ${outFilename} (${humanSize(blob.size)})`);
  } catch (err) {
    logEntry("error", `PDF export failed: ${err.message}`);
  } finally {
    pdfBtn.disabled = false;
    pdfBtn.textContent = "PDF";
  }
}

function initFilePreview() {
  const modal = $("#file-preview-modal");
  if (!modal) return;

  // Close on backdrop click.
  const backdrop = modal.querySelector(".modal-backdrop");
  if (backdrop) backdrop.addEventListener("click", () => modal.classList.add("hidden"));

  // Close on ✕ button.
  const closeBtn = modal.querySelector(".modal-close");
  if (closeBtn) closeBtn.addEventListener("click", () => modal.classList.add("hidden"));

  // Close on Escape.
  document.addEventListener("keydown", (e) => {
    if (e.key === "Escape" && !modal.classList.contains("hidden")) {
      modal.classList.add("hidden");
    }
  });
}

/* ═══════════════════════════════════════════════════════════════
   MODEL & TIER SELECTORS
   ═══════════════════════════════════════════════════════════════ */

async function loadModels() {
  try {
    const res = await fetch("/api/orchestrator/models");
    if (!res.ok) return;
    const data = await res.json();
    availableModels = data.models || [];
    renderModelGrid();
    updateModelBtnLabel();
    console.log(`[Computer] Loaded ${availableModels.length} available models`);
  } catch (e) {
    console.warn("Failed to load models:", e);
  }
}

function providerIcon(provider) {
  const p = (provider || "").toLowerCase();
  if (p.includes("anthropic") || p.includes("claude")) return "🟣";
  if (p.includes("openai") || p.includes("gpt")) return "🟢";
  if (p.includes("google") || p.includes("gemini") || p.includes("vertex")) return "🔵";
  if (p.includes("groq")) return "🟠";
  if (p.includes("mistral")) return "🔴";
  if (p.includes("ollama") || p.includes("local")) return "⚪";
  if (p.includes("openrouter")) return "🟡";
  return "⬜";
}

function renderModelGrid() {
  const grid = $("#model-grid");
  grid.innerHTML = "";

  // Default option
  const defCard = document.createElement("div");
  defCard.className = "model-card" + (!selectedModel ? " selected" : "");
  defCard.dataset.modelId = "";
  defCard.innerHTML =
    '<div class="model-card-icon">⚙️</div>' +
    '<div class="model-card-info">' +
      '<div class="model-card-name">Default Model</div>' +
      '<div class="model-card-detail">Use the session\'s default model for all operations</div>' +
    '</div>' +
    (!selectedModel ? '<div class="model-card-badges"><span class="model-badge active">Active</span></div>' : '');
  defCard.addEventListener("click", () => selectModel(""));
  grid.appendChild(defCard);

  for (const m of availableModels) {
    if (m.id === "default" && availableModels.length === 1) continue; // skip if it's the only fallback
    const card = document.createElement("div");
    card.className = "model-card" + (m.id === selectedModel ? " selected" : "");
    card.dataset.modelId = m.id;

    const name = m.description || m.id;
    const detail = `${m.provider} / ${m.model}`;
    const badges = [];
    if (m.reasoning_level) badges.push(m.reasoning_level);
    if (m.model_type && m.model_type !== "llm") badges.push(m.model_type);
    if (m.id === selectedModel) badges.push("Active");

    card.innerHTML =
      '<div class="model-card-icon">' + providerIcon(m.provider) + '</div>' +
      '<div class="model-card-info">' +
        '<div class="model-card-name">' + esc(name) + '</div>' +
        '<div class="model-card-detail">' + esc(detail) + '</div>' +
      '</div>' +
      (badges.length
        ? '<div class="model-card-badges">' +
          badges.map(b => '<span class="model-badge' + (b === "Active" ? " active" : "") + '">' + esc(b) + '</span>').join("") +
          '</div>'
        : '');

    card.addEventListener("click", () => selectModel(m.id));
    grid.appendChild(card);
  }
}

function selectModel(modelId) {
  selectedModel = modelId;
  localStorage.setItem("computer-model", selectedModel);

  // Update visuals.
  renderModelGrid();
  updateModelBtnLabel();

  // Also set the model for the chat/agent session via WebSocket.
  if (modelId && connected) {
    send({ type: "set_model", selector: modelId });
  } else if (!modelId && connected) {
    // Reset to default — send the default model id.
    send({ type: "set_model", selector: "default" });
  }

  logEntry("system", `Model: ${selectedModel || '(default)'}`);
  closeModelModal();
}

function updateModelBtnLabel() {
  const label = $("#model-btn-label");
  if (!selectedModel) {
    label.textContent = "Default";
    return;
  }
  const m = availableModels.find(m => m.id === selectedModel);
  label.textContent = m ? (m.description || m.id) : selectedModel;
}

function openModelModal() {
  renderModelGrid();
  $("#model-modal").classList.add("active");
}

function closeModelModal() {
  $("#model-modal").classList.remove("active");
}

function initModelModal() {
  $("#model-btn").addEventListener("click", openModelModal);
  $("#model-modal .modal-close").addEventListener("click", closeModelModal);
  $("#model-modal .modal-backdrop").addEventListener("click", closeModelModal);
  document.addEventListener("keydown", (e) => {
    if (e.key === "Escape" && $("#model-modal").classList.contains("active")) {
      closeModelModal();
    }
  });
}

/* ═══════════════════════════════════════════════════════════════
   PERSONA SELECTOR
   ═══════════════════════════════════════════════════════════════ */

async function loadPersonas() {
  try {
    const res = await fetch("/api/user-personalities");
    if (!res.ok) return;
    const data = await res.json();
    availablePersonas = (Array.isArray(data) ? data : data.personalities || []).map(p => {
      p.id = String(p.user_id || p.id || "").trim();
      return p;
    });
    renderPersonaGrid();
    updatePersonaBtnLabel();
    console.log(`[Computer] Loaded ${availablePersonas.length} personas`);
  } catch (e) {
    console.warn("Failed to load personas:", e);
  }
}

function renderPersonaGrid() {
  const grid = $("#persona-grid");
  grid.innerHTML = "";

  // "No profile" option.
  const noneCard = document.createElement("div");
  noneCard.className = "model-card" + (!selectedPersona ? " selected" : "");
  noneCard.innerHTML =
    '<div class="model-card-icon">🚫</div>' +
    '<div class="model-card-info">' +
      '<div class="model-card-name">No profile</div>' +
      '<div class="model-card-detail">Generic context — no user profile applied</div>' +
    '</div>' +
    (!selectedPersona ? '<div class="model-card-badges"><span class="model-badge active">Active</span></div>' : '');
  noneCard.addEventListener("click", () => selectPersona(""));
  grid.appendChild(noneCard);

  for (const p of availablePersonas) {
    if (!p.id) continue;
    const card = document.createElement("div");
    card.className = "model-card" + (p.id === selectedPersona ? " selected" : "");

    const name = p.name || p.id;
    const detail = p.description || "";
    const badges = [];
    if (p.is_telegram) badges.push("Telegram");
    if (p.id === selectedPersona) badges.push("Active");

    card.innerHTML =
      '<div class="model-card-icon">👤</div>' +
      '<div class="model-card-info">' +
        '<div class="model-card-name">' + esc(name) + '</div>' +
        (detail ? '<div class="model-card-detail">' + esc(detail) + '</div>' : '') +
      '</div>' +
      (badges.length
        ? '<div class="model-card-badges">' +
          badges.map(b => '<span class="model-badge' + (b === "Active" ? " active" : "") + '">' + esc(b) + '</span>').join("") +
          '</div>'
        : '');

    card.addEventListener("click", () => selectPersona(p.id));
    grid.appendChild(card);
  }
}

function selectPersona(personaId) {
  selectedPersona = personaId;
  localStorage.setItem("computer-persona", selectedPersona);

  renderPersonaGrid();
  updatePersonaBtnLabel();

  if (connected) {
    send({ type: "set_personality", personality_id: personaId });
  }

  logEntry("system", `Persona: ${selectedPersona || '(none)'}`);
  closePersonaModal();
}

function updatePersonaBtnLabel() {
  const label = $("#persona-btn-label");
  if (!selectedPersona) {
    label.textContent = "No profile";
    return;
  }
  const p = availablePersonas.find(p => p.id === selectedPersona);
  label.textContent = p ? (p.name || p.id) : selectedPersona;
}

function openPersonaModal() {
  renderPersonaGrid();
  $("#persona-modal").classList.add("active");
}

function closePersonaModal() {
  $("#persona-modal").classList.remove("active");
}

function initPersonaModal() {
  $("#persona-btn").addEventListener("click", openPersonaModal);
  $("#persona-modal .modal-close").addEventListener("click", closePersonaModal);
  $("#persona-modal .modal-backdrop").addEventListener("click", closePersonaModal);
  document.addEventListener("keydown", (e) => {
    if (e.key === "Escape" && $("#persona-modal").classList.contains("active")) {
      closePersonaModal();
    }
  });
}

function initSelectors() {
  // Model modal.
  initModelModal();

  // Persona modal.
  initPersonaModal();

  // Tier popup.
  initTierPopup();

  // BYOK modal.
  initByokModal();

  // Playbook modal.
  initPlaybookModal();

  // Approval modal.
  initApprovalModal();
}

/* ── BYOK (Bring Your Own Key) ───────────────────────────────── */

function initByokModal() {
  $("#byok-btn").addEventListener("click", openByokModal);
  $("#byok-modal .modal-close").addEventListener("click", closeByokModal);
  $("#byok-modal .modal-backdrop").addEventListener("click", closeByokModal);
  $("#byok-apply-btn").addEventListener("click", applyByok);
  $("#byok-clear-btn").addEventListener("click", clearByok);
  $("#byok-key-toggle").addEventListener("click", function() {
    var inp = $("#byok-apikey");
    inp.type = inp.type === "password" ? "text" : "password";
  });
  $("#byok-toggle").addEventListener("change", toggleByok);
  // Pre-fill from localStorage.
  if (byokProvider) $("#byok-provider").value = byokProvider;
  if (byokModel) $("#byok-model").value = byokModel;
  if (byokApiKey) $("#byok-apikey").value = byokApiKey;
  syncByokToggleUI();
  updateByokBtnLabel();
}

function syncByokToggleUI() {
  var toggle = $("#byok-toggle");
  var fields = $("#byok-fields");
  var hasCreds = !!(byokProvider && byokModel && byokApiKey);
  toggle.checked = byokEnabled && hasCreds;
  // Always show fields so user can enter/edit credentials.
  // Dim them when toggle is off to hint they're inactive.
  fields.classList.toggle("byok-fields-dimmed", hasCreds && !byokEnabled);
}

function toggleByok() {
  var toggle = $("#byok-toggle");
  byokEnabled = toggle.checked;
  localStorage.setItem("computer-byok-enabled", byokEnabled ? "true" : "false");

  if (byokEnabled && byokProvider && byokModel && byokApiKey) {
    // Re-activate saved credentials.
    if (connected) {
      send({ type: "set_byok", provider: byokProvider, model: byokModel, api_key: byokApiKey });
    }
    logEntry("system", "BYOK enabled: " + byokProvider + "/" + byokModel);
  } else if (!byokEnabled) {
    // Switch to server default without clearing saved credentials.
    if (connected) {
      send({ type: "clear_byok" });
    }
    byokActive = false;
    logEntry("system", "BYOK disabled — using server default");
  }
  syncByokToggleUI();
  updateByokBtnLabel();
}

function openByokModal() {
  // Sync fields from localStorage (in case changed elsewhere).
  if (byokProvider) $("#byok-provider").value = byokProvider;
  if (byokModel) $("#byok-model").value = byokModel;
  if (byokApiKey) $("#byok-apikey").value = byokApiKey;
  syncByokToggleUI();
  $("#byok-status").textContent = "";
  $("#byok-modal").classList.add("active");
}

function closeByokModal() {
  $("#byok-modal").classList.remove("active");
}

function applyByok() {
  var provider = $("#byok-provider").value;
  var model = $("#byok-model").value.trim();
  var apiKey = $("#byok-apikey").value.trim();
  if (!model) {
    $("#byok-status").textContent = "Model name is required.";
    $("#byok-status").className = "byok-status byok-error";
    return;
  }
  if (!apiKey) {
    $("#byok-status").textContent = "API key is required.";
    $("#byok-status").className = "byok-status byok-error";
    return;
  }
  // Save to localStorage.
  byokProvider = provider;
  byokModel = model;
  byokApiKey = apiKey;
  byokEnabled = true;
  localStorage.setItem("computer-byok-provider", provider);
  localStorage.setItem("computer-byok-model", model);
  localStorage.setItem("computer-byok-apikey", apiKey);
  localStorage.setItem("computer-byok-enabled", "true");
  // Send to server.
  if (connected) {
    send({ type: "set_byok", provider: provider, model: model, api_key: apiKey });
    $("#byok-status").textContent = "Applying...";
    $("#byok-status").className = "byok-status";
  }
  syncByokToggleUI();
}

function clearByok() {
  byokProvider = "";
  byokModel = "";
  byokApiKey = "";
  byokEnabled = true;
  byokActive = false;
  localStorage.removeItem("computer-byok-provider");
  localStorage.removeItem("computer-byok-model");
  localStorage.removeItem("computer-byok-apikey");
  localStorage.removeItem("computer-byok-enabled");
  $("#byok-provider").value = "openai";
  $("#byok-model").value = "";
  $("#byok-apikey").value = "";
  if (connected) {
    send({ type: "clear_byok" });
  }
  syncByokToggleUI();
  updateByokBtnLabel();
  $("#byok-status").textContent = "Credentials cleared.";
  $("#byok-status").className = "byok-status";
  logEntry("system", "BYOK credentials cleared");
}

function handleByokStatus(data) {
  byokActive = !!data.active;
  updateByokBtnLabel();
  syncByokToggleUI();
  var statusEl = $("#byok-status");
  if (data.error) {
    statusEl.textContent = "Error: " + data.error;
    statusEl.className = "byok-status byok-error";
    logEntry("error", "BYOK error: " + data.error);
  } else if (data.active) {
    statusEl.textContent = "Active: " + data.provider + "/" + data.model;
    statusEl.className = "byok-status byok-ok";
    logEntry("system", "BYOK active: " + data.provider + "/" + data.model);
    closeByokModal();
  } else {
    statusEl.textContent = "Using server default.";
    statusEl.className = "byok-status";
  }
}

function updateByokBtnLabel() {
  var label = $("#byok-btn-label");
  if (byokActive && byokProvider) {
    label.textContent = byokProvider + "/" + (byokModel || "").split("/").pop();
  } else if (byokProvider && byokModel && !byokEnabled) {
    label.textContent = "BYOK (off)";
  } else {
    label.textContent = "BYOK";
  }
}

/* ── Markdown to HTML (basic) ────────────────────────────────── */

function markdownToHtml(md) {
  if (!md) return "";

  // Pre-process: extract tables before escaping (they need raw pipe chars).
  const tableBlocks = [];
  const TABLE_PLACEHOLDER = '\x00TABLE_';
  md = md.replace(
    /(^[ \t]*\|.+\|[ \t]*\n)([ \t]*\|[\s:]*[-:]+[-|\s:]*\|[ \t]*\n)((?:[ \t]*\|.+\|[ \t]*\n?)*)/gm,
    (match, headerLine, sepLine, bodyLines) => {
      const idx = tableBlocks.length;
      tableBlocks.push({ headerLine, sepLine, bodyLines });
      return TABLE_PLACEHOLDER + idx + '\n';
    }
  );

  // Pre-process: convert raw <img> HTML tags to markdown image syntax
  // so they survive the esc() call.  Handles both <img src="..." alt="...">
  // and <img src="..." alt="..." ...other attrs...> patterns.
  md = md.replace(/<img\s+[^>]*src\s*=\s*["']([^"']+)["'][^>]*>/gi, function(match, src) {
    var altMatch = match.match(/alt\s*=\s*["']([^"']*)["']/i);
    var alt = altMatch ? altMatch[1] : "Image";
    return "![" + alt + "](" + src + ")";
  });

  let html = esc(md);

  // Code blocks (``` ... ```) — must run before inline replacements.
  html = html.replace(/```(\w*)\n([\s\S]*?)```/g, "<pre><code>$2</code></pre>");

  // Images: ![alt](path) — route local paths through /api/media
  html = html.replace(/!\[([^\]]*)\]\(([^)]+)\)/g, function(_, alt, src) {
    var url = src;
    // file:/// URLs → strip scheme and route through /api/media
    if (/^file:\/\/\//i.test(src)) {
      url = '/api/media?path=' + encodeURIComponent(src.replace(/^file:\/\/\//i, '/'));
    } else if (/^saved\/|^output\/|^\//.test(src)) {
      url = '/api/media?path=' + encodeURIComponent(src);
    }
    return '<img src="' + url + '" alt="' + alt +
      '" style="max-width:100%;border-radius:8px;cursor:pointer;" ' +
      'onclick="window.open(this.src,\'_blank\')">';
  });

  // Attached image: [Attached image: /path/to/file.jpg]
  html = html.replace(/\[Attached image: ([^\]]+)\]/g, function(_, path) {
    var p = path.trim().replace(/^file:\/\/\//i, '/');
    var url = '/api/media?path=' + encodeURIComponent(p);
    return '<img src="' + url + '" alt="Attached image" ' +
      'style="max-width:100%;max-height:300px;border-radius:8px;cursor:pointer;display:block;margin:6px 0;" ' +
      'onclick="window.open(this.src,\'_blank\')">';
  });

  // Audio: [Audio: /path/to/file.mp3]
  html = html.replace(/\[Audio: ([^\]]+)\]/gi, function(_, path) {
    var p = path.trim();
    var url = /^https?:\/\//.test(p) ? p : '/api/media?path=' + encodeURIComponent(p);
    return '<audio controls style="display:block;width:100%;margin:6px 0;border-radius:8px;">' +
      '<source src="' + url + '">Your browser does not support audio.</audio>';
  });

  // Links to audio files: [text](path.mp3) → inline <audio> player
  html = html.replace(/\[([^\]]+)\]\(([^)]+\.(?:mp3|wav|ogg|flac|m4a|aac))\)/gi, function(_, label, src) {
    var url = /^https?:\/\//.test(src) ? src : '/api/media?path=' + encodeURIComponent(src);
    return '<div style="margin:6px 0;"><strong>' + label + '</strong>' +
      '<audio controls style="display:block;width:100%;margin:4px 0;border-radius:8px;">' +
      '<source src="' + url + '">Your browser does not support audio.</audio></div>';
  });

  // Auto-detect bare audio file paths → inline <audio> player.
  html = html.replace(/`?((?:\/|saved\/|output\/)[^\s<>&quot;&#39;`]+\.(?:mp3|wav|ogg|flac|m4a|aac))`?/gi, function(_, path) {
    var url = '/api/media?path=' + encodeURIComponent(path);
    return '<audio controls style="display:block;width:100%;margin:6px 0;border-radius:8px;">' +
      '<source src="' + url + '">Your browser does not support audio.</audio>';
  });

  // Auto-detect bare image file paths → inline <img>.
  // Matches paths like /path/to/image.jpg, saved/media/image.png, file:///path/image.webp
  // Consumes optional surrounding ** (bold) or ` (code) markers.
  html = html.replace(/(?:\*\*|`)?(?:file:\/\/\/)?((?:\/|saved\/|output\/)[^\s<>&"'`*]+\.(?:png|jpe?g|gif|webp|bmp|svg))(?:\*\*|`)?/gi, function(_, path) {
    var url = '/api/media?path=' + encodeURIComponent(path);
    return '<img src="' + url + '" alt="Image" ' +
      'style="max-width:100%;border-radius:8px;cursor:pointer;display:block;margin:8px 0;" ' +
      'onclick="window.open(this.src,\'_blank\')">';
  });

  // HTML file view buttons: [View: /path/to/file.html] or bare .html paths
  html = html.replace(/\[View: ([^\]]+\.html?)\]/gi, function(_, path) {
    var url = '/api/files/view?path=' + encodeURIComponent(path.trim());
    return '<div class="html-view-card" style="margin:8px 0;padding:8px 12px;border:1px solid var(--border);border-radius:8px;display:inline-flex;align-items:center;gap:8px;">' +
      '<span style="opacity:.7;">HTML</span> <span>' + esc(path.trim().split('/').pop()) + '</span> ' +
      '<a href="' + url + '" target="_blank" style="padding:3px 10px;border-radius:4px;background:var(--accent);color:#fff;text-decoration:none;font-size:.85em;">View</a></div>';
  });

  // Links: [text](url) — after images/audio so those aren't caught here.
  html = html.replace(/\[([^\]]+)\]\(([^)]+)\)/g, '<a href="$2" target="_blank">$1</a>');

  html = html.replace(/^### (.+)$/gm, "<h3>$1</h3>");
  html = html.replace(/^## (.+)$/gm, "<h2>$1</h2>");
  html = html.replace(/^# (.+)$/gm, "<h1>$1</h1>");

  html = html.replace(/\*\*(.+?)\*\*/g, "<strong>$1</strong>");
  html = html.replace(/\*(.+?)\*/g, "<em>$1</em>");

  html = html.replace(/`(.+?)`/g, "<code>$1</code>");

  html = html.replace(/^&gt; (.+)$/gm, "<blockquote>$1</blockquote>");

  html = html.replace(/^[-*] (.+)$/gm, "<li>$1</li>");
  html = html.replace(/(<li>.*<\/li>\n?)+/g, "<ul>$&</ul>");

  // Restore table blocks as HTML tables.
  for (let i = 0; i < tableBlocks.length; i++) {
    const tb = tableBlocks[i];
    const headerCells = parseTableRow(tb.headerLine);
    const bodyRows = tb.bodyLines.trim()
      ? tb.bodyLines.trim().split('\n').map(parseTableRow)
      : [];

    // Parse alignment from separator row.
    const sepCells = parseTableRow(tb.sepLine);
    const aligns = sepCells.map(s => {
      s = s.trim();
      if (s.startsWith(':') && s.endsWith(':')) return 'center';
      if (s.endsWith(':')) return 'right';
      return 'left';
    });

    let tableHtml = '<table><thead><tr>';
    headerCells.forEach((cell, ci) => {
      const align = aligns[ci] || 'left';
      tableHtml += `<th style="text-align:${align}">${_inlineMd(cell)}</th>`;
    });
    tableHtml += '</tr></thead><tbody>';
    for (const row of bodyRows) {
      tableHtml += '<tr>';
      row.forEach((cell, ci) => {
        const align = aligns[ci] || 'left';
        tableHtml += `<td style="text-align:${align}">${_inlineMd(cell)}</td>`;
      });
      tableHtml += '</tr>';
    }
    tableHtml += '</tbody></table>';

    html = html.replace(TABLE_PLACEHOLDER + i, tableHtml);
  }

  html = html.replace(/^(?!<[hupblot])([\s\S]+?)(?=\n\n|$)/gm, (match) => {
    const trimmed = match.trim();
    if (!trimmed || trimmed.startsWith("<")) return match;
    return `<p>${trimmed}</p>`;
  });

  html = html.replace(/\n{2,}/g, "\n");

  return html;
}

/** Apply inline markdown (bold, italic, code, links) to a table cell. */
function _inlineMd(text) {
  let s = esc(text);
  s = s.replace(/\*\*(.+?)\*\*/g, "<strong>$1</strong>");
  s = s.replace(/\*(.+?)\*/g, "<em>$1</em>");
  s = s.replace(/`(.+?)`/g, "<code>$1</code>");
  s = s.replace(/\[([^\]]+)\]\(([^)]+)\)/g, '<a href="$2" target="_blank">$1</a>');
  return s;
}

function parseTableRow(line) {
  line = line.trim();
  if (line.startsWith('|')) line = line.slice(1);
  if (line.endsWith('|')) line = line.slice(0, -1);
  return line.split('|').map(c => c.trim());
}

/* ── Utilities ───────────────────────────────────────────────── */

function humanTime(ms) {
  if (ms < 1000) return ms + "ms";
  const s = ms / 1000;
  if (s < 60) return s.toFixed(1) + "s";
  const m = Math.floor(s / 60);
  const rem = (s % 60).toFixed(1);
  return m + "m " + rem + "s";
}

function humanSize(chars) {
  if (chars < 1000) return chars + " chars";
  return (chars / 1000).toFixed(1) + "k chars";
}

function esc(s) {
  if (!s) return "";
  const d = document.createElement("div");
  d.textContent = s;
  return d.innerHTML;
}

/* ── Mobile helpers ──────────────────────────────────────────── */

function _isMobile() {
  return window.innerWidth <= 800;
}

function _scrollToOutput() {
  const el = document.getElementById("output-panel");
  if (el) {
    el.scrollIntoView({ behavior: "smooth", block: "start" });
  }
}

/* ── Stream panel (bottom slide-up) ──────────────────────────── */

let streamPanelActive = false;
let streamPanelSource = "";  // "answer" | "visual"

function _alignStreamPanel() {
  // On desktop, align the stream panel with the output panel.
  const panel = $("#stream-panel");
  if (_isMobile()) {
    panel.style.left = "0";
    return;
  }
  const output = $("#output-panel");
  if (output) {
    const rect = output.getBoundingClientRect();
    panel.style.left = rect.left + "px";
  }
}

function openStreamPanel(source) {
  const panel = $("#stream-panel");
  const content = $("#stream-panel-content");
  const meta = $("#stream-panel-meta");
  const body = $("#stream-panel-body");

  content.textContent = "";
  meta.textContent = source === "visual" ? "Visual generation" : "Agent thinking";
  streamPanelSource = source;
  streamPanelActive = true;

  _alignStreamPanel();
  panel.classList.remove("hidden");
  body.classList.add("streaming");
  document.body.classList.add("stream-panel-open");

  // Hide the re-open button while the panel is visible.
  const showBtn = $("#btn-show-stream");
  if (showBtn) showBtn.classList.add("hidden");

  playStreamStartSound();
}

function reopenStreamPanel() {
  const panel = $("#stream-panel");
  _alignStreamPanel();
  panel.classList.remove("hidden");
  document.body.classList.add("stream-panel-open");
  // Hide the re-open button.
  const showBtn = $("#btn-show-stream");
  if (showBtn) showBtn.classList.add("hidden");
  // Scroll to bottom of content.
  const body = $("#stream-panel-body");
  body.scrollTop = body.scrollHeight;
}

function closeStreamPanel() {
  const panel = $("#stream-panel");
  const body = $("#stream-panel-body");
  streamPanelActive = false;
  streamPanelSource = "";
  body.classList.remove("streaming");
  panel.classList.add("hidden");
  document.body.classList.remove("stream-panel-open");

  playStreamStopSound();

  // Show the re-open button if there is content to review.
  const content = $("#stream-panel-content");
  const showBtn = $("#btn-show-stream");
  if (showBtn) {
    if (content && content.textContent.trim()) {
      showBtn.classList.remove("hidden");
    } else {
      showBtn.classList.add("hidden");
    }
  }
}

function appendStreamText(text) {
  if (!streamPanelActive) return;
  const content = $("#stream-panel-content");
  content.textContent += text;
  playStreamChunkSound();
  // Auto-scroll to bottom.
  const body = $("#stream-panel-body");
  body.scrollTop = body.scrollHeight;
}

function finishStreamPanel(label) {
  const body = $("#stream-panel-body");
  const meta = $("#stream-panel-meta");
  body.classList.remove("streaming");
  if (label) meta.textContent = label;
  playStreamStopSound();
}

function initStreamPanel() {
  $("#stream-panel-close").addEventListener("click", closeStreamPanel);
  const showBtn = $("#btn-show-stream");
  if (showBtn) showBtn.addEventListener("click", reopenStreamPanel);
}

/* ── Auto-resize textarea ────────────────────────────────────── */

function initTextarea() {
  const textarea = $("#input-box");

  textarea.addEventListener("input", () => {
    textarea.style.height = "auto";
    textarea.style.height = Math.min(textarea.scrollHeight, 120) + "px";
  });

  textarea.addEventListener("keydown", (e) => {
    if (e.key === "Enter" && !e.shiftKey) {
      e.preventDefault();
      handleSend();
    }
  });

  $("#send-btn").addEventListener("click", () => handleSend());
  $("#stop-btn").addEventListener("click", () => handleStop());
}

/* ── Clear log button ────────────────────────────────────────── */

function initClearLog() {
  const clearBtn = $(".win-clear");
  if (clearBtn) {
    clearBtn.addEventListener("click", () => {
      $("#log-content").innerHTML = "";
      logEntry("system", "Log cleared");
    });
  }
}

/* ── Clear session button ────────────────────────────────────── */

function initClearSession() {
  const btn = $("#clear-session-btn");
  if (!btn) return;
  btn.addEventListener("click", () => {
    showConfirmDialog("Clear the current session? All context will be lost.", () => {
      // Send /clear command to backend.
      send({ type: "command", command: "/clear" });
      // Reset local UI state.
      explorationNodes.clear();
      currentNodeId = null;
      historyBranchNodeId = null;
      pendingExploration = null;
      lastPrompt = "";
      lastResult = "";
      agentCreatedHtmlFiles = [];
      setProcessing(false);
      // Reset answer to empty placeholder (not "Processing...").
      removeNextSteps();
      const ansEl = $("#answer-content");
      ansEl.innerHTML = '<span class="output-placeholder">Responses will appear here...</span>';
      ansEl.classList.add("output-empty");
      clearVisual();
      renderHistoryDrawer();
      $("#log-content").innerHTML = "";
      logEntry("system", "Session cleared");
    });
  });
}

/* ── Session settings modal ─────────────────────────────────── */

function initSessionSettings() {
  const btn = $("#session-settings-btn");
  const modal = $("#session-settings-modal");
  if (!btn || !modal) return;

  const backdrop = modal.querySelector(".modal-backdrop");
  const closeBtn = modal.querySelector(".modal-close");
  const cancelBtn = $("#ss-cancel");
  const saveBtn = $("#ss-save");
  const nameInput = $("#ss-name");
  const descInput = $("#ss-description");
  const instInput = $("#ss-instructions");

  function openModal() {
    // Populate fields from current state.
    nameInput.value = _sessionSettings.name;
    descInput.value = _sessionSettings.description;
    instInput.value = _sessionSettings.instructions;
    modal.classList.add("active");
  }

  function closeModal() {
    modal.classList.remove("active");
  }

  function saveSettings() {
    const payload = {
      type: "session_settings",
      session_name: nameInput.value.trim(),
      session_description: descInput.value.trim(),
      session_instructions: instInput.value.trim(),
    };
    send(payload);
    closeModal();
  }

  btn.addEventListener("click", openModal);
  if (backdrop) backdrop.addEventListener("click", closeModal);
  if (closeBtn) closeBtn.addEventListener("click", closeModal);
  if (cancelBtn) cancelBtn.addEventListener("click", closeModal);
  if (saveBtn) saveBtn.addEventListener("click", saveSettings);
}

function showConfirmDialog(message, onYes) {
  const overlay = document.createElement("div");
  overlay.className = "confirm-overlay";
  overlay.innerHTML = `
    <div class="confirm-box">
      <p>${esc(message)}</p>
      <div class="confirm-actions">
        <button class="confirm-yes">Yes, clear</button>
        <button class="confirm-no">Cancel</button>
      </div>
    </div>`;

  const close = () => overlay.remove();
  overlay.querySelector(".confirm-no").addEventListener("click", close);
  overlay.querySelector(".confirm-yes").addEventListener("click", () => {
    close();
    onYes();
  });
  // Click outside the box to cancel.
  overlay.addEventListener("click", (e) => {
    if (e.target === overlay) close();
  });
  // Escape key cancels.
  const escHandler = (e) => {
    if (e.key === "Escape") { close(); document.removeEventListener("keydown", escHandler); }
  };
  document.addEventListener("keydown", escHandler);

  document.body.appendChild(overlay);
  overlay.querySelector(".confirm-no").focus();
}

/* ── Processing animation ────────────────────────────────────── */

const SPINNER_FRAMES = ["⣾", "⣽", "⣻", "⢿", "⡿", "⣟", "⣯", "⣷"];
let spinnerIdx = 0;

function initSpinner() {
  setInterval(() => {
    const els = $$(".processing-dots");
    for (const el of els) {
      spinnerIdx = (spinnerIdx + 1) % SPINNER_FRAMES.length;
      el.textContent = SPINNER_FRAMES[spinnerIdx];
    }
  }, 100);
}

/* ── Attachment / paste / drag-drop ──────────────────────────── */

function resizeImageFile(file, maxSide) {
  return new Promise((resolve, reject) => {
    const img = new Image();
    const url = URL.createObjectURL(file);
    img.onload = () => {
      URL.revokeObjectURL(url);
      const w = img.width, h = img.height;
      const scale = (w > maxSide || h > maxSide) ? maxSide / Math.max(w, h) : 1;
      const nw = Math.round(w * scale), nh = Math.round(h * scale);
      const canvas = document.createElement("canvas");
      canvas.width = nw; canvas.height = nh;
      canvas.getContext("2d").drawImage(img, 0, 0, nw, nh);
      canvas.toBlob((blob) => {
        if (!blob) { resolve(file); return; }
        const outName = file.name.replace(/\.[^.]+$/, ".jpg");
        resolve(new File([blob], outName, { type: "image/jpeg" }));
      }, "image/jpeg", 0.85);
    };
    img.onerror = () => { URL.revokeObjectURL(url); reject(new Error("load")); };
    img.src = url;
  });
}

async function uploadFile(file) {
  if (!file) return;
  const ext = file.name.split(".").pop().toLowerCase();
  const isImage = _IMAGE_EXTS.indexOf(ext) !== -1;
  const isData = _DATA_EXTS.indexOf(ext) !== -1;

  if (!isImage && !isData) {
    logEntry("error", "Supported: images (.png, .jpg, .jpeg, .webp, .gif, .bmp), documents (.pdf, .docx, .pptx, .md, .txt), data (.csv, .xlsx), and archives (.zip).");
    return;
  }

  const attachBtn = $("#attach-btn");

  if (isImage) {
    attachBtn.disabled = true;
    attachBtn.textContent = "⏳";
    try { file = await resizeImageFile(file, 1024); } catch (_) {}
    const formData = new FormData();
    formData.append("file", file);
    try {
      const res = await fetch("/api/image/upload", { method: "POST", body: formData });
      const data = await res.json();
      if (!res.ok) {
        logEntry("error", "Upload failed: " + (data.error || "Unknown error"));
      } else {
        showAttachmentChip(file.name, data.path, false);
      }
    } catch (err) {
      logEntry("error", "Upload failed: " + err.message);
    } finally {
      attachBtn.disabled = false;
      attachBtn.textContent = "📎";
      $("#file-input").value = "";
    }
    return;
  }

  // Data file
  attachBtn.disabled = true;
  attachBtn.textContent = "⏳";
  const formData = new FormData();
  formData.append("file", file);
  try {
    const res = await fetch("/api/file/upload", { method: "POST", body: formData });
    const data = await res.json();
    if (!res.ok) {
      logEntry("error", "Upload failed: " + (data.error || "Unknown error"));
    } else {
      showAttachmentChip(file.name, data.path, true);
    }
  } catch (err) {
    logEntry("error", "Upload failed: " + err.message);
  } finally {
    attachBtn.disabled = false;
    attachBtn.textContent = "📎";
    $("#file-input").value = "";
  }
}

function showAttachmentChip(filename, path, isFile) {
  const entry = { path, name: filename };
  if (isFile) { pendingFiles.push(entry); } else { pendingImages.push(entry); }
  _renderAttachmentBar();
}

function _renderAttachmentBar() {
  const bar = $("#attachment-bar");
  const chip = $("#attachment-chip");
  chip.innerHTML = "";
  const all = [
    ...pendingImages.map((f, i) => ({ ...f, isFile: false, idx: i })),
    ...pendingFiles.map((f, i) => ({ ...f, isFile: true, idx: i })),
  ];
  if (all.length === 0) {
    bar.classList.add("hidden");
    return;
  }
  for (const item of all) {
    const wrap = document.createElement("span");
    wrap.className = "attachment-item";
    const span = document.createElement("span");
    span.className = "attachment-name";
    span.textContent = (item.isFile ? "📄 " : "🖼️ ") + item.name;
    wrap.appendChild(span);
    const removeBtn = document.createElement("button");
    removeBtn.className = "attachment-remove";
    removeBtn.textContent = "✕";
    removeBtn.title = "Remove";
    removeBtn.addEventListener("click", () => {
      if (item.isFile) pendingFiles.splice(item.idx, 1);
      else pendingImages.splice(item.idx, 1);
      _renderAttachmentBar();
    });
    wrap.appendChild(removeBtn);
    chip.appendChild(wrap);
  }
  bar.classList.remove("hidden");
}

function clearAttachment() {
  pendingImages = [];
  pendingFiles = [];
  const bar = $("#attachment-bar");
  if (bar) bar.classList.add("hidden");
  const chip = $("#attachment-chip");
  if (chip) chip.innerHTML = "";
  const fi = $("#file-input");
  if (fi) fi.value = "";
}

function initAttachments() {
  const attachBtn = $("#attach-btn");
  const fileInput = $("#file-input");
  const inputBox = $("#input-box");
  const workspace = $("#workspace");

  attachBtn.addEventListener("click", () => fileInput.click());
  fileInput.addEventListener("change", () => {
    if (fileInput.files) {
      for (const f of fileInput.files) uploadFile(f);
    }
  });

  // Paste from clipboard — handle multiple images/files.
  inputBox.addEventListener("paste", (e) => {
    const items = e.clipboardData && e.clipboardData.items;
    if (!items) return;
    let handled = false;
    for (let i = 0; i < items.length; i++) {
      if (items[i].type.indexOf("image/") === 0) {
        let file = items[i].getAsFile();
        if (file) {
          if (!file.name || file.name === "image.png") {
            const ext = (file.type.split("/")[1] || "png") === "jpeg" ? "jpg" : (file.type.split("/")[1] || "png");
            const ts = new Date().toISOString().replace(/[:.]/g, "-").slice(0, 19);
            file = new File([file], "pasted-" + ts + "-" + i + "." + ext, { type: file.type });
          }
          uploadFile(file);
          handled = true;
        }
      } else if (items[i].kind === "file") {
        const file = items[i].getAsFile();
        if (file) { uploadFile(file); handled = true; }
      }
    }
    if (handled) e.preventDefault();
  });

  // Drag & drop on workspace — handle multiple files.
  let dragCounter = 0;
  workspace.addEventListener("dragenter", (e) => { e.preventDefault(); dragCounter++; });
  workspace.addEventListener("dragover", (e) => { e.preventDefault(); });
  workspace.addEventListener("dragleave", (e) => { e.preventDefault(); dragCounter--; });
  workspace.addEventListener("drop", (e) => {
    e.preventDefault();
    dragCounter = 0;
    if (e.dataTransfer && e.dataTransfer.files) {
      for (const f of e.dataTransfer.files) uploadFile(f);
    }
  });
}

/* ── Folder browser ─────────────────────────────────────────── */

function openFolderModal() {
  $("#folder-modal").classList.add("active");
  loadFolderList();
  browseFolderDir("");
  initFolderTabs();
}

function closeFolderModal() {
  $("#folder-modal").classList.remove("active");
}

async function initFolderTabs() {
  // Check for Windows drives.
  try {
    const res = await fetch("/api/drives");
    const data = await res.json();
    const bar = $("#folder-drive-bar");
    if (data.drives && data.drives.length > 0) {
      bar.classList.remove("hidden");
      bar.innerHTML = data.drives.map(d =>
        '<button class="folder-drive-btn" data-drive="' + esc(d) + '">' + esc(d) + '</button>'
      ).join("");
      bar.querySelectorAll(".folder-drive-btn").forEach(btn => {
        btn.addEventListener("click", () => browseFolderDir(btn.dataset.drive + "\\"));
      });
    } else {
      bar.classList.add("hidden");
    }
  } catch (_) {
    $("#folder-drive-bar").classList.add("hidden");
  }

  // Check for GDrive availability.
  try {
    const res = await fetch("/api/gws-status");
    const data = await res.json();
    _gwsAvailable = !!data.available;
    $("#folder-tab-gdrive-btn").style.display = _gwsAvailable ? "" : "none";
  } catch (_) {
    _gwsAvailable = false;
    $("#folder-tab-gdrive-btn").style.display = "none";
  }

  // Tab switching.
  for (const tab of $$(".folder-modal-tab")) {
    tab.onclick = () => {
      for (const t of $$(".folder-modal-tab")) t.classList.remove("active");
      tab.classList.add("active");
      const target = tab.dataset.tab;
      $("#folder-tab-local").classList.toggle("hidden", target !== "local");
      $("#folder-tab-gdrive").classList.toggle("hidden", target !== "gdrive");
      if (target === "gdrive") loadGdriveTab();
    };
  }
}

async function loadFolderList() {
  const list = $("#folder-list");
  try {
    const res = await fetch("/api/read-folders");
    const data = await res.json();
    const dirs = data.dirs || [];
    if (!dirs.length) {
      list.innerHTML = '<div class="folder-empty">No extra folders configured</div>';
      return;
    }
    list.innerHTML = dirs.map(d => {
      const cls = d.exists ? "" : " missing";
      return '<div class="folder-entry' + cls + '">' +
        '<span class="folder-entry-path" title="' + esc(d.resolved) + '">' + esc(d.resolved) + '</span>' +
        (!d.exists ? '<span class="folder-entry-warn" title="Directory does not exist">⚠</span>' : '') +
        '<button class="folder-entry-remove" data-path="' + esc(d.resolved) + '" title="Remove">✕</button>' +
        '</div>';
    }).join("");
    list.querySelectorAll(".folder-entry-remove").forEach(btn => {
      btn.addEventListener("click", () => removeFolderDir(btn.dataset.path));
    });
  } catch (_) {
    list.innerHTML = '<div class="folder-empty">Failed to load</div>';
  }
}

async function browseFolderDir(path) {
  const dirsEl = $("#folder-dirs");
  dirsEl.innerHTML = '<div class="folder-empty">Loading...</div>';
  try {
    const url = "/api/browse" + (path ? "?path=" + encodeURIComponent(path) : "");
    const res = await fetch(url);
    const data = await res.json();
    if (data.error) { dirsEl.innerHTML = '<div class="folder-empty">' + esc(data.error) + '</div>'; return; }
    folderCurrentPath = data.path;
    $("#folder-selected-path").textContent = data.path;
    const addBtn = $("#folder-add-btn");
    addBtn.disabled = !!data.blocked;

    renderFolderBreadcrumb(data.path);

    let html = "";
    if (data.parent) {
      html += '<div class="folder-dir-item parent" data-path="' + esc(data.parent) + '"><span class="folder-dir-icon">⬆</span> ..</div>';
    }
    if (!data.dirs.length && !data.parent) {
      html += '<div class="folder-empty">No subdirectories</div>';
    }
    data.dirs.forEach(name => {
      const sep = data.path.includes("\\") ? "\\" : "/";
      const full = data.path + (data.path.endsWith(sep) ? "" : sep) + name;
      html += '<div class="folder-dir-item" data-path="' + esc(full) + '"><span class="folder-dir-icon">📁</span> ' + esc(name) + '</div>';
    });
    dirsEl.innerHTML = html;
    dirsEl.querySelectorAll(".folder-dir-item").forEach(el => {
      el.addEventListener("click", () => browseFolderDir(el.dataset.path));
    });
  } catch (e) {
    dirsEl.innerHTML = '<div class="folder-empty">Error: ' + esc(e.message) + '</div>';
  }
}

function renderFolderBreadcrumb(fullPath) {
  const bc = $("#folder-breadcrumb");
  const sep = fullPath.includes("\\") ? "\\" : "/";
  const parts = fullPath.split(sep).filter(Boolean);
  let html = "", built = "";
  if (fullPath.startsWith("/")) {
    html += '<span class="folder-bc-item" data-path="/">/</span>';
    built = "/";
  }
  parts.forEach((p, i) => {
    if (i === 0 && !fullPath.startsWith("/")) { built = p + sep; }
    else { built += (built.endsWith(sep) ? "" : sep) + p; }
    if (i > 0 || fullPath.startsWith("/")) html += '<span class="folder-bc-sep">' + sep + '</span>';
    html += '<span class="folder-bc-item" data-path="' + esc(built) + '">' + esc(p) + '</span>';
  });
  bc.innerHTML = html;
  bc.querySelectorAll(".folder-bc-item").forEach(el => {
    el.addEventListener("click", () => browseFolderDir(el.dataset.path));
  });
}

async function addCurrentFolder() {
  if (!folderCurrentPath) return;
  const btn = $("#folder-add-btn");
  btn.disabled = true; btn.textContent = "Adding...";
  try {
    const res = await fetch("/api/read-folders", {
      method: "POST", headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ path: folderCurrentPath }),
    });
    const data = await res.json();
    if (!res.ok || !data.ok) { logEntry("error", "Add folder failed: " + (data.error || "Unknown")); }
    else { logEntry("system", "Read folder added: " + data.path); loadFolderList(); }
  } catch (e) { logEntry("error", "Add folder failed: " + e.message); }
  finally { btn.disabled = false; btn.textContent = "Add this folder"; }
}

async function removeFolderDir(path) {
  try {
    const res = await fetch("/api/read-folders", {
      method: "DELETE", headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ path }),
    });
    const data = await res.json();
    if (!res.ok || !data.ok) { logEntry("error", "Remove folder failed: " + (data.error || "Unknown")); }
    else { logEntry("system", "Read folder removed: " + data.path); loadFolderList(); }
  } catch (e) { logEntry("error", "Remove folder failed: " + e.message); }
}

// ── Google Drive folder browser ──

async function loadGdriveTab() {
  loadGdriveFolderList();
  browseGdrive("root", "My Drive");
}

async function loadGdriveFolderList() {
  const list = $("#gdrive-list");
  try {
    const res = await fetch("/api/read-folders/gdrive");
    const data = await res.json();
    const folders = data.folders || [];
    if (!folders.length) { list.innerHTML = '<div class="folder-empty">No Google Drive folders configured</div>'; return; }
    list.innerHTML = folders.map(f =>
      '<div class="folder-entry"><span class="folder-entry-path" title="ID: ' + esc(f.id) + '">' +
      esc(f.name) + '</span><button class="folder-entry-remove" data-id="' + esc(f.id) + '" title="Remove">✕</button></div>'
    ).join("");
    list.querySelectorAll(".folder-entry-remove").forEach(btn => {
      btn.addEventListener("click", () => removeGdriveFolder(btn.dataset.id));
    });
  } catch (_) { list.innerHTML = '<div class="folder-empty">Failed to load</div>'; }
}

async function browseGdrive(folderId, folderName) {
  const dirsEl = $("#gdrive-dirs");
  dirsEl.innerHTML = '<div class="folder-empty">Loading...</div>';
  _gdriveCurrent = { id: folderId, name: folderName };
  $("#gdrive-selected-name").textContent = folderName;
  $("#gdrive-add-btn").disabled = (folderId === "root");

  if (folderId === "root") { _gdriveBcStack = [{ id: "root", name: "My Drive" }]; }
  else {
    const idx = _gdriveBcStack.findIndex(b => b.id === folderId);
    if (idx >= 0) _gdriveBcStack = _gdriveBcStack.slice(0, idx + 1);
    else _gdriveBcStack.push({ id: folderId, name: folderName });
  }
  renderGdriveBreadcrumb();

  try {
    const res = await fetch("/api/read-folders/gdrive/browse?folder_id=" + encodeURIComponent(folderId));
    const data = await res.json();
    if (data.error) { dirsEl.innerHTML = '<div class="folder-empty">' + esc(data.error) + '</div>'; return; }
    const folders = data.folders || [];
    const sharedDrives = data.shared_drives || [];
    let html = "";
    if (_gdriveBcStack.length > 1) {
      const parent = _gdriveBcStack[_gdriveBcStack.length - 2];
      html += '<div class="folder-dir-item parent" data-gid="' + esc(parent.id) + '" data-gname="' + esc(parent.name) + '"><span class="folder-dir-icon">⬆</span> ..</div>';
    }
    if (sharedDrives.length) {
      html += '<div class="folder-section-lbl">Shared Drives</div>';
      sharedDrives.forEach(d => {
        html += '<div class="folder-dir-item" data-gid="' + esc(d.id) + '" data-gname="' + esc(d.name) + '"><span class="folder-dir-icon">📤</span> ' + esc(d.name) + '</div>';
      });
      if (folders.length) html += '<div class="folder-section-lbl">My Drive</div>';
    }
    if (!folders.length && !sharedDrives.length && _gdriveBcStack.length <= 1) {
      html += '<div class="folder-empty">No subfolders</div>';
    }
    folders.forEach(f => {
      html += '<div class="folder-dir-item" data-gid="' + esc(f.id) + '" data-gname="' + esc(f.name) + '"><span class="folder-dir-icon">📁</span> ' + esc(f.name) + '</div>';
    });
    dirsEl.innerHTML = html;
    dirsEl.querySelectorAll(".folder-dir-item").forEach(el => {
      el.addEventListener("click", () => browseGdrive(el.dataset.gid, el.dataset.gname));
    });
  } catch (e) { dirsEl.innerHTML = '<div class="folder-empty">Error: ' + esc(e.message) + '</div>'; }
}

function renderGdriveBreadcrumb() {
  const bc = $("#gdrive-breadcrumb");
  let html = "";
  _gdriveBcStack.forEach((item, i) => {
    if (i > 0) html += '<span class="folder-bc-sep">/</span>';
    html += '<span class="folder-bc-item" data-gid="' + esc(item.id) + '" data-gname="' + esc(item.name) + '">' + esc(item.name) + '</span>';
  });
  bc.innerHTML = html;
  bc.querySelectorAll(".folder-bc-item").forEach(el => {
    el.addEventListener("click", () => browseGdrive(el.dataset.gid, el.dataset.gname));
  });
}

async function addGdriveFolder() {
  if (_gdriveCurrent.id === "root") return;
  const btn = $("#gdrive-add-btn");
  btn.disabled = true; btn.textContent = "Adding...";
  try {
    const res = await fetch("/api/read-folders/gdrive", {
      method: "POST", headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ id: _gdriveCurrent.id, name: _gdriveCurrent.name }),
    });
    const data = await res.json();
    if (!res.ok || !data.ok) { logEntry("error", "Add GDrive folder failed: " + (data.error || "Unknown")); }
    else { logEntry("system", "Google Drive folder added: " + data.name); loadGdriveFolderList(); }
  } catch (e) { logEntry("error", "Add GDrive folder failed: " + e.message); }
  finally { btn.disabled = false; btn.textContent = "Add this folder"; }
}

async function removeGdriveFolder(id) {
  try {
    const res = await fetch("/api/read-folders/gdrive", {
      method: "DELETE", headers: { "Content-Type": "application/json" },
      body: JSON.stringify({ id }),
    });
    const data = await res.json();
    if (!res.ok || !data.ok) { logEntry("error", "Remove GDrive folder failed: " + (data.error || "Unknown")); }
    else { logEntry("system", "Google Drive folder removed"); loadGdriveFolderList(); }
  } catch (e) { logEntry("error", "Remove GDrive folder failed: " + e.message); }
}

function initFolderModal() {
  const folderBtn = $("#folder-btn");
  folderBtn.addEventListener("click", openFolderModal);

  // Close handlers.
  $("#folder-modal .modal-close").addEventListener("click", closeFolderModal);
  $("#folder-modal .modal-backdrop").addEventListener("click", closeFolderModal);

  // Add buttons.
  $("#folder-add-btn").addEventListener("click", addCurrentFolder);
  $("#gdrive-add-btn").addEventListener("click", addGdriveFolder);
}

/* ── Sound effects (Web Audio API) ────────────────────────────── */

let soundMode = "typewriter";  // "off" | "typewriter" | "modem"
let _audioCtx = null;

function _getAudioCtx() {
  if (!_audioCtx) _audioCtx = new (window.AudioContext || window.webkitAudioContext)();
  if (_audioCtx.state === "suspended") _audioCtx.resume();
  return _audioCtx;
}

/* ═══════════════════════════════════════════════════════════════
   THEME SOUND PROFILES
   Each theme defines parameters that shape all sound generators.
   Win11 is the baseline; other themes diverge in character.
   ═══════════════════════════════════════════════════════════════ */

const _themeSoundProfiles = {
  /* ── Win 11 — clean, modern, minimal (BASELINE) ─────────────── */
  win11: {
    key: { freqs: [1200, 900, 1500], qs: [1.8, 1.4, 2.0], dur: 0.05, vol: 0.35, decay: 2.5, filter: "bandpass" },
    tick: { freq: 800, freqEnd: 400, dur: 0.08, vol: 0.25, wave: "sine" },
    tool: { pins: [3, 6], pinDur: 0.008, gap: 0.012, hpFreq: [1800, 600], vol: 0.35 },
    llm:  { freq: 420, freqMid: 820, freqEnd: 580, dur: 0.12, vol: 0.4, wave: "sine" },
    misc: { dur: [0.03, 0.02], sparse: 0.3, hpFreq: 3500, lpFreq: 8000, vol: 0.45 },
    done: { notes: [523, 659, 784, 1047], gap: 0.2, sustain: 0.8, vol: 0.45, wave: "sine", harmonic: "triangle", harmVol: 0.15 },
    err:  { f1: 250, f1End: 80, f2: 200, f2End: 70, sub: 60, subEnd: 35, dur: 0.9, vol: 0.45, w1: "sawtooth", w2: "square", lp1: 800, lp2: 600 },
    warn: { freqs: [880, 740, 620], gap: 0.22, dur: 0.18, vol: 0.45, wave: "square", subDiv: 4, subVol: 0.3, lp: 2500 },
  },

  /* ── Amiga — Paula chip: bright 8-bit, punchy, warm lo-fi ──── */
  amiga: {
    key: { freqs: [800, 650, 1000], qs: [2.5, 2.0, 3.0], dur: 0.04, vol: 0.4, decay: 3.5, filter: "bandpass" },
    tick: { freq: 600, freqEnd: 300, dur: 0.06, vol: 0.3, wave: "square" },
    tool: { pins: [4, 7], pinDur: 0.006, gap: 0.01, hpFreq: [1200, 400], vol: 0.4 },
    llm:  { freq: 330, freqMid: 660, freqEnd: 440, dur: 0.15, vol: 0.4, wave: "square" },
    misc: { dur: [0.025, 0.015], sparse: 0.4, hpFreq: 2000, lpFreq: 5500, vol: 0.45 },
    done: { notes: [440, 554, 659, 880], gap: 0.18, sustain: 0.7, vol: 0.5, wave: "square", harmonic: "sawtooth", harmVol: 0.1 },
    err:  { f1: 200, f1End: 60, f2: 170, f2End: 50, sub: 50, subEnd: 25, dur: 0.8, vol: 0.5, w1: "square", w2: "sawtooth", lp1: 600, lp2: 400 },
    warn: { freqs: [800, 660, 520], gap: 0.2, dur: 0.15, vol: 0.5, wave: "square", subDiv: 4, subVol: 0.35, lp: 2000 },
  },

  /* ── Atari ST — Yamaha YM2149: sharp, tinny, chiptune ─────── */
  atarist: {
    key: { freqs: [1400, 1100, 1700], qs: [3.0, 2.5, 3.5], dur: 0.035, vol: 0.35, decay: 4.0, filter: "bandpass" },
    tick: { freq: 1000, freqEnd: 500, dur: 0.05, vol: 0.3, wave: "square" },
    tool: { pins: [3, 5], pinDur: 0.005, gap: 0.008, hpFreq: [2200, 700], vol: 0.35 },
    llm:  { freq: 500, freqMid: 1000, freqEnd: 700, dur: 0.1, vol: 0.35, wave: "square" },
    misc: { dur: [0.02, 0.01], sparse: 0.25, hpFreq: 4000, lpFreq: 9000, vol: 0.4 },
    done: { notes: [587, 740, 880, 1175], gap: 0.15, sustain: 0.5, vol: 0.4, wave: "square", harmonic: "square", harmVol: 0.08 },
    err:  { f1: 280, f1End: 90, f2: 230, f2End: 80, sub: 70, subEnd: 40, dur: 0.7, vol: 0.4, w1: "square", w2: "square", lp1: 900, lp2: 700 },
    warn: { freqs: [1000, 840, 700], gap: 0.18, dur: 0.14, vol: 0.4, wave: "square", subDiv: 4, subVol: 0.2, lp: 3000 },
  },

  /* ── C64 — SID chip: gritty, warm, resonant bass ──────────── */
  c64: {
    key: { freqs: [600, 500, 750], qs: [3.5, 3.0, 4.0], dur: 0.045, vol: 0.45, decay: 3.0, filter: "bandpass" },
    tick: { freq: 500, freqEnd: 250, dur: 0.07, vol: 0.35, wave: "sawtooth" },
    tool: { pins: [5, 8], pinDur: 0.007, gap: 0.015, hpFreq: [900, 300], vol: 0.45 },
    llm:  { freq: 260, freqMid: 520, freqEnd: 350, dur: 0.18, vol: 0.45, wave: "sawtooth" },
    misc: { dur: [0.035, 0.02], sparse: 0.45, hpFreq: 1500, lpFreq: 4500, vol: 0.5 },
    done: { notes: [392, 494, 587, 784], gap: 0.22, sustain: 0.9, vol: 0.5, wave: "sawtooth", harmonic: "square", harmVol: 0.12 },
    err:  { f1: 180, f1End: 50, f2: 150, f2End: 40, sub: 45, subEnd: 20, dur: 1.0, vol: 0.5, w1: "sawtooth", w2: "square", lp1: 500, lp2: 350 },
    warn: { freqs: [700, 580, 460], gap: 0.25, dur: 0.2, vol: 0.5, wave: "sawtooth", subDiv: 3, subVol: 0.4, lp: 1800 },
  },

  /* ── Classic Mac — soft, warm, friendly beeps ─────────────── */
  mac: {
    key: { freqs: [1000, 800, 1200], qs: [1.5, 1.2, 1.8], dur: 0.04, vol: 0.3, decay: 2.0, filter: "bandpass" },
    tick: { freq: 700, freqEnd: 450, dur: 0.06, vol: 0.2, wave: "sine" },
    tool: { pins: [2, 4], pinDur: 0.008, gap: 0.015, hpFreq: [1500, 500], vol: 0.3 },
    llm:  { freq: 380, freqMid: 700, freqEnd: 500, dur: 0.1, vol: 0.35, wave: "sine" },
    misc: { dur: [0.025, 0.015], sparse: 0.2, hpFreq: 3000, lpFreq: 7000, vol: 0.35 },
    done: { notes: [494, 622, 740, 988], gap: 0.22, sustain: 0.9, vol: 0.4, wave: "sine", harmonic: "sine", harmVol: 0.1 },
    err:  { f1: 220, f1End: 100, f2: 190, f2End: 80, sub: 55, subEnd: 30, dur: 0.7, vol: 0.35, w1: "sine", w2: "triangle", lp1: 700, lp2: 500 },
    warn: { freqs: [740, 622, 520], gap: 0.2, dur: 0.16, vol: 0.35, wave: "sine", subDiv: 4, subVol: 0.2, lp: 2200 },
  },

  /* ── Windows 3.1 — MIDI-like, bright, plasticky ──────────── */
  win31: {
    key: { freqs: [1100, 850, 1350], qs: [2.2, 1.8, 2.5], dur: 0.045, vol: 0.35, decay: 2.8, filter: "bandpass" },
    tick: { freq: 900, freqEnd: 450, dur: 0.07, vol: 0.28, wave: "square" },
    tool: { pins: [3, 6], pinDur: 0.007, gap: 0.011, hpFreq: [1600, 500], vol: 0.38 },
    llm:  { freq: 400, freqMid: 750, freqEnd: 540, dur: 0.13, vol: 0.38, wave: "triangle" },
    misc: { dur: [0.028, 0.018], sparse: 0.3, hpFreq: 3200, lpFreq: 7500, vol: 0.4 },
    done: { notes: [523, 659, 784, 1047], gap: 0.2, sustain: 0.7, vol: 0.42, wave: "triangle", harmonic: "square", harmVol: 0.08 },
    err:  { f1: 240, f1End: 85, f2: 200, f2End: 70, sub: 55, subEnd: 30, dur: 0.8, vol: 0.42, w1: "triangle", w2: "square", lp1: 750, lp2: 550 },
    warn: { freqs: [900, 760, 640], gap: 0.2, dur: 0.16, vol: 0.42, wave: "triangle", subDiv: 4, subVol: 0.25, lp: 2400 },
  },

  /* ── Hacker — dark, glitchy, matrix-like ──────────────────── */
  hacker: {
    key: { freqs: [1600, 1300, 2000], qs: [4.0, 3.5, 4.5], dur: 0.03, vol: 0.3, decay: 5.0, filter: "highpass" },
    tick: { freq: 1200, freqEnd: 600, dur: 0.04, vol: 0.25, wave: "sawtooth" },
    tool: { pins: [4, 8], pinDur: 0.004, gap: 0.006, hpFreq: [2800, 800], vol: 0.3 },
    llm:  { freq: 600, freqMid: 1200, freqEnd: 800, dur: 0.08, vol: 0.3, wave: "sawtooth" },
    misc: { dur: [0.02, 0.015], sparse: 0.5, hpFreq: 5000, lpFreq: 12000, vol: 0.35 },
    done: { notes: [440, 554, 659, 880], gap: 0.12, sustain: 0.4, vol: 0.35, wave: "sawtooth", harmonic: "sawtooth", harmVol: 0.06 },
    err:  { f1: 300, f1End: 60, f2: 260, f2End: 50, sub: 40, subEnd: 20, dur: 1.0, vol: 0.4, w1: "sawtooth", w2: "sawtooth", lp1: 1000, lp2: 800 },
    warn: { freqs: [1100, 900, 750], gap: 0.15, dur: 0.12, vol: 0.38, wave: "sawtooth", subDiv: 5, subVol: 0.2, lp: 3500 },
  },

  /* ── Modern — polished, subtle, UI-grade ──────────────────── */
  modern: {
    key: { freqs: [1300, 1000, 1600], qs: [1.2, 1.0, 1.5], dur: 0.035, vol: 0.25, decay: 2.0, filter: "bandpass" },
    tick: { freq: 900, freqEnd: 500, dur: 0.06, vol: 0.2, wave: "sine" },
    tool: { pins: [2, 4], pinDur: 0.006, gap: 0.01, hpFreq: [2000, 600], vol: 0.25 },
    llm:  { freq: 450, freqMid: 850, freqEnd: 620, dur: 0.1, vol: 0.3, wave: "sine" },
    misc: { dur: [0.02, 0.01], sparse: 0.2, hpFreq: 4000, lpFreq: 10000, vol: 0.3 },
    done: { notes: [554, 698, 831, 1109], gap: 0.22, sustain: 1.0, vol: 0.35, wave: "sine", harmonic: "sine", harmVol: 0.12 },
    err:  { f1: 230, f1End: 90, f2: 195, f2End: 75, sub: 55, subEnd: 30, dur: 0.7, vol: 0.35, w1: "sine", w2: "triangle", lp1: 700, lp2: 500 },
    warn: { freqs: [830, 700, 590], gap: 0.22, dur: 0.16, vol: 0.35, wave: "sine", subDiv: 4, subVol: 0.2, lp: 2200 },
  },

  /* ── macOS — refined, airy, Aqua-era glass tones ──────────── */
  macos: {
    key: { freqs: [1100, 900, 1400], qs: [1.3, 1.1, 1.6], dur: 0.04, vol: 0.28, decay: 2.0, filter: "bandpass" },
    tick: { freq: 750, freqEnd: 420, dur: 0.07, vol: 0.22, wave: "sine" },
    tool: { pins: [2, 4], pinDur: 0.007, gap: 0.012, hpFreq: [1800, 550], vol: 0.28 },
    llm:  { freq: 440, freqMid: 820, freqEnd: 600, dur: 0.12, vol: 0.32, wave: "sine" },
    misc: { dur: [0.022, 0.012], sparse: 0.2, hpFreq: 3800, lpFreq: 9500, vol: 0.32 },
    done: { notes: [587, 740, 880, 1175], gap: 0.24, sustain: 1.1, vol: 0.38, wave: "sine", harmonic: "triangle", harmVol: 0.15 },
    err:  { f1: 210, f1End: 95, f2: 180, f2End: 80, sub: 50, subEnd: 28, dur: 0.75, vol: 0.35, w1: "sine", w2: "sine", lp1: 650, lp2: 480 },
    warn: { freqs: [780, 660, 560], gap: 0.22, dur: 0.17, vol: 0.35, wave: "sine", subDiv: 4, subVol: 0.2, lp: 2100 },
  },

  /* ── iPhone — iOS haptic-paired, taps and plinks ──────────── */
  iphone: {
    key: { freqs: [1500, 1200, 1800], qs: [1.0, 0.8, 1.2], dur: 0.025, vol: 0.25, decay: 1.8, filter: "bandpass" },
    tick: { freq: 1000, freqEnd: 600, dur: 0.04, vol: 0.2, wave: "sine" },
    tool: { pins: [2, 3], pinDur: 0.005, gap: 0.008, hpFreq: [2200, 700], vol: 0.22 },
    llm:  { freq: 500, freqMid: 900, freqEnd: 680, dur: 0.08, vol: 0.28, wave: "sine" },
    misc: { dur: [0.015, 0.01], sparse: 0.15, hpFreq: 4500, lpFreq: 11000, vol: 0.28 },
    done: { notes: [659, 831, 988, 1319], gap: 0.18, sustain: 0.6, vol: 0.32, wave: "sine", harmonic: "sine", harmVol: 0.1 },
    err:  { f1: 260, f1End: 110, f2: 220, f2End: 90, sub: 65, subEnd: 35, dur: 0.6, vol: 0.3, w1: "sine", w2: "triangle", lp1: 750, lp2: 550 },
    warn: { freqs: [950, 800, 670], gap: 0.18, dur: 0.13, vol: 0.32, wave: "sine", subDiv: 5, subVol: 0.15, lp: 2800 },
  },

  /* ── Android — Material You: rounder, deeper, tactile ─────── */
  android: {
    key: { freqs: [950, 750, 1150], qs: [1.5, 1.2, 1.8], dur: 0.04, vol: 0.3, decay: 2.2, filter: "bandpass" },
    tick: { freq: 650, freqEnd: 350, dur: 0.07, vol: 0.25, wave: "triangle" },
    tool: { pins: [3, 5], pinDur: 0.007, gap: 0.012, hpFreq: [1400, 450], vol: 0.32 },
    llm:  { freq: 350, freqMid: 700, freqEnd: 500, dur: 0.13, vol: 0.35, wave: "triangle" },
    misc: { dur: [0.028, 0.015], sparse: 0.25, hpFreq: 2800, lpFreq: 7500, vol: 0.38 },
    done: { notes: [466, 587, 698, 932], gap: 0.22, sustain: 0.9, vol: 0.4, wave: "triangle", harmonic: "sine", harmVol: 0.12 },
    err:  { f1: 200, f1End: 75, f2: 170, f2End: 60, sub: 50, subEnd: 25, dur: 0.85, vol: 0.4, w1: "triangle", w2: "square", lp1: 650, lp2: 450 },
    warn: { freqs: [750, 630, 530], gap: 0.22, dur: 0.18, vol: 0.4, wave: "triangle", subDiv: 4, subVol: 0.25, lp: 2000 },
  },

  /* ── Nokia 7110 — tiny speaker: buzzy, thin, monophonic ───── */
  nokia7110: {
    key: { freqs: [2000, 1700, 2400], qs: [5.0, 4.5, 5.5], dur: 0.025, vol: 0.35, decay: 6.0, filter: "highpass" },
    tick: { freq: 1400, freqEnd: 800, dur: 0.035, vol: 0.3, wave: "square" },
    tool: { pins: [2, 4], pinDur: 0.004, gap: 0.005, hpFreq: [3000, 1000], vol: 0.35 },
    llm:  { freq: 700, freqMid: 1400, freqEnd: 1000, dur: 0.07, vol: 0.35, wave: "square" },
    misc: { dur: [0.015, 0.01], sparse: 0.35, hpFreq: 6000, lpFreq: 14000, vol: 0.38 },
    done: { notes: [784, 988, 1175, 1568], gap: 0.12, sustain: 0.35, vol: 0.4, wave: "square", harmonic: "square", harmVol: 0.05 },
    err:  { f1: 350, f1End: 120, f2: 300, f2End: 100, sub: 80, subEnd: 50, dur: 0.5, vol: 0.4, w1: "square", w2: "square", lp1: 1200, lp2: 900 },
    warn: { freqs: [1200, 1000, 840], gap: 0.14, dur: 0.1, vol: 0.42, wave: "square", subDiv: 6, subVol: 0.15, lp: 4000 },
  },

  /* ── Nokia Communicator — richer than 7110, dual-speaker feel */
  nokiacomm: {
    key: { freqs: [1600, 1300, 1900], qs: [3.5, 3.0, 4.0], dur: 0.035, vol: 0.35, decay: 4.0, filter: "bandpass" },
    tick: { freq: 1100, freqEnd: 600, dur: 0.05, vol: 0.28, wave: "square" },
    tool: { pins: [3, 5], pinDur: 0.005, gap: 0.008, hpFreq: [2400, 700], vol: 0.35 },
    llm:  { freq: 550, freqMid: 1100, freqEnd: 780, dur: 0.1, vol: 0.35, wave: "triangle" },
    misc: { dur: [0.02, 0.012], sparse: 0.3, hpFreq: 4500, lpFreq: 10000, vol: 0.38 },
    done: { notes: [659, 831, 988, 1319], gap: 0.16, sustain: 0.5, vol: 0.4, wave: "triangle", harmonic: "square", harmVol: 0.08 },
    err:  { f1: 300, f1End: 100, f2: 250, f2End: 85, sub: 70, subEnd: 40, dur: 0.65, vol: 0.4, w1: "square", w2: "triangle", lp1: 1000, lp2: 750 },
    warn: { freqs: [1000, 840, 700], gap: 0.17, dur: 0.14, vol: 0.4, wave: "square", subDiv: 5, subVol: 0.2, lp: 3200 },
  },
};

/** Get sound profile for current theme, falling back to win11 */
function _snd() {
  return _themeSoundProfiles[currentTheme] || _themeSoundProfiles.win11;
}

/* ── Typewriter ──────────────────────────────────────────────── */

function _playTypewriterKey() {
  if (soundMode !== "typewriter") return;
  const ctx = _getAudioCtx();
  const now = ctx.currentTime;
  const p = _snd().key;
  const variant = Math.floor(Math.random() * 3);

  const bufLen = Math.floor(ctx.sampleRate * p.dur);
  const buf = ctx.createBuffer(1, bufLen, ctx.sampleRate);
  const data = buf.getChannelData(0);
  for (let i = 0; i < bufLen; i++) {
    data[i] = (Math.random() * 2 - 1) * Math.pow(1 - i / bufLen, p.decay);
  }

  const noise = ctx.createBufferSource();
  noise.buffer = buf;

  const filter = ctx.createBiquadFilter();
  filter.type = p.filter;
  filter.frequency.value = p.freqs[variant];
  filter.Q.value = p.qs[variant];

  const gain = ctx.createGain();
  gain.gain.setValueAtTime(p.vol, now);
  gain.gain.exponentialRampToValueAtTime(0.001, now + p.dur);

  noise.connect(filter);
  filter.connect(gain);
  gain.connect(ctx.destination);

  noise.start(now);
  noise.stop(now + p.dur);
}

let _lastTypewriterTime = 0;
const _typewriterMinInterval = 40;

function playTypewriterSound() {
  const now = performance.now();
  if (now - _lastTypewriterTime < _typewriterMinInterval) return;
  _lastTypewriterTime = now;
  _playTypewriterKey();
}

/* ── Modem ───────────────────────────────────────────────────── */

/* Modem dial-up handshake: plays the classic sequence of tones and noise
   that continues looping until stopModemSound() is called. */
let _modemNodes = null;  // { oscs, gains, noise, master, ctx } — active modem session

function startModemSound() {
  if (soundMode !== "modem") return;
  stopModemSound();

  const ctx = _getAudioCtx();
  const now = ctx.currentTime;
  const master = ctx.createGain();
  master.gain.setValueAtTime(0.3, now);
  master.connect(ctx.destination);

  // Helper: schedule an oscillator with gain envelope.
  function _osc(type, freqEvents, gainEvents, startT, stopT) {
    const o = ctx.createOscillator();
    o.type = type;
    for (const [method, val, t] of freqEvents) {
      if (method === "set") o.frequency.setValueAtTime(val, now + t);
      else o.frequency.linearRampToValueAtTime(val, now + t);
    }
    const g = ctx.createGain();
    for (const [method, val, t] of gainEvents) {
      if (method === "set") g.gain.setValueAtTime(val, now + t);
      else g.gain.linearRampToValueAtTime(val, now + t);
    }
    o.connect(g); g.connect(master);
    o.start(now + startT); o.stop(now + stopT);
  }

  /* ── Phase 1 (0–2.4s): Dial tone ─────────────────────────────
     Classic US dial tone: 350 Hz + 440 Hz continuous. */
  _osc("sine",
    [["set", 350, 0]],
    [["set", 0.5, 0], ["set", 0.5, 2.2], ["ramp", 0, 2.4]],
    0, 2.4);
  _osc("sine",
    [["set", 440, 0]],
    [["set", 0.5, 0], ["set", 0.5, 2.2], ["ramp", 0, 2.4]],
    0, 2.4);

  /* ── Phase 2 (2.5–7s): DTMF dialing — "phone number" touch tones
     Real DTMF uses two simultaneous freqs per digit. */
  const dtmfPairs = [
    [941, 1336], // 0-row tones
    [697, 1209], [697, 1336], [697, 1477],
    [770, 1209], [770, 1336], [770, 1477],
    [852, 1209],
  ];
  const digitDur = 0.12;
  const digitGap = 0.08;
  let dt = 2.6;
  // Dial 10 digits.
  for (let d = 0; d < 10; d++) {
    const pair = dtmfPairs[Math.floor(Math.random() * dtmfPairs.length)];
    _osc("sine",
      [["set", pair[0], dt]],
      [["set", 0.4, dt], ["set", 0.4, dt + digitDur - 0.01], ["ramp", 0, dt + digitDur]],
      dt, dt + digitDur);
    _osc("sine",
      [["set", pair[1], dt]],
      [["set", 0.4, dt], ["set", 0.4, dt + digitDur - 0.01], ["ramp", 0, dt + digitDur]],
      dt, dt + digitDur);
    dt += digitDur + digitGap;
  }

  /* ── Phase 3 (5.2–7s): Ringing — US ringback (440+480 Hz, 2s on 4s off,
     we just do one partial ring) */
  const ringStart = dt + 0.3;
  _osc("sine",
    [["set", 440, ringStart]],
    [["set", 0.3, ringStart], ["set", 0.3, ringStart + 1.8], ["ramp", 0, ringStart + 2.0]],
    ringStart, ringStart + 2.0);
  _osc("sine",
    [["set", 480, ringStart]],
    [["set", 0.3, ringStart], ["set", 0.3, ringStart + 1.8], ["ramp", 0, ringStart + 2.0]],
    ringStart, ringStart + 2.0);

  /* ── Phase 4 (ring+2.2 – +5.5s): CNG tone (calling fax/modem) →
     answering CED tone. The classic 1100 Hz "beeeee" then 2100 Hz "EEEEE". */
  const p4 = ringStart + 2.2;
  // CNG: 1100 Hz, pulsed 0.5s on / 0.5s off × 3
  for (let i = 0; i < 3; i++) {
    const t = p4 + i * 1.0;
    _osc("sine",
      [["set", 1100, t]],
      [["set", 0.45, t], ["set", 0.45, t + 0.45], ["ramp", 0, t + 0.5]],
      t, t + 0.5);
  }
  // CED: 2100 Hz continuous answer tone.
  const cedStart = p4 + 3.2;
  _osc("sine",
    [["set", 2100, cedStart]],
    [["set", 0, cedStart], ["ramp", 0.5, cedStart + 0.1],
     ["set", 0.5, cedStart + 2.5], ["ramp", 0, cedStart + 2.8]],
    cedStart, cedStart + 2.8);

  /* ── Phase 5 (ced+3 – +7s): V.32 handshake — the iconic screech.
     Multiple carriers probing, alternating tones, scrambled training. */
  const p5 = cedStart + 3.0;

  // Carrier 1: sweeping 1800→900→2400→1200 Hz (sawtooth for gritty harmonics).
  _osc("sawtooth",
    [["set", 1800, p5], ["ramp", 900, p5 + 1.0],
     ["ramp", 2400, p5 + 2.0], ["ramp", 1200, p5 + 3.0],
     ["ramp", 1800, p5 + 4.0]],
    [["set", 0, p5], ["ramp", 0.12, p5 + 0.2],
     ["set", 0.12, p5 + 3.8], ["ramp", 0, p5 + 4.0]],
    p5, p5 + 4.0);

  // Carrier 2: counter-sweep.
  _osc("sawtooth",
    [["set", 2400, p5 + 0.1], ["ramp", 1650, p5 + 1.5],
     ["ramp", 980, p5 + 2.5], ["ramp", 2100, p5 + 3.5]],
    [["set", 0, p5], ["ramp", 0.10, p5 + 0.3],
     ["set", 0.10, p5 + 3.5], ["ramp", 0, p5 + 4.0]],
    p5 + 0.1, p5 + 4.0);

  // Carrier 3: rapid alternating tones simulating scrambler training.
  const scrStart = p5 + 1.0;
  const scrEnd = p5 + 3.5;
  const scrOsc = ctx.createOscillator();
  scrOsc.type = "square";
  // Rapid frequency stepping.
  const scrFreqs = [1200, 2400, 1800, 600, 2100, 980, 1650, 2400, 1200, 1800,
                    2100, 600, 1650, 980, 2400, 1200, 1800, 2100, 980, 1650];
  const scrStep = (scrEnd - scrStart) / scrFreqs.length;
  for (let i = 0; i < scrFreqs.length; i++) {
    scrOsc.frequency.setValueAtTime(scrFreqs[i], now + scrStart + i * scrStep);
  }
  const scrGain = ctx.createGain();
  scrGain.gain.setValueAtTime(0, now);
  scrGain.gain.setValueAtTime(0.06, now + scrStart);
  scrGain.gain.setValueAtTime(0.06, now + scrEnd - 0.1);
  scrGain.gain.linearRampToValueAtTime(0, now + scrEnd);
  scrOsc.connect(scrGain); scrGain.connect(master);
  scrOsc.start(now + scrStart); scrOsc.stop(now + scrEnd);

  /* ── Phase 6 (ongoing): Data transfer noise ──────────────────
     Multi-layered: FSK-like tone modulation + shaped noise + warble.
     This is the sustained "shhhhkrkrkrkr" of an active connection. */
  const p6 = p5 + 4.2;

  // Layer 1: FSK data carrier — two tones rapidly switching (like V.21).
  const fskBufDur = 6;
  const fskBufLen = Math.floor(ctx.sampleRate * fskBufDur);
  const fskBuf = ctx.createBuffer(1, fskBufLen, ctx.sampleRate);
  const fd = fskBuf.getChannelData(0);
  for (let i = 0; i < fskBufLen; i++) {
    const t = i / ctx.sampleRate;
    // Pseudo-random bit switching between mark (1200Hz) and space (2200Hz).
    const bit = Math.sin(t * 2 * Math.PI * 37.5) > 0 ? 1 : 0;  // ~37.5 baud switching
    const freq = bit ? 1200 : 2200;
    fd[i] = Math.sin(t * 2 * Math.PI * freq) * 0.5;
  }
  const fskNode = ctx.createBufferSource();
  fskNode.buffer = fskBuf;
  fskNode.loop = true;
  const fskGain = ctx.createGain();
  fskGain.gain.setValueAtTime(0, now);
  fskGain.gain.setValueAtTime(0, now + p6 - 0.1);
  fskGain.gain.linearRampToValueAtTime(0.22, now + p6 + 0.5);
  fskNode.connect(fskGain); fskGain.connect(master);
  fskNode.start(now + p6 - 0.1);

  // Layer 2: Broadband data noise — filtered, with amplitude modulation
  // simulating bursty data packets.
  const noiseDur = 6;
  const noiseBufLen = Math.floor(ctx.sampleRate * noiseDur);
  const noiseBuf = ctx.createBuffer(1, noiseBufLen, ctx.sampleRate);
  const nd = noiseBuf.getChannelData(0);
  for (let i = 0; i < noiseBufLen; i++) {
    const t = i / ctx.sampleRate;
    // Irregular amplitude envelope: mix of several modulation rates.
    const mod = 0.3
      + 0.25 * Math.sin(t * 2 * Math.PI * 5.3)
      + 0.20 * Math.sin(t * 2 * Math.PI * 13.7)
      + 0.15 * Math.sin(t * 2 * Math.PI * 31.1)
      + 0.10 * Math.sin(t * 2 * Math.PI * 67.3);
    nd[i] = (Math.random() * 2 - 1) * Math.max(0, mod);
  }
  const noiseNode = ctx.createBufferSource();
  noiseNode.buffer = noiseBuf;
  noiseNode.loop = true;

  // Dual bandpass: telephone band 300–3400 Hz.
  const noiseBP1 = ctx.createBiquadFilter();
  noiseBP1.type = "bandpass";
  noiseBP1.frequency.value = 1700;
  noiseBP1.Q.value = 0.6;

  // Second narrower peak around the carrier.
  const noiseBP2 = ctx.createBiquadFilter();
  noiseBP2.type = "peaking";
  noiseBP2.frequency.value = 1800;
  noiseBP2.gain.value = 8;
  noiseBP2.Q.value = 2.0;

  const noiseGain = ctx.createGain();
  noiseGain.gain.setValueAtTime(0, now);
  noiseGain.gain.setValueAtTime(0, now + p6 - 0.1);
  noiseGain.gain.linearRampToValueAtTime(0.3, now + p6 + 0.5);

  noiseNode.connect(noiseBP1);
  noiseBP1.connect(noiseBP2);
  noiseBP2.connect(noiseGain);
  noiseGain.connect(master);
  noiseNode.start(now + p6 - 0.1);

  // Layer 3: Slow warbling carrier — gives the characteristic
  // "woooo-woooo" pitch drift of a real connection.
  const warble = ctx.createOscillator();
  warble.type = "sine";
  warble.frequency.setValueAtTime(1800, now + p6);
  // Slow LFO-driven pitch modulation.
  const lfo = ctx.createOscillator();
  lfo.type = "sine";
  lfo.frequency.value = 0.4;  // slow wobble
  const lfoGain = ctx.createGain();
  lfoGain.gain.value = 80;  // ±80 Hz deviation
  lfo.connect(lfoGain);
  lfoGain.connect(warble.frequency);

  const warbleGain = ctx.createGain();
  warbleGain.gain.setValueAtTime(0, now);
  warbleGain.gain.setValueAtTime(0, now + p6);
  warbleGain.gain.linearRampToValueAtTime(0.08, now + p6 + 1.0);

  warble.connect(warbleGain); warbleGain.connect(master);
  lfo.start(now + p6); warble.start(now + p6);

  _modemNodes = {
    sources: [fskNode, noiseNode, warble, lfo],
    gains: [fskGain, noiseGain, warbleGain],
    master, ctx
  };
}

function stopModemSound() {
  if (!_modemNodes) return;
  const { sources, gains, master, ctx } = _modemNodes;
  const now = ctx.currentTime;
  try {
    // Fade all layers out over 400ms.
    for (const g of gains) {
      g.gain.cancelScheduledValues(now);
      g.gain.setValueAtTime(g.gain.value, now);
      g.gain.linearRampToValueAtTime(0, now + 0.4);
    }
    for (const s of sources) {
      try { s.stop(now + 0.5); } catch (_) {}
    }
  } catch (_) {}
  _modemNodes = null;
}

/* ── Activity sounds (7 types, randomly selected per task) ──── */

let _activeActivitySound = null; // name of current sound for this task

const _activitySounds = {

  /* 1. Soft click / Geiger counter tick */
  softClick() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const buf = ctx.createBuffer(1, ctx.sampleRate * 0.025, ctx.sampleRate);
    const d = buf.getChannelData(0);
    for (let i = 0; i < d.length; i++) {
      d[i] = (Math.random() * 2 - 1) * Math.exp(-i / (d.length * 0.08));
    }
    const src = ctx.createBufferSource();
    src.buffer = buf;
    const bp = ctx.createBiquadFilter();
    bp.type = "bandpass";
    bp.frequency.value = 2800 + Math.random() * 800;
    bp.Q.value = 3;
    const g = ctx.createGain();
    g.gain.setValueAtTime(0.45, now);
    g.gain.exponentialRampToValueAtTime(0.001, now + 0.025);
    src.connect(bp); bp.connect(g); g.connect(ctx.destination);
    src.start(now);
  },

  /* 2. Dot matrix printer — short burst of pin impacts */
  dotMatrix() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const pins = 3 + Math.floor(Math.random() * 4); // 3-6 pin strikes
    for (let i = 0; i < pins; i++) {
      const t = now + i * 0.012;
      const buf = ctx.createBuffer(1, ctx.sampleRate * 0.008, ctx.sampleRate);
      const d = buf.getChannelData(0);
      for (let j = 0; j < d.length; j++) {
        d[j] = (Math.random() * 2 - 1) * Math.exp(-j / (d.length * 0.15));
      }
      const src = ctx.createBufferSource();
      src.buffer = buf;
      const hp = ctx.createBiquadFilter();
      hp.type = "highpass";
      hp.frequency.value = 1800 + Math.random() * 600;
      const g = ctx.createGain();
      g.gain.setValueAtTime(0.35, t);
      g.gain.exponentialRampToValueAtTime(0.001, t + 0.008);
      src.connect(hp); hp.connect(g); g.connect(ctx.destination);
      src.start(t);
    }
  },

  /* 3. Terminal bell — classic \a beep */
  terminalBell() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const osc = ctx.createOscillator();
    osc.type = "square";
    osc.frequency.setValueAtTime(1760, now);
    const g = ctx.createGain();
    g.gain.setValueAtTime(0.18, now);
    g.gain.setValueAtTime(0.18, now + 0.04);
    g.gain.exponentialRampToValueAtTime(0.001, now + 0.1);
    const lp = ctx.createBiquadFilter();
    lp.type = "lowpass";
    lp.frequency.value = 3000;
    osc.connect(lp); lp.connect(g); g.connect(ctx.destination);
    osc.start(now);
    osc.stop(now + 0.1);
  },

  /* 4. Bubble / pop — chat message arrival */
  bubblePop() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const osc = ctx.createOscillator();
    osc.type = "sine";
    osc.frequency.setValueAtTime(420, now);
    osc.frequency.exponentialRampToValueAtTime(820, now + 0.03);
    osc.frequency.exponentialRampToValueAtTime(580, now + 0.08);
    const g = ctx.createGain();
    g.gain.setValueAtTime(0.4, now);
    g.gain.exponentialRampToValueAtTime(0.001, now + 0.12);
    osc.connect(g); g.connect(ctx.destination);
    osc.start(now);
    osc.stop(now + 0.12);
  },

  /* 5. Morse code dit — very short tone pip */
  morseDit() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const osc = ctx.createOscillator();
    osc.type = "sine";
    osc.frequency.value = 700 + Math.random() * 100;
    const g = ctx.createGain();
    g.gain.setValueAtTime(0, now);
    g.gain.linearRampToValueAtTime(0.35, now + 0.005);
    g.gain.setValueAtTime(0.35, now + 0.04);
    g.gain.exponentialRampToValueAtTime(0.001, now + 0.07);
    osc.connect(g); g.connect(ctx.destination);
    osc.start(now);
    osc.stop(now + 0.07);
  },

  /* 6. Hard drive seek — mechanical head movement chatter */
  hardDriveSeek() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const steps = 2 + Math.floor(Math.random() * 3); // 2-4 seek steps
    for (let i = 0; i < steps; i++) {
      const t = now + i * 0.035;
      // Click from head movement
      const buf = ctx.createBuffer(1, ctx.sampleRate * 0.015, ctx.sampleRate);
      const d = buf.getChannelData(0);
      for (let j = 0; j < d.length; j++) {
        const env = Math.exp(-j / (d.length * 0.06));
        d[j] = (Math.random() * 2 - 1) * env;
        // Add a sharp transient at the start
        if (j < 8) d[j] += (Math.random() * 2 - 1) * 0.8;
      }
      const src = ctx.createBufferSource();
      src.buffer = buf;
      const bp = ctx.createBiquadFilter();
      bp.type = "bandpass";
      bp.frequency.value = 800 + Math.random() * 400;
      bp.Q.value = 1.5;
      const g = ctx.createGain();
      g.gain.setValueAtTime(0.5, t);
      g.gain.exponentialRampToValueAtTime(0.001, t + 0.015);
      src.connect(bp); bp.connect(g); g.connect(ctx.destination);
      src.start(t);
    }
  },

  /* 7. CRT scan line — tiny electrical zap/crackle */
  crtScanLine() {
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    const dur = 0.03 + Math.random() * 0.02;
    const buf = ctx.createBuffer(1, ctx.sampleRate * dur, ctx.sampleRate);
    const d = buf.getChannelData(0);
    for (let i = 0; i < d.length; i++) {
      const env = Math.exp(-i / (d.length * 0.12));
      // Crackling: sparse random impulses
      d[i] = (Math.random() < 0.3 ? (Math.random() * 2 - 1) : 0) * env;
    }
    const src = ctx.createBufferSource();
    src.buffer = buf;
    const hp = ctx.createBiquadFilter();
    hp.type = "highpass";
    hp.frequency.value = 3500;
    const lp = ctx.createBiquadFilter();
    lp.type = "lowpass";
    lp.frequency.value = 8000;
    const g = ctx.createGain();
    g.gain.setValueAtTime(0.45, now);
    g.gain.exponentialRampToValueAtTime(0.001, now + dur);
    src.connect(hp); hp.connect(lp); lp.connect(g); g.connect(ctx.destination);
    src.start(now);
  },
};

const _activitySoundNames = Object.keys(_activitySounds);

/** Pick activity sound — no longer needed but kept for modem start */
function pickActivitySound() {
  _activeActivitySound = "crtScanLine";
}

/** Play themed activity sound based on log entry type:
 *  tool → dot-matrix style (themed)
 *  thinking/stream → bubble/pop (themed)
 *  everything else → crackle/zap (themed) */
function playActivitySound(logType) {
  if (soundMode === "off") return;
  try {
    const s = _snd();
    if (logType === "tool") {
      _playThemedTool(s.tool);
    } else if (logType === "thinking" || logType === "stream") {
      _playThemedLlm(s.llm);
    } else {
      _playThemedMisc(s.misc);
    }
  } catch (_) {}
}

/** Themed tool sound — pin-strike burst */
function _playThemedTool(p) {
  const ctx = _getAudioCtx();
  const now = ctx.currentTime;
  const pins = p.pins[0] + Math.floor(Math.random() * (p.pins[1] - p.pins[0] + 1));
  for (let i = 0; i < pins; i++) {
    const t = now + i * p.gap;
    const buf = ctx.createBuffer(1, ctx.sampleRate * p.pinDur, ctx.sampleRate);
    const d = buf.getChannelData(0);
    for (let j = 0; j < d.length; j++) {
      d[j] = (Math.random() * 2 - 1) * Math.exp(-j / (d.length * 0.15));
    }
    const src = ctx.createBufferSource();
    src.buffer = buf;
    const hp = ctx.createBiquadFilter();
    hp.type = "highpass";
    hp.frequency.value = p.hpFreq[0] + Math.random() * p.hpFreq[1];
    const g = ctx.createGain();
    g.gain.setValueAtTime(p.vol, t);
    g.gain.exponentialRampToValueAtTime(0.001, t + p.pinDur);
    src.connect(hp); hp.connect(g); g.connect(ctx.destination);
    src.start(t);
  }
}

/** Themed LLM sound — bubble/pop tone sweep */
function _playThemedLlm(p) {
  const ctx = _getAudioCtx();
  const now = ctx.currentTime;
  const osc = ctx.createOscillator();
  osc.type = p.wave;
  osc.frequency.setValueAtTime(p.freq, now);
  osc.frequency.exponentialRampToValueAtTime(p.freqMid, now + p.dur * 0.25);
  osc.frequency.exponentialRampToValueAtTime(p.freqEnd, now + p.dur * 0.67);
  const g = ctx.createGain();
  g.gain.setValueAtTime(p.vol, now);
  g.gain.exponentialRampToValueAtTime(0.001, now + p.dur);
  osc.connect(g); g.connect(ctx.destination);
  osc.start(now);
  osc.stop(now + p.dur);
}

/** Themed misc sound — crackle/zap */
function _playThemedMisc(p) {
  const ctx = _getAudioCtx();
  const now = ctx.currentTime;
  const dur = p.dur[0] + Math.random() * p.dur[1];
  const buf = ctx.createBuffer(1, ctx.sampleRate * dur, ctx.sampleRate);
  const d = buf.getChannelData(0);
  for (let i = 0; i < d.length; i++) {
    const env = Math.exp(-i / (d.length * 0.12));
    d[i] = (Math.random() < p.sparse ? (Math.random() * 2 - 1) : 0) * env;
  }
  const src = ctx.createBufferSource();
  src.buffer = buf;
  const hp = ctx.createBiquadFilter();
  hp.type = "highpass";
  hp.frequency.value = p.hpFreq;
  const lp = ctx.createBiquadFilter();
  lp.type = "lowpass";
  lp.frequency.value = p.lpFreq;
  const g = ctx.createGain();
  g.gain.setValueAtTime(p.vol, now);
  g.gain.exponentialRampToValueAtTime(0.001, now + dur);
  src.connect(hp); hp.connect(lp); lp.connect(g); g.connect(ctx.destination);
  src.start(now);
}

/* ── Public API called by streaming code ─────────────────────── */

function playStreamChunkSound() {
  if (soundMode === "typewriter") playTypewriterSound();
  // Modem: continuous sound already running, nothing per-chunk.
}

function playStreamStartSound() {
  pickActivitySound(); // new random activity sound each task
  if (soundMode === "modem") startModemSound();
}

function playStreamStopSound() {
  if (soundMode === "modem") stopModemSound();
}

/* ── Notification sounds (completion / error / warning) ──────── */

/** Task completion: rising arpeggio chime (themed) */
function playCompletionSound() {
  if (soundMode === "off") return;
  try {
    const p = _snd().done;
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    for (let i = 0; i < p.notes.length; i++) {
      const freq = p.notes[i];
      const t = now + i * p.gap;
      const vol = p.vol * (0.85 + i * 0.05); // slight crescendo
      // Main tone
      const osc = ctx.createOscillator();
      osc.type = p.wave;
      osc.frequency.setValueAtTime(freq, t);
      const g = ctx.createGain();
      g.gain.setValueAtTime(0, now);
      g.gain.setValueAtTime(vol, t);
      g.gain.exponentialRampToValueAtTime(vol * 0.5, t + p.sustain * 0.35);
      g.gain.exponentialRampToValueAtTime(0.001, t + p.sustain);
      osc.connect(g); g.connect(ctx.destination);
      osc.start(t); osc.stop(t + p.sustain);
      // Harmonic overtone
      const h = ctx.createOscillator();
      h.type = p.harmonic;
      h.frequency.setValueAtTime(freq * 2, t);
      const gh = ctx.createGain();
      gh.gain.setValueAtTime(0, now);
      gh.gain.setValueAtTime(p.harmVol, t);
      gh.gain.exponentialRampToValueAtTime(0.001, t + p.sustain * 0.7);
      h.connect(gh); gh.connect(ctx.destination);
      h.start(t); h.stop(t + p.sustain * 0.7);
    }
  } catch (_) {}
}

/** Error: descending dissonant buzz (themed) */
function playErrorSound() {
  if (soundMode === "off") return;
  try {
    const p = _snd().err;
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    // Primary descending tone
    const o1 = ctx.createOscillator();
    o1.type = p.w1;
    o1.frequency.setValueAtTime(p.f1, now);
    o1.frequency.exponentialRampToValueAtTime(p.f1End, now + p.dur * 0.9);
    const g1 = ctx.createGain();
    g1.gain.setValueAtTime(p.vol, now);
    g1.gain.exponentialRampToValueAtTime(p.vol * 0.33, now + p.dur * 0.55);
    g1.gain.exponentialRampToValueAtTime(0.001, now + p.dur);
    const lp = ctx.createBiquadFilter();
    lp.type = "lowpass"; lp.frequency.value = p.lp1;
    o1.connect(lp); lp.connect(g1); g1.connect(ctx.destination);
    o1.start(now); o1.stop(now + p.dur);
    // Dissonant second tone
    const o2 = ctx.createOscillator();
    o2.type = p.w2;
    o2.frequency.setValueAtTime(p.f2, now);
    o2.frequency.exponentialRampToValueAtTime(p.f2End, now + p.dur * 0.78);
    const g2 = ctx.createGain();
    g2.gain.setValueAtTime(p.vol * 0.55, now + 0.05);
    g2.gain.exponentialRampToValueAtTime(p.vol * 0.18, now + p.dur * 0.55);
    g2.gain.exponentialRampToValueAtTime(0.001, now + p.dur * 0.9);
    const lp2 = ctx.createBiquadFilter();
    lp2.type = "lowpass"; lp2.frequency.value = p.lp2;
    o2.connect(lp2); lp2.connect(g2); g2.connect(ctx.destination);
    o2.start(now + 0.05); o2.stop(now + p.dur * 0.9);
    // Sub-bass rumble
    const sub = ctx.createOscillator();
    sub.type = "sine";
    sub.frequency.setValueAtTime(p.sub, now);
    sub.frequency.exponentialRampToValueAtTime(p.subEnd, now + p.dur * 0.9);
    const gs = ctx.createGain();
    gs.gain.setValueAtTime(p.vol * 0.78, now);
    gs.gain.exponentialRampToValueAtTime(0.001, now + p.dur);
    sub.connect(gs); gs.connect(ctx.destination);
    sub.start(now); sub.stop(now + p.dur);
  } catch (_) {}
}

/** Warning/blocked: descending beep sequence (themed) */
function playWarningSound() {
  if (soundMode === "off") return;
  try {
    const p = _snd().warn;
    const ctx = _getAudioCtx();
    const now = ctx.currentTime;
    for (let i = 0; i < p.freqs.length; i++) {
      const t = now + i * p.gap;
      const freq = p.freqs[i];
      // Main tone
      const osc = ctx.createOscillator();
      osc.type = p.wave;
      osc.frequency.setValueAtTime(freq, t);
      const g = ctx.createGain();
      g.gain.setValueAtTime(p.vol, t);
      g.gain.setValueAtTime(p.vol, t + p.dur * 0.55);
      g.gain.exponentialRampToValueAtTime(0.001, t + p.dur);
      const lp = ctx.createBiquadFilter();
      lp.type = "lowpass"; lp.frequency.value = p.lp;
      osc.connect(lp); lp.connect(g); g.connect(ctx.destination);
      osc.start(t); osc.stop(t + p.dur);
      // Sub-bass layer
      const sub = ctx.createOscillator();
      sub.type = "sine";
      sub.frequency.setValueAtTime(freq / p.subDiv, t);
      const gs = ctx.createGain();
      gs.gain.setValueAtTime(p.subVol, t);
      gs.gain.exponentialRampToValueAtTime(0.001, t + p.dur);
      sub.connect(gs); gs.connect(ctx.destination);
      sub.start(t); sub.stop(t + p.dur);
    }
  } catch (_) {}
}

/* ── Init ────────────────────────────────────────────────────── */

/* ── Generic toolbar popup helper ────────────────────────────── */

/** Close any open toolbar popups */
function _closeAllPopups() {
  document.querySelectorAll(".toolbar-popup").forEach((p) => p.classList.add("hidden"));
}

function _initToolbarPopup(btnId, panelId, optClass, dataAttr, values, {
  defaultVal, storageKey, onSelect, labelFn
}) {
  const btn = $("#" + btnId);
  const panel = $("#" + panelId);
  if (!btn || !panel) return;
  const opts = panel.querySelectorAll("." + optClass);
  let current = defaultVal;

  const saved = storageKey ? localStorage.getItem(storageKey) : null;
  if (saved && values.includes(saved)) current = saved;

  _mark(current);
  if (labelFn) btn.textContent = labelFn(current);
  if (onSelect) onSelect(current, true); // initial load

  btn.addEventListener("click", (e) => {
    e.stopPropagation();
    const open = !panel.classList.contains("hidden");
    _closeAllPopups();
    if (open) return;
    const r = btn.getBoundingClientRect();
    panel.style.top = r.bottom + 2 + "px";
    panel.style.right = (window.innerWidth - r.right) + "px";
    panel.classList.remove("hidden");
  });

  opts.forEach((o) => {
    o.addEventListener("click", (e) => {
      e.stopPropagation();
      const val = o.dataset[dataAttr];
      current = val;
      _mark(val);
      if (labelFn) btn.textContent = labelFn(val);
      if (storageKey) localStorage.setItem(storageKey, val);
      if (onSelect) onSelect(val, false);
      panel.classList.add("hidden");
    });
  });

  document.addEventListener("click", () => panel.classList.add("hidden"));

  function _mark(val) {
    opts.forEach((o) => o.classList.toggle("active", o.dataset[dataAttr] === val));
  }
}

/* ── Tier popup ──────────────────────────────────────────────── */

const _tierLabels = { micro: "📐 Micro", minimal: "📐 Minimal", standard: "📐 Standard", generous: "📐 Generous" };

function initTierPopup() {
  _initToolbarPopup("tier-btn", "tier-panel", "tier-opt", "tier",
    ["micro", "minimal", "standard", "generous"], {
      defaultVal: selectedTokenTier,
      storageKey: "computer-token-tier",
      labelFn: (v) => _tierLabels[v] || "📐 " + v,
      onSelect: (v, isInit) => {
        selectedTokenTier = v;
        if (!isInit) logEntry("system", `Token tier: ${v}`);
      }
    });
}

/* ── Sound popup ─────────────────────────────────────────────── */

const _soundLabels = { off: "🔇 Off", typewriter: "🔊 Typewriter", modem: "🔊 Modem" };

/* ── Text size ───────────────────────────────────────────────── */

const _textsizeLabels = { standard: "Aa Standard", large: "Aa Large", larger: "Aa Larger", laaaarger: "Aa Laaaarger" };

function initTextSize() {
  _initToolbarPopup("textsize-btn", "textsize-panel", "textsize-opt", "size",
    ["standard", "large", "larger", "laaaarger"], {
      defaultVal: "standard",
      storageKey: "computer-textsize",
      labelFn: (v) => _textsizeLabels[v] || "Aa " + v,
      onSelect: (v) => setTextSize(v)
    });
}

function setTextSize(size) {
  document.body.classList.remove("textsize-large", "textsize-larger", "textsize-laaaarger");
  if (size === "large") document.body.classList.add("textsize-large");
  else if (size === "larger") document.body.classList.add("textsize-larger");
  else if (size === "laaaarger") document.body.classList.add("textsize-laaaarger");
}

function initSound() {
  _initToolbarPopup("sound-btn", "sound-panel", "sound-opt", "sound",
    ["off", "typewriter", "modem"], {
      defaultVal: soundMode,
      storageKey: "computer-sound",
      labelFn: (v) => _soundLabels[v] || "🔊 " + v,
      onSelect: (v, isInit) => {
        soundMode = v;
        // Stop modem if switching away while it's playing.
        if (!isInit && v !== "modem") stopModemSound();
      }
    });
}

/* ── Approval modal ──────────────────────────────────────────── */

let _currentApprovalId = null;
const _approvedPlaybooks = new Set();  // Track already-approved playbooks this session

function showApprovalModal(id, message, category) {
  // Auto-approve if this playbook was already approved in this session.
  if (category === "playbook") {
    // Extract playbook name(s) from message for dedup key.
    const key = message.replace(/\s+/g, " ").trim();
    if (_approvedPlaybooks.has(key)) {
      send({ type: "approval_response", id, approved: true });
      logEntry("system", `✓ Auto-approved (already approved): ${category}`);
      return;
    }
    // Store for future dedup after user approves.
    _currentApprovalKey = key;
  }

  _currentApprovalId = id;
  const msgEl = $("#approval-message");
  if (msgEl) msgEl.innerHTML = markdownToHtml(message);
  $("#approval-modal").classList.add("active");
  logEntry("system", `⏳ Approval requested: ${category}`);
}

let _currentApprovalKey = null;

function respondApproval(approved) {
  if (_currentApprovalId) {
    send({ type: "approval_response", id: _currentApprovalId, approved });
    logEntry("system", approved ? "✓ Approved" : "✕ Declined");
    // Remember approval so we don't ask again for the same playbook.
    if (approved && _currentApprovalKey) {
      _approvedPlaybooks.add(_currentApprovalKey);
    }
  }
  _currentApprovalId = null;
  _currentApprovalKey = null;
  $("#approval-modal").classList.remove("active");
}

function initApprovalModal() {
  const approveBtn = $("#approval-approve-btn");
  const declineBtn = $("#approval-decline-btn");
  const backdrop = $("#approval-modal .modal-backdrop");
  if (approveBtn) approveBtn.addEventListener("click", () => respondApproval(true));
  if (declineBtn) declineBtn.addEventListener("click", () => respondApproval(false));
  // Backdrop click = decline (safe default).
  if (backdrop) backdrop.addEventListener("click", () => respondApproval(false));
  document.addEventListener("keydown", (e) => {
    if (!$("#approval-modal").classList.contains("active")) return;
    if (e.key === "Enter") { e.preventDefault(); respondApproval(true); }
    if (e.key === "Escape") { e.preventDefault(); respondApproval(false); }
  });
}

/* ── Playbook selector ────────────────────────────────────────── */

async function loadPlaybooks() {
  try {
    const res = await fetch("/api/playbooks");
    if (!res.ok) return;
    availablePlaybooks = await res.json();
    renderPlaybookGrid();
    updatePlaybookBtnLabel();
    console.log(`[Computer] Loaded ${availablePlaybooks.length} playbooks`);
  } catch (e) {
    console.warn("Failed to load playbooks:", e);
  }
}

function renderPlaybookGrid() {
  const grid = $("#playbook-grid");
  if (!grid) return;
  grid.innerHTML = "";

  // Auto card.
  grid.appendChild(_pbCard("", "📋", "Auto", "System picks best match"));

  // None card.
  grid.appendChild(_pbCard("__none__", "🚫", "None", "No playbook guidance"));

  // Group by task_type.
  if (availablePlaybooks.length) {
    const grouped = {};
    availablePlaybooks.forEach(pb => {
      const t = pb.task_type || "other";
      if (!grouped[t]) grouped[t] = [];
      grouped[t].push(pb);
    });
    for (const [type, pbs] of Object.entries(grouped).sort()) {
      const label = document.createElement("div");
      label.className = "pb-section-label";
      label.textContent = type;
      grid.appendChild(label);
      pbs.forEach(pb => {
        const desc = pb.trigger_description || pb.tags || "";
        grid.appendChild(_pbCard(pb.id, "📋", pb.name, desc));
      });
    }
  }
}

function _pbCard(pbid, icon, name, detail) {
  const card = document.createElement("div");
  card.className = "model-card" + (selectedPlaybookId === pbid ? " selected" : "");
  card.dataset.pbid = pbid;
  card.innerHTML = `
    <div class="model-card-icon">${icon}</div>
    <div class="model-card-info">
      <div class="model-card-name">${name}</div>
      <div class="model-card-detail">${detail}</div>
    </div>`;
  card.addEventListener("click", () => selectPlaybook(pbid));
  return card;
}

function selectPlaybook(pbid) {
  selectedPlaybookId = pbid;
  localStorage.setItem("computer-playbook", pbid);
  updatePlaybookBtnLabel();
  renderPlaybookGrid();
  closePlaybookModal();

  if (connected) {
    send({ type: "set_playbook", playbook_id: pbid });
  }
  const label = pbid === "__none__" ? "None"
    : pbid ? (availablePlaybooks.find(p => p.id === pbid)?.name || pbid)
    : "Auto";
  logEntry("system", `Playbook: ${label}`);
}

function updatePlaybookBtnLabel() {
  const label = $("#playbook-btn-label");
  if (!label) return;
  if (!selectedPlaybookId) {
    label.textContent = "Auto";
  } else if (selectedPlaybookId === "__none__") {
    label.textContent = "None";
  } else {
    const pb = availablePlaybooks.find(p => p.id === selectedPlaybookId);
    label.textContent = pb ? pb.name : selectedPlaybookId.slice(0, 8);
  }
}

function openPlaybookModal() {
  renderPlaybookGrid();
  $("#playbook-modal").classList.add("active");
}

function closePlaybookModal() {
  $("#playbook-modal").classList.remove("active");
}

function initPlaybookModal() {
  $("#playbook-btn").addEventListener("click", openPlaybookModal);
  $("#playbook-modal .modal-close").addEventListener("click", closePlaybookModal);
  $("#playbook-modal .modal-backdrop").addEventListener("click", closePlaybookModal);
  document.addEventListener("keydown", (e) => {
    if (e.key === "Escape" && $("#playbook-modal").classList.contains("active")) {
      closePlaybookModal();
    }
  });
}

/* ═══════════════════════════════════════════════════════════════
   INIT
   ═══════════════════════════════════════════════════════════════ */

document.addEventListener("DOMContentLoaded", async () => {
  // Restore theme.
  const saved = localStorage.getItem("computer-theme");
  if (saved && saved !== currentTheme) {
    setTheme(saved);
  }

  // Theme modal.
  initThemeModal();

  // Init subsystems.
  initTextarea();
  initTabs();
  initTodoTab();
  initDataTab();
  initInsightsTab();
  initResize();
  initClearLog();
  initClearSession();
  initSessionSettings();
  initSpinner();
  initFullscreen();
  initStreamPanel();
  initMap();
  initFiles();
  initFilePreview();
  initSelectors();
  initAttachments();
  initFolderModal();
  initSound();
  initTextSize();

  // Load available models, personas, and playbooks for selectors.
  loadModels();
  loadPersonas();
  loadPlaybooks();

  // Listen for postMessage from visual iframe.
  window.addEventListener("message", handleIframeMessage);

  // Mobile titlebar toggle.
  const _tbToggle = document.getElementById("titlebar-toggle");
  const _tbRight = document.getElementById("titlebar-right");
  if (_tbToggle && _tbRight) {
    _tbToggle.addEventListener("click", (e) => {
      e.stopPropagation();
      _tbRight.classList.toggle("open");
    });
    // Close menu when clicking outside.
    document.addEventListener("click", (e) => {
      if (!_tbRight.contains(e.target) && e.target !== _tbToggle) {
        _tbRight.classList.remove("open");
      }
    });
    // Close menu after selecting an item.
    _tbRight.addEventListener("click", () => {
      setTimeout(() => _tbRight.classList.remove("open"), 150);
    });
    _tbRight.addEventListener("change", () => {
      setTimeout(() => _tbRight.classList.remove("open"), 150);
    });
  }

  // Boot sequence.
  await runBoot();

  // Connect to agent.
  connect();

  // Focus input.
  $("#input-box").focus();
});
