chat-ui.js 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966
  1. class ChatUI {
  2. constructor() {
  3. this.messageContainer = document.getElementById("chat-messages");
  4. this.input = document.getElementById("chat-input");
  5. this.sendButton = document.getElementById("send-button");
  6. this.stopButton = document.getElementById("stop-button");
  7. this.promptButton = document.getElementById("prompt-button");
  8. this.promptPanel = document.getElementById("prompt-panel");
  9. this.uploadButton = document.getElementById("upload-button");
  10. this.fileInput = document.getElementById("file-input");
  11. this.ContentInputs = []
  12. this.iframeInfo = null
  13. this.excelData = [];
  14. // 确保AI服务已经初始化
  15. if (!window.aiService) {
  16. throw new Error("AI Service not initialized");
  17. }
  18. this.aiService = window.aiService;
  19. this.typingSpeed = 50;
  20. this.inputWrapper = document.querySelector(".input-wrapper");
  21. // 先添加过渡动画类
  22. this.input.classList.add("input-transition");
  23. // 设置初始高度
  24. this.input.style.height = "40px";
  25. // 获取总结按钮
  26. this.summarizeButton = document.querySelector(".summarize-button");
  27. // 支持的Excel文件类型
  28. this.excelTypes = {
  29. xlsx: "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
  30. xls: "application/vnd.ms-excel",
  31. csv: "text/csv",
  32. };
  33. // 初始化其他组件
  34. this.setupEventListeners();
  35. this.init();
  36. // 使用微任务确保DOM更新后再调整高度
  37. Promise.resolve().then(() => {
  38. // 第一帧:等待过渡动画类生效
  39. requestAnimationFrame(() => {
  40. // 第二帧:执行高度调整
  41. requestAnimationFrame(() => {
  42. this.adjustInputHeight();
  43. // 触发一次输入框focus以确保正确的高度
  44. this.input.focus();
  45. });
  46. });
  47. });
  48. // 添加新消息指示器
  49. this.createNewMessageIndicator();
  50. // 监听滚动事件
  51. this.messageContainer.addEventListener(
  52. "scroll",
  53. this.handleScroll.bind(this)
  54. );
  55. // 创建打断按钮
  56. // this.stopButton.className = "stop-button";
  57. // this.stopButton.innerHTML = `
  58. // <svg viewBox="0 0 24 24" fill="currentColor">
  59. // <path d="M6 6h12v12H6z"/>
  60. // </svg>
  61. // <span>停止生成</span>
  62. // `;
  63. // 将打断按钮添加到输入按钮区域
  64. // document.querySelector(".input-buttons").appendChild(this.stopButton);
  65. // 添加打断按钮事件监听
  66. // this.stopButton.addEventListener("click", () => this.handleStop());
  67. this.isTyping = false; // 添加打字状态标记
  68. this.loadingDiv = null; // 添加加载动画的引用
  69. // 初始化页面信息卡片
  70. this.initPageInfoCard();
  71. }
  72. async init() {
  73. try {
  74. await this.aiService.init();
  75. await this.checkApiKey();
  76. } catch (error) {
  77. console.error("ChatUI initialization failed:", error);
  78. this.addMessage("初始化失败,请刷新页面重试。", "assistant");
  79. }
  80. }
  81. setupEventListeners() {
  82. // 发送消息
  83. this.sendButton.addEventListener("click", () => this.handleSend());
  84. this.input.addEventListener("keydown", (e) => {
  85. if (e.key === "Enter" && !e.shiftKey) {
  86. e.preventDefault();
  87. this.handleSend();
  88. }
  89. });
  90. // 自动调整输入框高度
  91. this.input.addEventListener("input", () => this.adjustInputHeight());
  92. // 添加focus事件监听,确保在获得焦点时高度正确
  93. this.input.addEventListener("focus", () => {
  94. requestAnimationFrame(() => this.adjustInputHeight());
  95. });
  96. // 快捷指令面板
  97. this.promptButton.addEventListener("click", () => this.togglePromptPanel());
  98. document.querySelector(".close-prompt").addEventListener("click", () => {
  99. this.promptPanel.classList.remove("show");
  100. });
  101. // 点击快捷指令
  102. document.querySelectorAll(".prompt-item").forEach((item) => {
  103. item.addEventListener("click", () => {
  104. // 获取文本内容并处理格式
  105. const text = item.textContent
  106. .trim() // 移除首尾空白
  107. .replace(/\s+/g, " ") // 将多个空白字符替换为单个空格
  108. .replace(/[\n\r]+/g, ""); // 移除所有换行符
  109. // 设置到输入框
  110. this.input.value = text;
  111. this.promptPanel.classList.remove("show");
  112. // 调整输入框高度并聚焦
  113. this.adjustInputHeight();
  114. this.input.focus();
  115. });
  116. });
  117. // 点击外部关闭快捷指令面板
  118. document.addEventListener("click", (e) => {
  119. if (
  120. !this.promptPanel.contains(e.target) &&
  121. !this.promptButton.contains(e.target)
  122. ) {
  123. this.promptPanel.classList.remove("show");
  124. }
  125. });
  126. // 添加总结按钮点击事件
  127. if (this.summarizeButton) {
  128. this.summarizeButton.addEventListener("click", () => {
  129. this.handleSummarize();
  130. });
  131. }
  132. // 工具箱面板
  133. const toolsButton = document.getElementById("tools-button");
  134. const toolsPanel = document.getElementById("tools-panel");
  135. const closeTools = document.querySelector(".close-tools");
  136. toolsButton.addEventListener("click", () => {
  137. toolsPanel.classList.toggle("show");
  138. // 关闭其他面板
  139. this.promptPanel.classList.remove("show");
  140. });
  141. closeTools.addEventListener("click", () => {
  142. toolsPanel.classList.remove("show");
  143. });
  144. // 点击外部关闭工具箱面板
  145. document.addEventListener("click", (e) => {
  146. if (!toolsPanel.contains(e.target) && !toolsButton.contains(e.target)) {
  147. toolsPanel.classList.remove("show");
  148. }
  149. });
  150. // 上传按钮点击事件
  151. this.uploadButton.addEventListener("click", async () => {
  152. this.iframeInfo = await new Promise((resolve) => {
  153. // 创建一次性消息监听器
  154. const messageHandler = (event) => {
  155. if (event.data.type === "PAGE_ANALYSIS_RESULT") {
  156. window.removeEventListener("message", messageHandler);
  157. resolve(event.data.pageInfo);
  158. }
  159. };
  160. console.log(this.iframeInfo);
  161. // 添加消息监听
  162. window.addEventListener("message", messageHandler);
  163. // 发送分析请求到父页面
  164. window.parent.postMessage({ type: "ANALYZE_PAGE" }, "*");
  165. console.log(2222285);
  166. });
  167. console.log(this.iframeInfo);
  168. this.fileInput.click();
  169. });
  170. // 文件选择事件
  171. this.fileInput.addEventListener("change", (event) => {
  172. console.log(event);
  173. const files = event.target.files;
  174. if (files.length > 0) {
  175. this.handleFileUpload(files);
  176. }
  177. // 清空文件输入框,确保同一文件可以重复上传
  178. event.target.value = "";
  179. });
  180. }
  181. async handleSend() {
  182. const message = this.input.value.trim();
  183. if (!message) return;
  184. // 处理命令
  185. if (message.startsWith("/setapi ")) {
  186. const apiKey = message.substring(8).trim();
  187. await this.aiService.setApiKey(apiKey);
  188. // 根据是否使用默认密钥显示不同消息
  189. if (apiKey === CONFIG.AI_API.DEFAULT_API_KEY) {
  190. this.addMessage("已恢复使用默认API密钥", "assistant", false);
  191. } else {
  192. this.addMessage("自定义API密钥已设置", "assistant", false);
  193. }
  194. this.input.value = "";
  195. return;
  196. }
  197. // 添加用户消息
  198. this.addMessage(message, "user", false);
  199. // 更新上下文
  200. this.aiService.updateContext(message, "user");
  201. // 清空输入框
  202. this.input.value = "";
  203. this.adjustInputHeight();
  204. try {
  205. // this.stopButton.classList.add("show");
  206. this.setInputState(true); // 禁用输入框
  207. this.loadingDiv = ''
  208. console.log(this.aiService.currentExcelData);
  209. // 如果存在Excel数据,添加上下文提示
  210. let prompt = message;
  211. if (this.aiService.currentExcelData) {
  212. const { fileName, headers, totalRows, rows } =
  213. this.aiService.currentExcelData;
  214. prompt = `请记住之前我们正在讨论的Excel文件:
  215. - 文件名:${fileName}
  216. - 列标题:${headers.join(", ")}
  217. - 列内容:${rows.join(", ")}
  218. - 总行数:${totalRows}
  219. 基于这个Excel文件的内容,请回答以下问题:
  220. ${this.iframeInfo}`;
  221. }
  222. const response = await this.aiService.sendMessage(prompt);
  223. if (this.loadingDiv) {
  224. this.loadingDiv.remove();
  225. this.loadingDiv = null;
  226. }
  227. if (!response) return;
  228. this.isTyping = false;
  229. await this.addMessage(response, "assistant", true);
  230. // this.stopButton.classList.remove("show");
  231. this.setInputState(false); // 这里会自动聚焦输入框
  232. this.aiService.updateContext(response, "assistant");
  233. } catch (error) {
  234. if (this.loadingDiv) {
  235. this.loadingDiv.remove();
  236. this.loadingDiv = null;
  237. }
  238. // this.stopButton.classList.remove("show");
  239. this.setInputState(false); // 这里也会自动聚焦输入框
  240. this.isTyping = false;
  241. if (error.message === "REQUEST_ABORTED") {
  242. return;
  243. }
  244. this.addMessage("抱歉,发生了一些错误,请稍后重试。", "assistant", false);
  245. console.error("AI response error:", error);
  246. }
  247. }
  248. /**
  249. * 添加消息到聊天界面
  250. * @param {string} content 消息内容
  251. * @param {string} type 消息类型(user/assistant)
  252. * @param {boolean} typing 是否使用打字效果
  253. * @param {boolean} isInterrupted 是否是中断消息
  254. */
  255. async addMessage(content, type, typing = type === "assistant", isInterrupted = false) {
  256. const messageDiv = document.createElement("div");
  257. messageDiv.className = `message ${type}`;
  258. const messageContent = document.createElement("div");
  259. messageContent.className = "message-content";
  260. if (isInterrupted) {
  261. messageContent.classList.add("interrupted");
  262. }
  263. const paragraph = document.createElement("p");
  264. messageContent.appendChild(paragraph);
  265. // 创建操作栏(时间戳和复制按钮)
  266. const actionsDiv = document.createElement("div");
  267. actionsDiv.className = "message-actions";
  268. // 只为非中断的AI消息添加复制按钮
  269. if (type === "assistant" && !isInterrupted) {
  270. const copyButton = this.createCopyButton(content);
  271. actionsDiv.appendChild(copyButton);
  272. console.log(content);
  273. console.log(Object.prototype.toString.call(content));
  274. if (Object.prototype.toString.call(content) === "[object Object]") {
  275. actionsDiv.appendChild(this.createFillButton(paragraph));
  276. }
  277. }
  278. // 添加时间戳
  279. const timestamp = document.createElement("div");
  280. timestamp.className = "message-timestamp";
  281. timestamp.textContent = this.formatTime(new Date());
  282. actionsDiv.appendChild(timestamp);
  283. messageContent.appendChild(actionsDiv);
  284. messageDiv.appendChild(messageContent);
  285. this.messageContainer.appendChild(messageDiv);
  286. if (typing) {
  287. messageContent.classList.add("typing");
  288. if (typeof content !== 'string') {
  289. const signal = content.controller.signal;
  290. try {
  291. const iterator = content.iterator();
  292. for await (const chunk of iterator) {
  293. if (chunk) {
  294. const decodedChunk = chunk.choices[0].delta.content;
  295. if (decodedChunk) {
  296. paragraph.innerHTML += decodedChunk;
  297. }
  298. }
  299. }
  300. } catch (error) {
  301. if (signal.aborted) {
  302. console.log("Stream reading aborted");
  303. } else {
  304. console.error("Error reading stream:", error);
  305. }
  306. }
  307. } else {
  308. await this.typeMessage(paragraph, content);
  309. }
  310. messageContent.classList.remove("typing");
  311. } else {
  312. paragraph.innerHTML = this.formatMessage(content);
  313. }
  314. const { scrollTop, scrollHeight, clientHeight } = this.messageContainer;
  315. const wasAtBottom = scrollHeight - scrollTop - clientHeight < 100;
  316. if (wasAtBottom) {
  317. this.scrollToBottom();
  318. } else {
  319. this.newMessageIndicator.classList.add("show");
  320. }
  321. return paragraph
  322. }
  323. /**
  324. * 实现打字机效果
  325. * @param {HTMLElement} element 要添加文字的元素
  326. * @param {string} text 要显示的文字
  327. */
  328. async typeMessage(element, text) {
  329. return element.innerHTML = text;
  330. let index = 0;
  331. const rawText = text;
  332. const tempDiv = document.createElement("div");
  333. this.isTyping = true; // 开始打字
  334. return new Promise((resolve) => {
  335. const type = () => {
  336. // 检查是否被中断
  337. if (!this.isTyping) {
  338. resolve();
  339. return;
  340. }
  341. if (index < rawText.length) {
  342. const currentText = rawText.substring(0, index + 1);
  343. tempDiv.innerHTML = this.formatMessage(currentText);
  344. element.innerHTML = tempDiv.innerHTML;
  345. index++;
  346. this.scrollToBottom();
  347. setTimeout(type, this.typingSpeed);
  348. } else {
  349. this.isTyping = false; // 打字结束
  350. resolve();
  351. }
  352. };
  353. type();
  354. });
  355. }
  356. adjustInputHeight() {
  357. const scrollPos = this.input.scrollTop;
  358. // 获取当前高度
  359. const currentHeight = this.input.style.height;
  360. // 重置高度
  361. this.input.style.height = "auto";
  362. // 计算新高度
  363. const newHeight = Math.min(this.input.scrollHeight, 120);
  364. // 如果高度有变化才设置
  365. if (currentHeight !== `${newHeight}px`) {
  366. this.input.style.height = `${newHeight}px`;
  367. }
  368. // 恢复滚动位置
  369. this.input.scrollTop = scrollPos;
  370. }
  371. scrollToBottom() {
  372. this.messageContainer.scrollTop = this.messageContainer.scrollHeight;
  373. }
  374. togglePromptPanel() {
  375. this.promptPanel.classList.toggle("show");
  376. }
  377. escapeHtml(html) {
  378. const div = document.createElement("div");
  379. div.textContent = html;
  380. return div.innerHTML;
  381. }
  382. async checkApiKey() {
  383. // 只有当没有任何API密钥时才显示提示
  384. if (!this.aiService.apiKey) {
  385. this.addMessage(
  386. "请先设置DeepSeek API密钥。输入格式:/setapi YOUR_API_KEY",
  387. "assistant"
  388. );
  389. }
  390. }
  391. /**
  392. * 创建加载动画消息
  393. */
  394. createLoadingMessage() {
  395. const loadingDiv = document.createElement("div");
  396. loadingDiv.className = "message assistant";
  397. loadingDiv.innerHTML = `
  398. <div class="message-content loading">
  399. <div class="typing-indicator">
  400. <span></span>
  401. <span></span>
  402. <span></span>
  403. </div>
  404. </div>
  405. `;
  406. this.messageContainer.appendChild(loadingDiv);
  407. this.scrollToBottom();
  408. return loadingDiv;
  409. }
  410. /**
  411. * 格式化时间
  412. */
  413. formatTime(date) {
  414. const hours = date.getHours().toString().padStart(2, "0");
  415. const minutes = date.getMinutes().toString().padStart(2, "0");
  416. return `${hours}:${minutes}`;
  417. }
  418. createNewMessageIndicator() {
  419. this.newMessageIndicator = document.createElement("div");
  420. this.newMessageIndicator.className = "new-messages-indicator";
  421. this.newMessageIndicator.textContent = "新消息";
  422. this.newMessageIndicator.addEventListener("click", () => {
  423. this.scrollToBottom();
  424. });
  425. document
  426. .querySelector(".chat-container")
  427. .appendChild(this.newMessageIndicator);
  428. }
  429. handleScroll() {
  430. const { scrollTop, scrollHeight, clientHeight } = this.messageContainer;
  431. const isNearBottom = scrollHeight - scrollTop - clientHeight < 100;
  432. if (isNearBottom) {
  433. this.newMessageIndicator.classList.remove("show");
  434. }
  435. }
  436. /**
  437. * 格式化消息内容
  438. * @param {string} text 原始文本
  439. * @returns {string} 格式化后的HTML
  440. */
  441. formatMessage(text) {
  442. if (!text) return "";
  443. return (
  444. text
  445. // 处理标题 (h1 ~ h6)
  446. .replace(/^#{1,6}\s+(.+)$/gm, (match, content) => {
  447. const level = match.trim().split("#").length - 1;
  448. return `<h${level}>${content.trim()}</h${level}>`;
  449. })
  450. // 处理换行
  451. .replace(/\n/g, "<br>")
  452. // 处理连续空格
  453. .replace(/ {2,}/g, (match) => "&nbsp;".repeat(match.length))
  454. // 处理代码块
  455. .replace(
  456. /```([\s\S]*?)```/g,
  457. (match, code) =>
  458. `<pre><code>${this.escapeHtml(code.trim())}</code></pre>`
  459. )
  460. // 处理行内代码
  461. .replace(
  462. /`([^`]+)`/g,
  463. (match, code) => `<code>${this.escapeHtml(code)}</code>`
  464. )
  465. // 处理粗体
  466. .replace(/\*\*(.*?)\*\*/g, "<strong>$1</strong>")
  467. // 处理斜体
  468. .replace(/\*(.*?)\*/g, "<em>$1</em>")
  469. // 处理链接
  470. .replace(
  471. /\[([^\]]+)\]\(([^)]+)\)/g,
  472. '<a href="$2" target="_blank">$1</a>'
  473. )
  474. // 处理无序列表
  475. .replace(/^[*-]\s+(.+)$/gm, "<li>$1</li>")
  476. .replace(/(<li>.*<\/li>)/gs, "<ul>$1</ul>")
  477. // 处理有序列表
  478. .replace(/^\d+\.\s+(.+)$/gm, "<li>$1</li>")
  479. .replace(/(<li>.*<\/li>)/gs, "<ol>$1</ol>")
  480. // 处理分隔线
  481. .replace(/^---+$/gm, "<hr>")
  482. // 处理引用
  483. .replace(/^>\s+(.+)$/gm, "<blockquote>$1</blockquote>")
  484. );
  485. }
  486. createFillButton(paragraph) {
  487. const button = document.createElement("button");
  488. button.className = "fill-button";
  489. button.innerHTML = `
  490. <svg viewBox="0 0 24 24" fill="currentColor">
  491. <path d="M3 17.25V21h3.75L17.81 9.94l-3.75-3.75L3 17.25zM20.71 7.04c.39-.39.39-1.02 0-1.41l-2.34-2.34c-.39-.39-1.02-.39-1.41 0l-1.83 1.83 3.75 3.75 1.83-1.83z"/>
  492. </svg>
  493. <span>填充</span>
  494. `;
  495. button.addEventListener("click", () => {
  496. window.parent.postMessage({
  497. type: "HANDLE_FILL_INPUT", data: {
  498. formData: JSON.parse(paragraph.innerHTML.split('json')[1].split('```')[0]) //根据不同返回修改res
  499. }
  500. }, "*");
  501. });
  502. return button;
  503. }
  504. /**
  505. * 创建复制按钮
  506. * @param {string} content 要复制的内容
  507. * @returns {HTMLElement} 复制按钮元素
  508. */
  509. createCopyButton(content) {
  510. const button = document.createElement("button");
  511. button.className = "copy-button";
  512. button.innerHTML = `
  513. <svg viewBox="0 0 24 24" fill="currentColor">
  514. <path d="M16 1H4C2.9 1 2 1.9 2 3v14h2V3h12V1zm3 4H8C6.9 5 6 5.9 6 7v14c0 1.1.9 2 2 2h11c1.1 0 2-.9 2-2V7c0-1.1-.9-2-2-2zm0 16H8V7h11v14z"/>
  515. </svg>
  516. <span>复制</span>
  517. `;
  518. button.addEventListener("click", async () => {
  519. try {
  520. // 直接使用传入的原始内容
  521. await this.copyToClipboard(content);
  522. // 显示复制成功状态
  523. button.classList.add("copied");
  524. button.innerHTML = `
  525. <svg viewBox="0 0 24 24" fill="currentColor">
  526. <path d="M9 16.2L4.8 12l-1.4 1.4L9 19 21 7l-1.4-1.4L9 16.2z"/>
  527. </svg>
  528. <span>已复制</span>
  529. `;
  530. // 2秒后恢复原始状态
  531. setTimeout(() => {
  532. button.classList.remove("copied");
  533. button.innerHTML = `
  534. <svg viewBox="0 0 24 24" fill="currentColor">
  535. <path d="M16 1H4C2.9 1 2 1.9 2 3v14h2V3h12V1zm3 4H8C6.9 5 6 5.9 6 7v14c0 1.1.9 2 2 2h11c1.1 0 2-.9 2-2V7c0-1.1-.9-2-2-2zm0 16H8V7h11v14z"/>
  536. </svg>
  537. <span>复制</span>
  538. `;
  539. }, 2000);
  540. } catch (err) {
  541. console.error("Failed to copy text:", err);
  542. }
  543. });
  544. return button;
  545. }
  546. /**
  547. * 复制文本到剪贴板
  548. * @param {string} text 要复制的文本
  549. */
  550. async copyToClipboard(text) {
  551. try {
  552. // 通过postMessage发送复制请求到content script
  553. window.parent.postMessage(
  554. {
  555. type: "COPY_TO_CLIPBOARD",
  556. text: text,
  557. },
  558. "*"
  559. );
  560. return true;
  561. } catch (err) {
  562. console.error("Failed to copy text:", err);
  563. throw err;
  564. }
  565. }
  566. // 处理打断请求
  567. async handleStop() {
  568. try {
  569. this.aiService.abortRequest();
  570. if (this.isTyping) {
  571. this.isTyping = false;
  572. }
  573. if (this.loadingDiv) {
  574. this.loadingDiv.remove();
  575. this.loadingDiv = null;
  576. }
  577. // this.stopButton.classList.remove("show");
  578. this.setInputState(false); // 中断后也自动聚焦输入框
  579. this.addMessage("用户手动停止生成", "assistant", false, true);
  580. } catch (error) {
  581. console.error("Failed to stop generation:", error);
  582. }
  583. }
  584. // 设置输入框状态
  585. setInputState(isGenerating) {
  586. this.input.disabled = isGenerating;
  587. this.input.placeholder = isGenerating ? "回复生成中..." : "输入消息...";
  588. if (isGenerating) {
  589. this.inputWrapper.classList.add("generating");
  590. this.stopButton.classList.add("show");
  591. } else {
  592. this.inputWrapper.classList.remove("generating");
  593. // this.stopButton.classList.remove("show");
  594. // AI回复完成后,自动聚焦到输入框
  595. this.input.focus();
  596. }
  597. }
  598. /**
  599. * 初始化页面信息卡片
  600. */
  601. initPageInfoCard() {
  602. // 从父窗口获取页面信息
  603. window.addEventListener("message", (event) => {
  604. if (event.data.type === "PAGE_INFO") {
  605. const pageInfo = event.data.pageInfo;
  606. this.iframeInfo = event.data.pageInfo.iframe
  607. this.ContentInputs = pageInfo.inputs
  608. // 更新卡片内容
  609. const favicon = document.querySelector(".page-favicon");
  610. const title = document.querySelector(".page-title");
  611. // 设置网站图标
  612. favicon.src = pageInfo.favicon;
  613. favicon.onerror = () => {
  614. // 如果图标加载失败,使用默认图标
  615. favicon.src = chrome.runtime.getURL("images/icon16.png");
  616. };
  617. // 设置页面标题
  618. title.textContent = pageInfo.title;
  619. }
  620. });
  621. }
  622. /**
  623. * 获取页面favicon
  624. * @returns {string} favicon URL
  625. */
  626. getPageFavicon() {
  627. // 尝试获取页面favicon
  628. const iconLink = document.querySelector('link[rel*="icon"]');
  629. if (iconLink) return iconLink.href;
  630. // 如果没有找到,返回网站根目录的favicon.ico
  631. const url = new URL(window.location.href);
  632. return `${url.protocol}//${url.hostname}/favicon.ico`;
  633. }
  634. /**
  635. * 处理总结请求
  636. */
  637. async handleSummarize() {
  638. try {
  639. // 显示加载状态
  640. this.setInputState(true);
  641. this.loadingDiv = ''
  642. // 通过postMessage请求页面分析结果
  643. this.iframeInfo = await new Promise((resolve) => {
  644. // 创建一次性消息监听器
  645. const messageHandler = (event) => {
  646. if (event.data.type === "PAGE_ANALYSIS_RESULT") {
  647. window.removeEventListener("message", messageHandler);
  648. resolve(event.data.pageInfo);
  649. }
  650. };
  651. // 添加消息监听
  652. window.addEventListener("message", messageHandler);
  653. // 发送分析请求到父页面
  654. window.parent.postMessage({ type: "ANALYZE_PAGE" }, "*");
  655. });
  656. // 构建提示词
  657. const prompt = this.aiService.getSummaryPrompt(pageInfo);
  658. // 发送分析请求
  659. const response = await this.aiService.sendMessage(prompt);
  660. // 显示分析结果
  661. if (this.loadingDiv) {
  662. this.loadingDiv.remove();
  663. this.loadingDiv = null;
  664. }
  665. await this.addMessage(response, "assistant", false);
  666. // 将总结内容添加到上下文
  667. this.aiService.updateContext(prompt, "user");
  668. this.aiService.updateContext(response, "assistant");
  669. // 保存页面信息到上下文
  670. this.aiService.setPageInfo(pageInfo);
  671. this.setInputState(false);
  672. } catch (error) {
  673. console.error("Failed to summarize page:", error);
  674. this.addMessage(
  675. "抱歉,页面总结过程中出现错误,请稍后重试。",
  676. "assistant",
  677. false
  678. );
  679. this.setInputState(false);
  680. }
  681. }
  682. /**
  683. * 处理文件上传
  684. * @param {FileList} files 上传的文件列表
  685. */
  686. async handleFileUpload(files) {
  687. console.log(files);
  688. try {
  689. for (const file of files) {
  690. const extension = file.name.split(".").pop().toLowerCase();
  691. // 显示文件上传消息
  692. this.addMessage(`已上传文件:${file.name}`, "user", false);
  693. if (this.excelTypes[extension]) {
  694. try {
  695. this.setInputState(true);
  696. this.loadingDiv = this.createLoadingMessage();
  697. // 读取Excel文件
  698. const data = await this.readExcelFile(file);
  699. // 保存Excel数据到AI服务
  700. this.aiService.setExcelData({
  701. fileName: file.name,
  702. headers: data[0],
  703. rows: data.slice(1),
  704. totalRows: data.length - 1,
  705. });
  706. // 构建Excel理解提示词
  707. const prompt = this.buildExcelUnderstandingPrompt(data, file.name);
  708. // 调用AI服务理解数据
  709. const response = await this.aiService.sendMessage(prompt);
  710. console.log(response);
  711. if (this.loadingDiv) {
  712. this.loadingDiv.remove();
  713. this.loadingDiv = null;
  714. }
  715. 0 && window.parent.postMessage({
  716. type: "HANDLE_FILL_INPUT", data: {
  717. excelData: this.excelData,
  718. formData: JSON.parse(response.split('json')[1].split('```')[0]) //根据不同返回修改res
  719. }
  720. }, "*");
  721. // 显示AI的理解结果
  722. this.addMessage(response, "assistant", true);
  723. } catch (error) {
  724. console.error("Excel processing error:", error);
  725. this.addMessage(
  726. "Excel文件处理过程中出现错误,请重试。",
  727. "assistant",
  728. false
  729. );
  730. } finally {
  731. this.setInputState(false);
  732. }
  733. }
  734. }
  735. } catch (error) {
  736. console.error("File upload error:", error);
  737. this.addMessage("文件上传过程中出现错误,请重试。", "assistant", false);
  738. }
  739. }
  740. /**
  741. * 构建Excel理解提示词
  742. */
  743. buildExcelUnderstandingPrompt(data, fileName) {
  744. if (!data || data.length < 2) {
  745. return "这是一个空的Excel文件,请检查文件内容。";
  746. }
  747. const headers = data[0];
  748. const rows = data.slice(1);
  749. const sampleRows = rows.slice(0, 2);
  750. console.log();
  751. data[0].forEach((header, i) => {
  752. if (!this.excelData[header]) this.excelData[header] = []
  753. this.excelData[header].push(data[1][i])
  754. })
  755. window.parent.postMessage({
  756. type: "EXCEL_DATA", data: {
  757. excelData: this.excelData,
  758. }
  759. }, "*");
  760. return `我将向你展示一个通过SheetJS库读取的Excel文件内容和一个form表单。请帮我理解这些数据:
  761. 文件名:${fileName}
  762. 列标题:${headers.join(", ")}
  763. 表单内容:
  764. ${this.iframeInfo}
  765. 要求:
  766. 1. 请分析表单中实际可操作的表单项,
  767. 2. 并根据实际可操作的表单项的所有信息与上传excel文件中的列标题进行匹配,生成表单项与excel文件中列标题的数组,并使用findBy告诉我通过表单项的什么字段信息匹配到的,使用findByValue告诉我匹配到的表单项字段值,使用excelColumn字段告诉我excel文件中列标题的值。在一个字段内返回
  768. 3. 并去除没有匹配到的表单项和excel文件中没有匹配到的列,
  769. 4. 通过type字段告诉我输入项的类型
  770. 5. 如果表单项有label标签,同时返回label,通过label字段告诉我label元素的文本
  771. 5. 仅返回数组,不要返回任何其他内容。`
  772. }
  773. /**
  774. * 读取Excel文件 * @param {File} file
  775. * @returns {Promise<Array>}
  776. */
  777. async readExcelFile(file) {
  778. return new Promise((resolve, reject) => {
  779. const reader = new FileReader();
  780. reader.onload = (e) => {
  781. try {
  782. const data = new Uint8Array(e.target.result);
  783. const workbook = XLSX.read(data, {
  784. type: "array",
  785. cellDates: true,
  786. cellNF: false,
  787. cellText: false,
  788. });
  789. // 获取第一个工作表
  790. const firstSheet = workbook.Sheets[workbook.SheetNames[0]];
  791. // 转换为JSON数据
  792. const jsonData = XLSX.utils.sheet_to_json(firstSheet, {
  793. header: 1,
  794. raw: true,
  795. defval: "",
  796. });
  797. resolve(jsonData);
  798. } catch (error) {
  799. reject(error);
  800. }
  801. };
  802. reader.onerror = () => reject(reader.error);
  803. reader.readAsArrayBuffer(file);
  804. });
  805. }
  806. }
  807. // 数据行数:${rows.length}
  808. // 示例数据(前2行):
  809. // ${sampleRows
  810. // .map((row, index) => {
  811. // return `第${index + 1}行: ${row
  812. // .map((cell, i) => `${headers[i]}=${cell}`)
  813. // .join(", ")}`;
  814. // })
  815. // .join("\n")}
  816. // 等待DOM加载完成后再初始化
  817. document.addEventListener("DOMContentLoaded", () => {
  818. try {
  819. window.chatUI = new ChatUI();
  820. } catch (error) {
  821. console.error("Failed to initialize ChatUI:", error);
  822. }
  823. });