Update plugins/gpt-4-turbo.js

plugins/gpt-4-turbo.js

@@ -2,6 +2,8 @@ const axios = require('axios');
 const fs = require('fs');
 const puppeteer = require('puppeteer');
 
+const delay = (ms) => new Promise(resolve => setTimeout(resolve, ms));
+
 async function gpt4Skywork(question, chatId = null, imagePath = null) {
     const isNewChat = !chatId;
     const targetUrl = isNewChat 
@@ -22,29 +24,28 @@ async function gpt4Skywork(question, chatId = null, imagePath = null) {
     try {
         const page = await browser.newPage();
         await page.goto(targetUrl, { waitUntil: 'networkidle2' });
-        await page.waitForTimeout(2000);
+        await delay(2000);
 
         if (imageBase64) {
             const fileInput = await page.$('input[type="file"][accept="image/*"]');
             const buffer = Buffer.from(imageBase64, 'base64');
-            await fileInput.uploadFile({
-                name: 'image.png',
-                type: 'image/png',
-                data: buffer
-            });
-            await page.waitForTimeout(3000);
+            const tempPath = `/tmp/upload_${Date.now()}.png`;
+            fs.writeFileSync(tempPath, buffer);
+            await fileInput.uploadFile(tempPath);
+            fs.unlinkSync(tempPath);
+            await delay(3000);
         }
 
         await page.waitForSelector('textarea.scrollbar-custom', { timeout: 10000 });
         await page.type('textarea.scrollbar-custom', question);
-        await page.waitForTimeout(500);
+        await delay(500);
 
         await page.click('button[type=submit][aria-label="Send message"]');
 
         if (isNewChat) {
-            await page.waitForNavigation({ waitUntil: 'networkidle2', timeout: 10000 });
+            await page.waitForNavigation({ waitUntil: 'networkidle2', timeout: 10000 }).catch(() => {});
         }
-        await page.waitForTimeout(3000);
+        await delay(3000);
 
         let isGenerating = true;
         let attempts = 0;
@@ -53,14 +54,15 @@ async function gpt4Skywork(question, chatId = null, imagePath = null) {
             if (!stopBtn) {
                 isGenerating = false;
             } else {
-                await page.waitForTimeout(1000);
+                await delay(1000);
                 attempts++;
             }
         }
 
-        await page.waitForTimeout(2000);
-        const responseDiv = await page.$('div.relative.flex.min-w-\\[60px\\]');
-        const responseText = await page.evaluate(el => el.innerText, responseDiv);
+        await delay(2000);
+        const responseDivs = await page.$$('div.relative.flex.min-w-\\[60px\\]');
+        const lastDiv = responseDivs[responseDivs.length - 1];
+        const responseText = await page.evaluate(el => el.innerText, lastDiv);
 
         const currentUrl = page.url();
         const chatIdMatch = currentUrl.match(/\/conversation\/([^/?]+)/);
@@ -97,7 +99,8 @@ const handler = async (req, res) => {
             success: true,
             data: {
                 msg: result.response,
-                chatId: result.chatId
+                chatId: result.chatId,
+                url: result.url
             }
         });