@jackwener/opencli 1.6.0 → 1.6.2

package/dist/clis/gemini/utils.js

@@ -1,3 +1,4 @@
+import { CommandExecutionError } from '../../errors.js';
 export const GEMINI_DOMAIN = 'gemini.google.com';
 export const GEMINI_APP_URL = 'https://gemini.google.com/app';
 const GEMINI_RESPONSE_NOISE_PATTERNS = [
@@ -6,6 +7,57 @@ const GEMINI_RESPONSE_NOISE_PATTERNS = [
     /Google Privacy Policy/gi,
     /Opens in a new window/gi,
 ];
+const GEMINI_TRANSCRIPT_CHROME_MARKERS = ['gemini', '我的内容', '对话', 'google terms', 'google privacy policy'];
+const GEMINI_COMPOSER_SELECTORS = [
+    '.ql-editor[contenteditable="true"]',
+    '.ql-editor[role="textbox"]',
+    '.ql-editor[aria-label*="Gemini"]',
+    '[contenteditable="true"][aria-label*="Gemini"]',
+    '[aria-label="Enter a prompt for Gemini"]',
+    '[aria-label*="prompt for Gemini"]',
+];
+const GEMINI_COMPOSER_MARKER_ATTR = 'data-opencli-gemini-composer';
+const GEMINI_COMPOSER_PREPARE_ATTEMPTS = 4;
+const GEMINI_COMPOSER_PREPARE_WAIT_SECONDS = 1;
+function buildGeminiComposerLocatorScript() {
+    const selectorsJson = JSON.stringify(GEMINI_COMPOSER_SELECTORS);
+    const markerAttrJson = JSON.stringify(GEMINI_COMPOSER_MARKER_ATTR);
+    return `
+      const isVisible = (el) => {
+        if (!(el instanceof HTMLElement)) return false;
+        const style = window.getComputedStyle(el);
+        if (style.display === 'none' || style.visibility === 'hidden') return false;
+        const rect = el.getBoundingClientRect();
+        return rect.width > 0 && rect.height > 0;
+      };
+
+      const markerAttr = ${markerAttrJson};
+      const clearComposerMarkers = (active) => {
+        document.querySelectorAll('[' + markerAttr + ']').forEach((node) => {
+          if (node !== active) node.removeAttribute(markerAttr);
+        });
+      };
+
+      const markComposer = (node) => {
+        if (!(node instanceof HTMLElement)) return null;
+        clearComposerMarkers(node);
+        node.setAttribute(markerAttr, '1');
+        return node;
+      };
+
+      const findComposer = () => {
+        const marked = document.querySelector('[' + markerAttr + '="1"]');
+        if (marked instanceof HTMLElement && isVisible(marked)) return marked;
+
+        const selectors = ${selectorsJson};
+        for (const selector of selectors) {
+          const node = Array.from(document.querySelectorAll(selector)).find((candidate) => candidate instanceof HTMLElement && isVisible(candidate));
+          if (node instanceof HTMLElement) return markComposer(node);
+        }
+        return null;
+      };
+  `;
+}
 export function sanitizeGeminiResponseText(value, promptText) {
     let sanitized = value;
     for (const pattern of GEMINI_RESPONSE_NOISE_PATTERNS) {
@@ -29,31 +81,146 @@ export function collectGeminiTranscriptAdditions(beforeLines, currentLines, prom
     const beforeSet = new Set(beforeLines);
     const additions = currentLines
         .filter((line) => !beforeSet.has(line))
-        .map((line) => sanitizeGeminiResponseText(line, promptText))
+        .map((line) => extractGeminiTranscriptLineCandidate(line, promptText))
         .filter((line) => line && line !== promptText);
     return additions.join('\n').trim();
 }
+export function collapseAdjacentGeminiTurns(turns) {
+    const collapsed = [];
+    for (const turn of turns) {
+        if (!turn || typeof turn.Role !== 'string' || typeof turn.Text !== 'string')
+            continue;
+        const previous = collapsed.at(-1);
+        if (previous?.Role === turn.Role && previous.Text === turn.Text)
+            continue;
+        collapsed.push(turn);
+    }
+    return collapsed;
+}
+function hasGeminiTurnPrefix(before, current) {
+    if (before.length > current.length)
+        return false;
+    return before.every((turn, index) => (turn.Role === current[index]?.Role
+        && turn.Text === current[index]?.Text));
+}
+function findLastMatchingGeminiTurnIndex(turns, target) {
+    if (!target)
+        return null;
+    for (let index = turns.length - 1; index >= 0; index -= 1) {
+        const turn = turns[index];
+        if (turn?.Role === target.Role && turn.Text === target.Text) {
+            return index;
+        }
+    }
+    return null;
+}
+function diffTrustedStructuredTurns(before, current) {
+    if (!before.structuredTurnsTrusted || !current.structuredTurnsTrusted) {
+        return {
+            appendedTurns: [],
+            hasTrustedAppend: false,
+            hasNewUserTurn: false,
+            hasNewAssistantTurn: false,
+        };
+    }
+    if (!hasGeminiTurnPrefix(before.turns, current.turns)) {
+        return {
+            appendedTurns: [],
+            hasTrustedAppend: false,
+            hasNewUserTurn: false,
+            hasNewAssistantTurn: false,
+        };
+    }
+    const appendedTurns = current.turns.slice(before.turns.length);
+    return {
+        appendedTurns,
+        hasTrustedAppend: appendedTurns.length > 0,
+        hasNewUserTurn: appendedTurns.some((turn) => turn.Role === 'User'),
+        hasNewAssistantTurn: appendedTurns.some((turn) => turn.Role === 'Assistant'),
+    };
+}
+function diffTranscriptLines(before, current) {
+    const beforeLines = new Set(before.transcriptLines);
+    return current.transcriptLines.filter((line) => !beforeLines.has(line));
+}
+function isLikelyGeminiTranscriptChrome(line) {
+    const lower = line.toLowerCase();
+    const markerHits = GEMINI_TRANSCRIPT_CHROME_MARKERS.filter((marker) => lower.includes(marker)).length;
+    return markerHits >= 2;
+}
+function extractGeminiTranscriptLineCandidate(transcriptLine, promptText) {
+    const candidate = transcriptLine.trim();
+    if (!candidate)
+        return '';
+    const prompt = promptText.trim();
+    const sanitized = sanitizeGeminiResponseText(candidate, promptText);
+    if (!prompt)
+        return sanitized;
+    if (!candidate.includes(prompt))
+        return sanitized;
+    if (sanitized && sanitized !== prompt && sanitized !== candidate)
+        return sanitized;
+    if (isLikelyGeminiTranscriptChrome(candidate))
+        return '';
+    // Some transcript snapshots flatten "prompt + answer" into a single line.
+    // Recover the answer only when the line starts with the current prompt.
+    if (candidate.startsWith(prompt)) {
+        const tail = candidate.slice(prompt.length).replace(/^[\s:：,，-]+/, '').trim();
+        return tail ? sanitizeGeminiResponseText(tail, '') : '';
+    }
+    return sanitized;
+}
 function getStateScript() {
     return `
     (() => {
+      ${buildGeminiComposerLocatorScript()}
+
       const signInNode = Array.from(document.querySelectorAll('a, button')).find((node) => {
         const text = (node.textContent || '').trim().toLowerCase();
         const aria = (node.getAttribute('aria-label') || '').trim().toLowerCase();
         const href = node.getAttribute('href') || '';
         return text === 'sign in'
           || aria === 'sign in'
+          || text === '登录'
+          || aria === '登录'
           || href.includes('accounts.google.com/ServiceLogin');
       });
 
-      const composer = document.querySelector('[aria-label="Enter a prompt for Gemini"], [aria-label*="prompt for Gemini"], .ql-editor[aria-label*="Gemini"], [contenteditable="true"][aria-label*="Gemini"]');
-      const sendButton = document.querySelector('button[aria-label="Send message"]');
+      const composer = findComposer();
 
       return {
         url: window.location.href,
         title: document.title || '',
         isSignedIn: signInNode ? false : (composer ? true : null),
         composerLabel: composer?.getAttribute('aria-label') || '',
-        canSend: !!(sendButton && !sendButton.disabled),
+        canSend: !!composer,
+      };
+    })()
+  `;
+}
+function readGeminiSnapshotScript() {
+    return `
+    (() => {
+      ${buildGeminiComposerLocatorScript()}
+      const composer = findComposer();
+      const composerText = composer?.textContent?.replace(/\\u00a0/g, ' ').trim() || '';
+      const isGenerating = !!Array.from(document.querySelectorAll('button, [role="button"]')).find((node) => {
+        const text = (node.textContent || '').trim().toLowerCase();
+        const aria = (node.getAttribute('aria-label') || '').trim().toLowerCase();
+        return text === 'stop response'
+          || aria === 'stop response'
+          || text === '停止回答'
+          || aria === '停止回答';
+      });
+      const turns = ${getTurnsScript().trim()};
+      const transcriptLines = ${getTranscriptLinesScript().trim()};
+
+      return {
+        turns,
+        transcriptLines,
+        composerHasText: composerText.length > 0,
+        isGenerating,
+        structuredTurnsTrusted: turns.length > 0 || transcriptLines.length === 0,
       };
     })()
   `;
@@ -159,7 +326,16 @@ function getTurnsScript() {
       ];
 
       const roots = selectors.flatMap((selector) => Array.from(document.querySelectorAll(selector)));
-      const unique = roots.filter((el, index, all) => all.indexOf(el) === index).filter(isVisible);
+      const unique = roots
+        .filter((el, index, all) => all.indexOf(el) === index)
+        .filter(isVisible)
+        .sort((left, right) => {
+          if (left === right) return 0;
+          const relation = left.compareDocumentPosition(right);
+          if (relation & Node.DOCUMENT_POSITION_FOLLOWING) return -1;
+          if (relation & Node.DOCUMENT_POSITION_PRECEDING) return 1;
+          return 0;
+        });
 
       const turns = unique.map((el) => {
         const text = clean(el.innerText || el.textContent || '');
@@ -179,52 +355,184 @@ function getTurnsScript() {
         return role ? { Role: role, Text: text } : null;
       }).filter(Boolean);
 
-      const deduped = [];
-      const seen = new Set();
-      for (const turn of turns) {
-        const key = turn.Role + '::' + turn.Text;
-        if (seen.has(key)) continue;
-        seen.add(key);
-        deduped.push(turn);
-      }
-      return deduped;
+      return turns;
     })()
   `;
 }
-function fillAndSubmitComposerScript(text) {
+function prepareComposerScript() {
     return `
-    ((inputText) => {
-      const cleanInsert = (el) => {
-        if (!(el instanceof HTMLElement)) throw new Error('Composer is not editable');
-        el.focus();
+    (() => {
+      ${buildGeminiComposerLocatorScript()}
+      const composer = findComposer();
+
+      if (!(composer instanceof HTMLElement)) {
+        return { ok: false, reason: 'Could not find Gemini composer' };
+      }
+
+      try {
+        composer.focus();
         const selection = window.getSelection();
         const range = document.createRange();
-        range.selectNodeContents(el);
+        range.selectNodeContents(composer);
         range.collapse(false);
         selection?.removeAllRanges();
         selection?.addRange(range);
-        el.textContent = '';
-        document.execCommand('insertText', false, inputText);
-        el.dispatchEvent(new InputEvent('input', { bubbles: true, data: inputText, inputType: 'insertText' }));
+        composer.textContent = '';
+        composer.dispatchEvent(new InputEvent('input', { bubbles: true, data: '', inputType: 'deleteContentBackward' }));
+      } catch (error) {
+        return {
+          ok: false,
+          reason: error instanceof Error ? error.message : String(error),
+        };
+      }
+
+      return {
+        ok: true,
+        label: composer.getAttribute('aria-label') || '',
       };
+    })()
+  `;
+}
+function composerHasTextScript() {
+    return `
+    (() => {
+      ${buildGeminiComposerLocatorScript()}
+      const composer = findComposer();
+
+      return {
+        hasText: !!(composer && ((composer.textContent || '').trim() || (composer.innerText || '').trim())),
+      };
+    })()
+  `;
+}
+function insertComposerTextFallbackScript(text) {
+    return `
+    ((inputText) => {
+      ${buildGeminiComposerLocatorScript()}
+      const composer = findComposer();
+
+      if (!(composer instanceof HTMLElement)) {
+        return { hasText: false, reason: 'Could not find Gemini composer' };
+      }
+
+      const selection = window.getSelection();
+      const range = document.createRange();
+      range.selectNodeContents(composer);
+      range.collapse(false);
+      selection?.removeAllRanges();
+      selection?.addRange(range);
+
+      composer.focus();
+      composer.textContent = '';
+      const execResult = typeof document.execCommand === 'function'
+        ? document.execCommand('insertText', false, inputText)
+        : false;
+
+      if (!execResult) {
+        const paragraph = document.createElement('p');
+        const lines = String(inputText).split(/\\n/);
+        for (const [index, line] of lines.entries()) {
+          if (index > 0) paragraph.appendChild(document.createElement('br'));
+          paragraph.appendChild(document.createTextNode(line));
+        }
+        composer.replaceChildren(paragraph);
+      }
+
+      composer.dispatchEvent(new InputEvent('beforeinput', { bubbles: true, data: inputText, inputType: 'insertText' }));
+      composer.dispatchEvent(new InputEvent('input', { bubbles: true, data: inputText, inputType: 'insertText' }));
+      composer.dispatchEvent(new Event('change', { bubbles: true }));
+
+      return {
+        hasText: !!((composer.textContent || '').trim() || (composer.innerText || '').trim()),
+      };
+    })(${JSON.stringify(text)})
+  `;
+}
+function submitComposerScript() {
+    return `
+    (() => {
+      ${buildGeminiComposerLocatorScript()}
+      const composer = findComposer();
 
-      const composer = document.querySelector('[aria-label="Enter a prompt for Gemini"], [aria-label*="prompt for Gemini"], .ql-editor[aria-label*="Gemini"], [contenteditable="true"][aria-label*="Gemini"]');
       if (!(composer instanceof HTMLElement)) {
         throw new Error('Could not find Gemini composer');
       }
 
-      cleanInsert(composer);
+      const composerRect = composer.getBoundingClientRect();
+      const rootCandidates = [
+        composer.closest('form'),
+        composer.closest('[role="form"]'),
+        composer.closest('.input-area-container'),
+        composer.closest('.textbox-container'),
+        composer.closest('.input-wrapper'),
+        composer.parentElement,
+        composer.parentElement?.parentElement,
+      ].filter(Boolean);
+
+      const seen = new Set();
+      const buttons = [];
+      for (const root of rootCandidates) {
+        root.querySelectorAll('button, [role="button"]').forEach((node) => {
+          if (!(node instanceof HTMLElement)) return;
+          if (seen.has(node)) return;
+          seen.add(node);
+          buttons.push(node);
+        });
+      }
+
+      const excludedPattern = /main menu|主菜单|microphone|麦克风|upload|上传|mode|模式|tools|工具|settings|临时对话|new chat|新对话/i;
+      const submitPattern = /send|发送|submit|提交/i;
+      let bestButton = null;
+      let bestScore = -1;
+
+      for (const button of buttons) {
+        if (!isVisible(button)) continue;
+        if (button instanceof HTMLButtonElement && button.disabled) continue;
+        if (button.getAttribute('aria-disabled') === 'true') continue;
+
+        const label = ((button.getAttribute('aria-label') || '') + ' ' + ((button.textContent || '').trim())).trim();
+        if (excludedPattern.test(label)) continue;
+
+        const rect = button.getBoundingClientRect();
+        const verticalDistance = Math.abs((rect.top + rect.bottom) / 2 - (composerRect.top + composerRect.bottom) / 2);
+        if (verticalDistance > 160) continue;
+
+        let score = 0;
+        if (submitPattern.test(label)) score += 10;
+        if (rect.left >= composerRect.right - 160) score += 3;
+        if (rect.left >= composerRect.left) score += 1;
+        if (rect.width <= 96 && rect.height <= 96) score += 1;
 
-      const sendButton = document.querySelector('button[aria-label="Send message"]');
-      if (sendButton instanceof HTMLButtonElement && !sendButton.disabled) {
-        sendButton.click();
+        if (score > bestScore) {
+          bestScore = score;
+          bestButton = button;
+        }
+      }
+
+      if (bestButton instanceof HTMLElement && bestScore >= 3) {
+        bestButton.click();
         return 'button';
       }
 
-      composer.dispatchEvent(new KeyboardEvent('keydown', { key: 'Enter', code: 'Enter', keyCode: 13, bubbles: true }));
-      composer.dispatchEvent(new KeyboardEvent('keyup', { key: 'Enter', code: 'Enter', keyCode: 13, bubbles: true }));
       return 'enter';
-    })(${JSON.stringify(text)})
+    })()
+  `;
+}
+function dispatchComposerEnterScript() {
+    return `
+    (() => {
+      ${buildGeminiComposerLocatorScript()}
+      const composer = findComposer();
+
+      if (!(composer instanceof HTMLElement)) {
+        throw new Error('Could not find Gemini composer');
+      }
+
+      composer.focus();
+      composer.dispatchEvent(new KeyboardEvent('keydown', { key: 'Enter', code: 'Enter', keyCode: 13, which: 13, bubbles: true }));
+      composer.dispatchEvent(new KeyboardEvent('keyup', { key: 'Enter', code: 'Enter', keyCode: 13, which: 13, bubbles: true }));
+      return 'enter';
+    })()
   `;
 }
 function clickNewChatScript() {
@@ -241,7 +549,14 @@ function clickNewChatScript() {
       const candidates = Array.from(document.querySelectorAll('button, a')).filter((node) => {
         const text = (node.textContent || '').trim().toLowerCase();
         const aria = (node.getAttribute('aria-label') || '').trim().toLowerCase();
-        return isVisible(node) && (text === 'new chat' || aria === 'new chat');
+        return isVisible(node) && (
+          text === 'new chat'
+          || aria === 'new chat'
+          || text === '发起新对话'
+          || aria === '发起新对话'
+          || text === '新对话'
+          || aria === '新对话'
+        );
       });
 
       const target = candidates.find((node) => !node.hasAttribute('disabled')) || candidates[0];
@@ -288,23 +603,133 @@ export async function startNewGeminiChat(page) {
     return action;
 }
 export async function getGeminiVisibleTurns(page) {
-    await ensureGeminiPage(page);
-    const turns = await page.evaluate(getTurnsScript());
+    const turns = await getGeminiStructuredTurns(page);
     if (Array.isArray(turns) && turns.length > 0)
         return turns;
     const lines = await getGeminiTranscriptLines(page);
     return lines.map((line) => ({ Role: 'System', Text: line }));
 }
+async function getGeminiStructuredTurns(page) {
+    await ensureGeminiPage(page);
+    const turns = collapseAdjacentGeminiTurns(await page.evaluate(getTurnsScript()));
+    return Array.isArray(turns) ? turns : [];
+}
 export async function getGeminiTranscriptLines(page) {
     await ensureGeminiPage(page);
     return await page.evaluate(getTranscriptLinesScript());
 }
+export async function readGeminiSnapshot(page) {
+    await ensureGeminiPage(page);
+    return await page.evaluate(readGeminiSnapshotScript());
+}
+function findLastUserTurnIndex(turns) {
+    for (let index = turns.length - 1; index >= 0; index -= 1) {
+        if (turns[index]?.Role === 'User')
+            return index;
+    }
+    return null;
+}
+function findLastUserTurn(turns) {
+    const index = findLastUserTurnIndex(turns);
+    return index === null ? null : turns[index] ?? null;
+}
+export async function waitForGeminiSubmission(page, before, timeoutSeconds) {
+    const preSendAssistantCount = before.turns.filter((turn) => turn.Role === 'Assistant').length;
+    const maxPolls = Math.max(1, Math.ceil(timeoutSeconds));
+    for (let index = 0; index < maxPolls; index += 1) {
+        await page.wait(index === 0 ? 0.5 : 1);
+        const current = await readGeminiSnapshot(page);
+        const structuredAppend = diffTrustedStructuredTurns(before, current);
+        const transcriptDelta = diffTranscriptLines(before, current);
+        if (structuredAppend.hasTrustedAppend && structuredAppend.hasNewUserTurn) {
+            return {
+                snapshot: current,
+                preSendAssistantCount,
+                userAnchorTurn: findLastUserTurn(current.turns),
+                reason: 'user_turn',
+            };
+        }
+        if (!current.composerHasText && current.isGenerating) {
+            return {
+                snapshot: current,
+                preSendAssistantCount,
+                userAnchorTurn: findLastUserTurn(current.turns),
+                reason: 'composer_generating',
+            };
+        }
+        if (!current.composerHasText && transcriptDelta.length > 0) {
+            return {
+                snapshot: current,
+                preSendAssistantCount,
+                userAnchorTurn: findLastUserTurn(current.turns),
+                reason: 'composer_transcript',
+            };
+        }
+    }
+    return null;
+}
 export async function sendGeminiMessage(page, text) {
     await ensureGeminiPage(page);
-    const submittedBy = await page.evaluate(fillAndSubmitComposerScript(text));
+    let prepared;
+    for (let attempt = 0; attempt < GEMINI_COMPOSER_PREPARE_ATTEMPTS; attempt += 1) {
+        prepared = await page.evaluate(prepareComposerScript());
+        if (prepared?.ok)
+            break;
+        if (attempt < GEMINI_COMPOSER_PREPARE_ATTEMPTS - 1)
+            await page.wait(GEMINI_COMPOSER_PREPARE_WAIT_SECONDS);
+    }
+    if (!prepared?.ok) {
+        throw new CommandExecutionError(prepared?.reason || 'Could not find Gemini composer');
+    }
+    let hasText = false;
+    if (page.nativeType) {
+        try {
+            await page.nativeType(text);
+            await page.wait(0.2);
+            const nativeState = await page.evaluate(composerHasTextScript());
+            hasText = !!nativeState?.hasText;
+        }
+        catch { }
+    }
+    if (!hasText) {
+        const fallbackState = await page.evaluate(insertComposerTextFallbackScript(text));
+        hasText = !!fallbackState?.hasText;
+    }
+    if (!hasText) {
+        throw new CommandExecutionError('Failed to insert text into Gemini composer');
+    }
+    const submitAction = await page.evaluate(submitComposerScript());
+    if (submitAction === 'button') {
+        await page.wait(1);
+        return 'button';
+    }
+    if (page.nativeKeyPress) {
+        try {
+            await page.nativeKeyPress('Enter');
+        }
+        catch {
+            await page.evaluate(dispatchComposerEnterScript());
+        }
+    }
+    else {
+        await page.evaluate(dispatchComposerEnterScript());
+    }
     await page.wait(1);
-    return submittedBy;
+    return 'enter';
 }
+export const __test__ = {
+    GEMINI_COMPOSER_SELECTORS,
+    GEMINI_COMPOSER_MARKER_ATTR,
+    collapseAdjacentGeminiTurns,
+    clickNewChatScript,
+    diffTranscriptLines,
+    diffTrustedStructuredTurns,
+    hasGeminiTurnPrefix,
+    readGeminiSnapshot,
+    readGeminiSnapshotScript,
+    submitComposerScript,
+    insertComposerTextFallbackScript,
+};
 export async function getGeminiVisibleImageUrls(page) {
     await ensureGeminiPage(page);
     return await page.evaluate(`
@@ -429,35 +854,77 @@ export async function exportGeminiImages(page, urls) {
     })(${urlsJson})
   `);
 }
-export async function waitForGeminiResponse(page, beforeLines, promptText, timeoutSeconds) {
-    const getCandidate = async () => {
-        const turns = await getGeminiVisibleTurns(page);
-        const assistantCandidate = [...turns].reverse().find((turn) => turn.Role === 'Assistant');
-        const visibleCandidate = assistantCandidate
-            ? sanitizeGeminiResponseText(assistantCandidate.Text, promptText)
-            : '';
-        if (visibleCandidate && visibleCandidate !== promptText)
-            return visibleCandidate;
-        const lines = await getGeminiTranscriptLines(page);
-        return collectGeminiTranscriptAdditions(beforeLines, lines, promptText);
+export async function waitForGeminiResponse(page, baseline, promptText, timeoutSeconds) {
+    if (timeoutSeconds <= 0)
+        return '';
+    // Reply ownership must survive Gemini prepending older history later.
+    // Re-anchor on the submitted user turn when possible, and otherwise only
+    // accept assistants that are appended to the exact submission snapshot.
+    const pickStructuredReplyCandidate = (current) => {
+        if (!current.structuredTurnsTrusted)
+            return '';
+        const userAnchorTurnIndex = findLastMatchingGeminiTurnIndex(current.turns, baseline.userAnchorTurn);
+        if (userAnchorTurnIndex !== null) {
+            const candidate = current.turns
+                .slice(userAnchorTurnIndex + 1)
+                .filter((turn) => turn.Role === 'Assistant')
+                .at(-1);
+            return candidate ? sanitizeGeminiResponseText(candidate.Text, promptText) : '';
+        }
+        if (hasGeminiTurnPrefix(baseline.snapshot.turns, current.turns)) {
+            const appendedAssistant = current.turns
+                .slice(baseline.snapshot.turns.length)
+                .filter((turn) => turn.Role === 'Assistant')
+                .at(-1);
+            if (appendedAssistant) {
+                return sanitizeGeminiResponseText(appendedAssistant.Text, promptText);
+            }
+        }
+        return '';
     };
-    const pollIntervalSeconds = 2;
-    const maxPolls = Math.max(1, Math.ceil(timeoutSeconds / pollIntervalSeconds));
-    let lastCandidate = '';
-    let stableCount = 0;
+    const pickFallbackGeminiTranscriptReply = (current) => current.transcriptLines
+        .filter((line) => !baseline.snapshot.transcriptLines.includes(line))
+        .map((line) => extractGeminiTranscriptLineCandidate(line, promptText))
+        .filter(Boolean)
+        .join('\n')
+        .trim();
+    const maxPolls = Math.max(1, Math.ceil(timeoutSeconds / 2));
+    let lastStructured = '';
+    let structuredStableCount = 0;
+    let lastTranscript = '';
+    let transcriptStableCount = 0;
+    let transcriptMissCount = 0;
     for (let index = 0; index < maxPolls; index += 1) {
-        await page.wait(index === 0 ? 1.5 : pollIntervalSeconds);
-        const candidate = await getCandidate();
-        if (!candidate)
+        await page.wait(index === 0 ? 1 : 2);
+        const current = await readGeminiSnapshot(page);
+        const structuredCandidate = pickStructuredReplyCandidate(current);
+        if (structuredCandidate) {
+            if (structuredCandidate === lastStructured)
+                structuredStableCount += 1;
+            else {
+                lastStructured = structuredCandidate;
+                structuredStableCount = 1;
+            }
+            if (!current.isGenerating && structuredStableCount >= 2) {
+                return structuredCandidate;
+            }
             continue;
-        if (candidate === lastCandidate)
-            stableCount += 1;
+        }
+        transcriptMissCount += 1;
+        if (transcriptMissCount < 2)
+            continue;
+        const transcriptCandidate = pickFallbackGeminiTranscriptReply(current);
+        if (!transcriptCandidate)
+            continue;
+        if (transcriptCandidate === lastTranscript)
+            transcriptStableCount += 1;
         else {
-            lastCandidate = candidate;
-            stableCount = 1;
+            lastTranscript = transcriptCandidate;
+            transcriptStableCount = 1;
+        }
+        if (!current.isGenerating && transcriptStableCount >= 2) {
+            return transcriptCandidate;
         }
-        if (stableCount >= 2 || index === maxPolls - 1)
-            return candidate;
     }
-    return lastCandidate;
+    return '';
 }