Google Search with AI-Generated Annotation via Gemini
当前为
// ==UserScript==
// @name Google Search Enhanced via Gemini AI
// @description Google Search with AI-Generated Annotation via Gemini
// @version 0.6
// @license MIT
// @namespace djshigel
// @match https://www.google.com/search*
// @run-at document-end
// @grant GM.setValue
// @grant GM.getValue
// ==/UserScript==
(async () => {
let GEMINI_API_KEY = await GM.getValue("GEMINI_API_KEY") ;
if (!GEMINI_API_KEY || !Object.keys(GEMINI_API_KEY).length) {
GEMINI_API_KEY = window.prompt('Get Generative Language Client API key from Google AI Studio\nhttps://ai.google.dev/aistudio', '');
await GM.setValue("GEMINI_API_KEY", GEMINI_API_KEY);
}
const apiUrl = `https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key=${GEMINI_API_KEY}`;
const delay = (ms) => new Promise(resolve => setTimeout(resolve, ms));
// ########## Results ##########
const processArticle = async (article, title, url) => {
try {
document.querySelector('#gemini-ticker').style.opacity = '1';
const response = await fetch(apiUrl, {
method: 'POST',
headers: { 'Content-Type': 'application/json' },
body: JSON.stringify({
contents: [{
parts: [{
text: `I'm searching about ${document.querySelector('textarea').value}.
Follow the steps below to execute step by step for each URL.
1 If the URL cannot be accessed, do not output the results
2 Summarize in 200 characters or so like an academic
3 Output only the results
${title} with URL: ${url}`
}],
}]
}),
});
if (!response.ok) throw new Error('Network response was not ok');
const reader = response.body.getReader();
let result = '', done = false, decoder = new TextDecoder();
while (!done) {
const { value, done: doneReading } = await reader.read();
done = doneReading;
if (value) result += decoder.decode(value, { stream: true });
}
result += decoder.decode();
const data = JSON.parse(result);
let summary = (data.candidates[0]?.content?.parts[0]?.text || '').replace(/\*\*/g, '').replace(/##/g, '');
console.log(`summary: ${summary}`);
let targetElement = article.parentElement.parentElement.parentElement.parentElement
.nextSibling?.querySelectorAll('div>span')[1] ||
article.parentElement.parentElement.parentElement.parentElement
.nextSibling?.querySelectorAll('div>span')[0];
if (!targetElement) return;
targetElement.parentElement.setAttribute('style', '-webkit-line-clamp: 30');
article.classList.add('gemini-annotated');
let displayText = '✦ ';
for (const char of summary) {
document.querySelector('#gemini-ticker').style.opacity = '1';
while (document.querySelector('#rso').classList.contains('hover')) {
await delay(100);
}
displayText += char + '●';
targetElement.textContent = displayText;
await delay(2);
displayText = displayText.slice(0, -1);
document.querySelector('#gemini-ticker').style.opacity = '0';
}
targetElement.textContent = displayText;
} catch (error) {
document.querySelector('#gemini-ticker').style.opacity = '0';
await delay(5000);
console.error('Error:', error);
}
};
const throttledProcessArticle = async (article, title, url, interval) => {
await delay(interval);
return processArticle(article, title, url);
};
// ########## Ticker ##########
const insertTickerElement = () => {
const ticker = document.createElement('div');
ticker.id = 'gemini-ticker';
ticker.style.position = 'fixed';
ticker.style.right = '20px';
ticker.style.bottom = '10px';
ticker.style.fontSize = '1.5em';
ticker.style.color = '#77777777';
ticker.style.transition = 'opacity .3s';
ticker.style.zIndex = '100';
ticker.innerHTML = '✦';
document.querySelector('body').appendChild(ticker);
};
// ########## Main ##########
await delay(1000);
insertTickerElement();
for (let j = 0; j < 30 ; j++) {
console.log(`######## attempt: ${j+1} ########`)
const articles = Array.from(document.querySelectorAll('#rso>div'))
.map(result=>result.querySelector('span>a:not(.gemini-annotated)'));
if (articles.length == 0) break;
document.querySelector('#rso').addEventListener('mouseover', ()=>{
document.querySelector('#rso').classList.add('hover')
});
document.querySelector('#rso').addEventListener('mouseout', ()=>{
document.querySelector('#rso').classList.remove('hover')
});
const promises = articles.map((targetLink, i) => {
if (!targetLink) return Promise.resolve();
const href = targetLink.getAttribute('href');
const title = targetLink.querySelector('h3').textContent;
console.log(`title: ${title}`);
console.log(`url: ${href}`);
if (!href) return Promise.resolve();
return throttledProcessArticle(targetLink, title, href, i * 1000);
});
await Promise.all(promises);
document.querySelector('#gemini-ticker').style.opacity = '0';
}
document.querySelector('#gemini-ticker').style.opacity = '0';
})();