ChatGPT Utils

Modifies the behavior of the chat interface on the OpenAI website

当前为 2022-12-24 提交的版本,查看 最新版本

  1. // ==UserScript==
  2. // @name ChatGPT Utils
  3. // @description Modifies the behavior of the chat interface on the OpenAI website
  4. // @namespace ChatGPTUtils
  5. // @version 1.6.0
  6. // @author CriDos
  7. // @match https://chat.openai.com/*
  8. // @icon https://www.google.com/s2/favicons?sz=64&domain=chat.openai.com
  9. // @grant GM_xmlhttpRequest
  10. // @run-at document-end
  11. // @license MIT
  12. // ==/UserScript==
  13.  
  14. 'use strict';
  15.  
  16. console.log(`ChatGPT Utils initializing...`);
  17.  
  18. const debug = false;
  19.  
  20. setInterval(() => {
  21. try {
  22. addAutoTranslate();
  23. } catch (error) {
  24. console.error(error);
  25. }
  26.  
  27. try {
  28. findAndHookTextareaElement();
  29. } catch (error) {
  30. console.error(error);
  31. }
  32. }, 100);
  33.  
  34. function addAutoTranslate() {
  35. var messages = document.querySelectorAll(".markdown.prose");
  36.  
  37. for (var i = 0; i < messages.length; i++) {
  38. const msgMarkdownNode = messages[i];
  39. const parentMsgMarkdown = msgMarkdownNode.parentElement;
  40.  
  41. if (parentMsgMarkdown.isAutoTranslate) {
  42. continue;
  43. }
  44. parentMsgMarkdown.isAutoTranslate = true;
  45.  
  46. setInterval(async () => {
  47. const translateClassName = "translate-markdown";
  48.  
  49. const msgMarkdownContent = msgMarkdownNode.outerHTML;
  50. if (msgMarkdownNode.storeContent == msgMarkdownContent) {
  51. return;
  52. }
  53. msgMarkdownNode.storeContent = msgMarkdownContent;
  54.  
  55. var translateNode = parentMsgMarkdown.querySelector(`.${translateClassName}`);
  56. if (translateNode == null) {
  57. translateNode = msgMarkdownNode.cloneNode(true);
  58. translateNode.classList.add(translateClassName);
  59. parentMsgMarkdown.insertBefore(translateNode, parentMsgMarkdown.firstChild);
  60. }
  61.  
  62. var clone = msgMarkdownNode.cloneNode(true);
  63. clone.classList.add(translateClassName);
  64.  
  65. const translatedContent = await translateHTML(clone.outerHTML, "auto", navigator.language)
  66.  
  67. let index = translatedContent.lastIndexOf('</div>');
  68. let before = translatedContent.slice(0, index);
  69. const endTranslate = `<p>.......... конец_перевода ..........</p>`;
  70.  
  71. translateNode.outerHTML = before.concat(endTranslate, '</div>');
  72. }, 500);
  73. }
  74. }
  75.  
  76. function findAndHookTextareaElement() {
  77. const targetElement = document.querySelector("textarea");
  78. if (targetElement === null) {
  79. return;
  80. }
  81.  
  82. if (targetElement.isAddHookKeydownEvent === true) {
  83. return;
  84. }
  85.  
  86. targetElement.isAddHookKeydownEvent = true;
  87.  
  88. console.log(`Textarea element found. Adding keydown event listener.`);
  89. targetElement.addEventListener("keydown", async event => await handleSubmit(event, targetElement), true);
  90. }
  91.  
  92. async function handleSubmit(event, targetElement) {
  93. console.log(`Keydown event detected: type - ${event.type}, key - ${event.key}`);
  94.  
  95. if (event.shiftKey && event.key === "Enter") {
  96. return;
  97. }
  98.  
  99. if (window.isActiveOnSubmit === true) {
  100. return;
  101. }
  102.  
  103. if (event.key === "Enter") {
  104. window.isActiveOnSubmit = true;
  105. event.stopImmediatePropagation();
  106.  
  107. const request = targetElement.value;
  108. targetElement.value = "";
  109.  
  110. const translatedText = await translateText(request, "ru", "en");
  111.  
  112. targetElement.focus();
  113. targetElement.value = translatedText;
  114. const enterEvent = new KeyboardEvent("keydown", {
  115. bubbles: true,
  116. cancelable: true,
  117. key: "Enter",
  118. code: "Enter"
  119. });
  120. targetElement.dispatchEvent(enterEvent);
  121.  
  122. window.isActiveOnSubmit = false;
  123. }
  124. }
  125.  
  126. async function translateHTML(html, sLang, tLang) {
  127. const excludeTagRegex = /<(pre|code)[^>]*>([\s\S]*?)<\/(pre|code)>/g;
  128. const excludeTags = [];
  129. const excludePlaceholder = 'e0x';
  130.  
  131. const replaceTagRegex = /<[^>]*>/g;
  132. const replaceTags = [];
  133. const replacePlaceholder = 'r0e';
  134.  
  135. let translateHTML = html;
  136.  
  137. let excludeTagsMatch;
  138. while (excludeTagsMatch = excludeTagRegex.exec(html)) {
  139. excludeTags.push(excludeTagsMatch[0]);
  140. translateHTML = translateHTML.replace(excludeTagsMatch[0], `[${excludePlaceholder}${excludeTags.length - 1}]`);
  141. }
  142.  
  143. let replaceTagsMatch;
  144. while (replaceTagsMatch = replaceTagRegex.exec(html)) {
  145. replaceTags.push(replaceTagsMatch[0]);
  146. translateHTML = translateHTML.replace(replaceTagsMatch[0], `[${replacePlaceholder}${replaceTags.length - 1}]`);
  147. }
  148.  
  149. if (debug) {
  150. console.log(`preTranslateHTML: ${html}`);
  151. }
  152.  
  153. translateHTML = await translateText(translateHTML, sLang, tLang);
  154. translateHTML = removeSpaces(translateHTML);
  155.  
  156. for (let i = 0; i < replaceTags.length; i++) {
  157. translateHTML = translateHTML.replace(`[${replacePlaceholder}${i}]`, replaceTags[i]);
  158. }
  159.  
  160. for (let i = 0; i < excludeTags.length; i++) {
  161. translateHTML = translateHTML.replace(`[${excludePlaceholder}${i}]`, excludeTags[i]);
  162. }
  163.  
  164. if (debug) {
  165. console.log(`postTranslateHTML: ${translateHTML}`);
  166. }
  167.  
  168. return translateHTML;
  169. }
  170.  
  171. async function translateText(text, sLang, tLang) {
  172. const url = `https://translate.googleapis.com/translate_a/single?client=gtx&sl=${sLang}&tl=${tLang}&dt=t&q=${encodeURIComponent(text)}`;
  173.  
  174.  
  175. try {
  176. if (debug) {
  177. console.log(`preTranslate: ${text}`);
  178. }
  179.  
  180. const response = await doXHR(url);
  181. const responseText = JSON.parse(response.responseText);
  182.  
  183. let postTranslate = "";
  184. responseText[0].forEach(part => {
  185. postTranslate += part[0];
  186. });
  187.  
  188. if (debug) {
  189. console.log(`postTranslate: ${postTranslate}`);
  190. }
  191.  
  192. return postTranslate;
  193. } catch (error) {
  194. console.error(error);
  195. }
  196. }
  197.  
  198. async function doXHR(url) {
  199. return new Promise((resolve, reject) => {
  200. const xhr = new XMLHttpRequest();
  201. xhr.open("GET", url);
  202. xhr.onload = () => resolve(xhr);
  203. xhr.onerror = () => reject(xhr.statusText);
  204. xhr.send();
  205. });
  206. }
  207.  
  208. function removeSpaces(string) {
  209. const regex = /\[([^\[\]]*)\]/g;
  210. let result;
  211.  
  212. while ((result = regex.exec(string)) !== null) {
  213. string = string.replace(result[1], result[1].replace(/\s/g, ''));
  214. }
  215.  
  216. return string;
  217. }