semantic search and retrieval of personal or organizational documents and chat
// ==UserScript== // @name chatgpt-retrieval-plugin // @namespace http://tampermonkey.net/ // @version 0.3 // @description semantic search and retrieval of personal or organizational documents and chat // @author [email protected] // @match https://*/* // @grant GM_xmlhttpRequest // @run-at document-idle // @license MIT // ==/UserScript== (function () { 'use strict'; if (!window.location.href.startsWith('https://chat.openai.com/')) { document.addEventListener('keydown', function (event) { if (event.ctrlKey && event.key === 'g') { event.preventDefault(); const currentUrl = window.location.href; const chatUrl = `https://chat.openai.com/chat?url=${encodeURIComponent(currentUrl)}`; window.open(chatUrl, '_blank'); } }); return } let monitorReply2TranslateBack = false, monitorReply2quest = false; let translateMode = true; let form, submitButton, inputField, submitButtonClone, inputFieldClone; interceptSubmitBtn(); const observer = new MutationObserver(mutations => { mutations.forEach(mutation => { if (mutation.type === 'childList') { console.log(mutation.addedNodes) mutation.addedNodes.forEach(addedNode => { if (addedNode.nodeType === Node.ELEMENT_NODE && ((addedNode.classList.contains('overflow-hidden') && addedNode.classList.contains('w-full') && addedNode.classList.contains('h-full') && addedNode.classList.contains('relative')))) { interceptSubmitBtn(); } if (addedNode.nodeType === Node.ELEMENT_NODE && (addedNode.classList.contains('px-2') && addedNode.classList.contains('py-10') && addedNode.classList.contains('relative') && addedNode.classList.contains('w-full') && addedNode.classList.contains('flex') && addedNode.classList.contains('flex-col') && addedNode.classList.contains('h-full'))) { interceptSubmitBtn(); } }); mutation.removedNodes.forEach(removedNode => { if (removedNode.nodeType === Node.ELEMENT_NODE && removedNode.classList.contains('btn') && removedNode.classList.contains('flex') && removedNode.classList.contains('justify-center') && removedNode.classList.contains('gap-2') && removedNode.classList.contains('btn-neutral') && removedNode.classList.contains('border-0') && removedNode.classList.contains('md:border') && removedNode.textContent.includes('Stop generating')) { const proseElements = document.querySelectorAll('.prose'); const lastProseElement = proseElements[proseElements.length - 1]; const form = document.querySelector('form.stretch'); const submitButton = form.querySelector('div div:nth-child(2) > button'); const inputField = form.querySelector('div:nth-child(2) textarea'); if (monitorReply2quest) { quest(lastProseElement.textContent); } else if (monitorReply2TranslateBack) { translateBack(lastProseElement.textContent); } } }) } }); }); observer.observe(document.body, { childList: true, subtree: true }); // 定义翻译函数，用于翻译给定的文本 const translateText = async (text) => { const url = 'https://translate.googleapis.com/translate_a/single?client=gtx&sl=zh-CN&tl=en&dt=t&q=' + encodeURI(text); const response = await fetch(url); const data = await response.json(); return data[0][0][0]; }; function interceptSubmitBtn() { form = document.querySelector('form.stretch'); submitButton = form.querySelector('div div:nth-child(2) > button'); inputField = form.querySelector('div:nth-child(2) textarea'); // 拦截表单提交事件，翻译输入内容并替换输入框中的内容 if (!submitButton || !inputField) { return; } submitButton.addEventListener('click', handleSubmit, { useCapture: true }); inputField.addEventListener("keydown", handleEnterDown, { useCapture: true }); inputField.addEventListener("keyup", async function (event) { if (event.key === "Enter") { event.preventDefault(); event.stopImmediatePropagation(); } }); inputField.addEventListener("keypress", async function (event) { if (event.key === "Enter") { event.preventDefault(); event.stopImmediatePropagation(); } }); } async function handleEnterDown(event) { if (event.key === "Enter") { handleSubmit(event); } } async function handleSubmit(event) { const stage = submitButton.getAttribute('data-stage') const texts = inputField.value.split('|'); if ((stage && stage != 0) || !translateMode) { return; } else if (texts.length > 1 && ['默认模式', 'default', 'd'].includes(texts[0])) { inputField.value = texts[1]; return; } else if (texts.length > 1 && ['切换模式', 'toggle', 't'].includes(texts[0])) { translateMode = !translateMode; inputField.value = texts[1]; return; } event.preventDefault(); event.stopImmediatePropagation(); if (!+stage) { if (!monitorReply2quest && !monitorReply2TranslateBack) { translate(); // Proceed with other steps like translation } } } function getUrlParameter(name) { const urlParams = new URLSearchParams(window.location.search); return urlParams.get(name); } async function translate() { submitButton.setAttribute('data-stage', 'translating'); const inputText = inputField.value.trim(); if (inputText.length === 0) { return; } inputField.value = "处理中..."; try { const apiResponseText = await enhanceQuestion(inputText); console.log('API response text:', apiResponseText); inputField.value = apiResponseText; submitButton.click(); submitButton.setAttribute('data-stage', 0); monitorReply2TranslateBack = false; } catch (error) { console.error('Error:', error); alert(error); } } function quest(reply) { submitButton.setAttribute('data-stage', 'questing'); if (reply.length === 0) { return; } inputField.value = 'answer using English, as specific as possible： "' + reply + '"'; submitButton.click(); monitorReply2quest = false; monitorReply2TranslateBack = true; } function translateBack(reply) { submitButton.setAttribute('data-stage', 'translateBacking'); inputField.value = '翻译成中文，只返回中文： "' + reply + '"'; submitButton.click(); monitorReply2TranslateBack = false; submitButton.setAttribute('data-stage', 0); } async function enhanceQuestion(text) { const filters = getAllUrlParameters(); return new Promise(async (resolve, reject) => { try { const response = await fetch('https://api.talkgpt.space/v1/questions/enhance', { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ text, filter: filters }) }); if (response.ok) { const data = await response.json(); resolve(data.enhancedText); } else { const errorMessage = (await response.json()).message; console.error(errorMessage); reject(errorMessage); } } catch (error) { const errorMessage = `Error enhancing question: ${error.message}`; console.error(errorMessage); reject(errorMessage); } }); } function getAllUrlParameters() { const urlParams = new URLSearchParams(window.location.search); let filters = {}; for (const [key, value] of urlParams.entries()) { filters[key] = value; } return filters; } })();
