Update index.js

index.js

@@ -2,6 +2,7 @@ import express from "express";
 import os from "os";
 import morgan from "morgan";
 import bytes from "bytes";
+import axios from "axios";
 import { FormData, Blob } from "formdata-node";
 import { fileTypeFromBuffer } from "file-type";
 import { Client } from "@gradio/client";
@@ -134,21 +135,47 @@ async function processImage2Img(imgBuffer, prompt) {
 
 async function askOpenGPT4o(prompt) {
   try {
-    process.env.GRADIO_CLIENT_DEBUG = 'true';
-    
-    const client = await Client.connect("KingNish/OpenGPT-4o");
-    const result = await client.predict("/chat", {
-      user_prompt: { text: prompt },
-      decoding_strategy: "Greedy",
-      temperature: 0,
-      max_new_tokens: 2048,
-      repetition_penalty: 0.01,
-      top_p: 0.01,
-      web_search: true,
+    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,
     });
-    return result.data;
+
+    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'));
+
+    if (processStartsLine) {
+      const processStartsData = JSON.parse(processStartsLine.replace('data: ', ''));
+      return processStartsData.output.data[0][0][1];
+    }
   } catch (error) {
-    console.error(error);
-    throw error; // Rethrow the error to allow for better error handling
+    console.error('Error occurred:', error);
+    return `Error: ${error.message}`;
   }
-}
+}
+