Update index.js

index.js

@@ -7,6 +7,7 @@ import { FormData, Blob } from "formdata-node";
 import { fileTypeFromBuffer } from "file-type";
 import { Client } from "@gradio/client";
 import { stablediff } from "./lib/diffusion.js"
+import { askOpenGPT4o } from "./lib/chatgpt.js"
 
 const app = express();
 app.set('json spaces', 4);
@@ -78,17 +79,45 @@ app.post('/api/img2img', async (req, res) => {
 app.post('/api/openai/gpt4', async (req, res) => {
 	try {
 		console.log(req.body)
-		const { prompt, status } = req.body
+		const { prompt, images status } = req.body
         if (!prompt) return res.json({ succese: false, message: 'Require an Promot text!'})
         if (!status) return res.json({ success: false, message: 'Required an status text!' })
 
         if(status !== apikey) return res.json({ success: false, message: 'Invalid status!' })
-        const response = await askOpenGPT4o(prompt);
-
-        res.json({
+        if(images) {
+          if (/^(https?|http):\/\//i.test(images)) {
+          const data_img = await axios.request({
+            method: "GET",
+            url: images,
+            responseType: "arraybuffer"
+          })
+          const response = await askOpenGPT4o(prompt, data_img.data)
+          //const type_img = await fileTypeFromBuffer(response)
+          //res.setHeader('Content-Type', type_img.mime)
+          res.json({
+            status: true,
+            result: response
+          })
+          } else if (images && typeof images == 'string' && isBase64(images)) {
+	   	    const response = await askOpenGPT4o(prompt, Buffer.from(images, "base64"))
+            //const type_img = await fileTypeFromBuffer(response)
+            //res.setHeader('Content-Type', type_img.mime)
+            res.json({
+              status: true,
+              result: response
+            })
+          } else {
+            res.json({
+              success: false, message: 'No url or base64 detected!!' 
+            })
+          }
+        } else if(!images) {
+          const response await askOpenGPT4o(prompt)
+          res.json({
           status: true,
           result: response
-        });
+          });
+        }
 	} catch (e) {
 		console.log(e)
 		e = String(e)
@@ -152,58 +181,4 @@ async function processImage2Img(imgBuffer, prompt) {
       reject(e.message);
     }
   });
-}
-
-async function askOpenGPT4o(prompt) {
-    try {
-      const session_hash = Math.random().toString(36).substring(2).slice(1);
-  
-      const resPrompt = await axios.post('https://kingnish-opengpt-4o.hf.space/run/predict?__theme=light', {
-        data: [{ text: prompt, files: [] }],
-        event_data: null,
-        fn_index: 3,
-        trigger_id: 34,
-        session_hash,
-      });
-  
-      const res = await axios.post('https://kingnish-opengpt-4o.hf.space/queue/join?__theme=light', {
-        data: [
-          null,
-          null,
-          'idefics2-8b-chatty',
-          'Top P Sampling',
-          0.5,
-          4096,
-          1,
-          0.9,
-          true,
-        ],
-        event_data: null,
-        fn_index: 5,
-        trigger_id: 34,
-        session_hash,
-      });
-  
-      const event_ID = res.data.event_id;
-  
-      const anu = await axios.get(`https://kingnish-opengpt-4o.hf.space/queue/data?session_hash=${session_hash}`);
-      const lines = anu.data.split('\n');
-      const processStartsLine = lines.find(line => line.includes('process_completed'));
-      const processStartsData = JSON.parse(processStartsLine.replace('data: ', ''));
-      
-      const processStartsLine_2 = lines.find(line => line.includes('process_generating'));
-      const processStartsData_2 = JSON.parse(processStartsLine_2.replace('data: ', ''));
-
-  
-      if (processStartsData?.success) {
-        return processStartsData.output.data[0][0][1];
-      } else if (processStartsData_2?.success) {
-        return processStartsData_2.output.data[0][0][1];
-      }
-      
-
-    } catch (error) {
-      console.error('Error occurred:', error);
-      return `Error: ${error.message}`;
-    }
-  }
+}