Create gpt-4-turbo.js

plugins/gpt-4-turbo.js

@@ -0,0 +1,123 @@
+const axios = require('axios');
+const fs = require('fs');
+const puppeteer = require('puppeteer');
+
+async function gpt4Skywork(question, chatId = null, imagePath = null) {
+    const isNewChat = !chatId;
+    const targetUrl = isNewChat 
+        ? 'https://tool-skywork.wemiaow.com' 
+        : `https://tool-skywork.wemiaow.com/conversation/${chatId}`;
+
+    let imageBase64 = null;
+    if (imagePath && fs.existsSync(imagePath)) {
+        const imageBuffer = fs.readFileSync(imagePath);
+        imageBase64 = imageBuffer.toString('base64');
+    }
+
+    const browser = await puppeteer.launch({
+        headless: true,
+        args: ['--no-sandbox', '--disable-setuid-sandbox']
+    });
+
+    try {
+        const page = await browser.newPage();
+        await page.goto(targetUrl, { waitUntil: 'networkidle2' });
+        await page.waitForTimeout(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);
+        }
+
+        await page.waitForSelector('textarea.scrollbar-custom', { timeout: 10000 });
+        await page.type('textarea.scrollbar-custom', question);
+        await page.waitForTimeout(500);
+
+        await page.click('button[type=submit][aria-label="Send message"]');
+
+        if (isNewChat) {
+            await page.waitForNavigation({ waitUntil: 'networkidle2', timeout: 10000 });
+        }
+        await page.waitForTimeout(3000);
+
+        let isGenerating = true;
+        let attempts = 0;
+        while (isGenerating && attempts < 180) {
+            const stopBtn = await page.$('button.stop-generating-btn');
+            if (!stopBtn) {
+                isGenerating = false;
+            } else {
+                await page.waitForTimeout(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);
+
+        const currentUrl = page.url();
+        const chatIdMatch = currentUrl.match(/\/conversation\/([^/?]+)/);
+        const extractedChatId = chatIdMatch ? chatIdMatch[1] : null;
+
+        await browser.close();
+
+        return {
+            response: responseText,
+            chatId: extractedChatId,
+            url: currentUrl
+        };
+    } catch (error) {
+        await browser.close();
+        throw error;
+    }
+}
+
+const handler = async (req, res) => {
+    try {
+        const { text, chatId, image } = req.query;
+
+        if (!text) {
+            return res.status(400).json({
+                success: false,
+                error: 'Missing required parameter: text'
+            });
+        }
+
+        const result = await gpt4Skywork(text, chatId || null, image || null);
+
+        res.json({
+            author: "Herza",
+            success: true,
+            data: {
+                msg: result.response,
+                chatId: result.chatId
+            }
+        });
+
+    } catch (error) {
+        res.status(500).json({
+            success: false,
+            error: error.message
+        });
+    }
+};
+
+module.exports = {
+    name: 'ChatGPT V4 Turbo',
+    description: 'Generate responses using GPT-4 Turbo AI with image support',
+    type: 'GET',
+    routes: ['api/AI/gpt-4-turbo'],
+    tags: ['ai', 'openai', 'vision'],
+    main: ['AI'],
+    parameters: ['text', 'chatId', 'image'],
+    enabled: true,
+    limit: 5,
+    handler
+};