Update index.js

index.js

@@ -13,9 +13,10 @@ app.use(express.json({ limit: "500mb" }));
 app.use(express.urlencoded({ limit: '500mb', extended: true }));
 app.use((req, res, next) => {
   next()
-})
+});
 
-const apikey = "@SadTeam77"
+const apikey = "@SadTeam77";
+process.env.GRADIO_CLIENT_DEBUG = 'true';
 
 app.all('/', (req, res) => {
 	const status = {}
@@ -72,6 +73,27 @@ app.post('/api/img2img', async (req, res) => {
 	}
 })
 
+app.post('/api/chatai/gpt4', async (req, res) => {
+	try {
+		console.log(req.body)
+		const { prompt, status } = req.body
+        if (!images) return res.json({ success: false, message: 'Required an images!' })
+        if (!prompt) return res.json({ succese: false, message: 'Require an Promot text!'})
+
+        if(status !== apikey) return res.json({ success: false, message: 'Invalid status!' })
+        const response = await chatgpt4(prompt);
+
+        res.json({
+          status: true,
+          result: response
+        });
+	} catch (e) {
+		console.log(e)
+		e = String(e)
+		res.json({ error: true, message: e === '[object Object]' ? 'Internal Server Error' : e })
+	}
+})
+
 const PORT = process.env.PORT || 7860
 app.listen(PORT, () => {
   console.log('App running on port', PORT)
@@ -93,8 +115,6 @@ async function processImage2Img(imgBuffer, prompt) {
     try {
       const imageArray = Buffer.from(imgBuffer);
 
-      process.env.GRADIO_CLIENT_DEBUG = 'true';
-
       const app = await Client.connect("Manjushri/SDXL-Turbo-Img2Img-CPU");
       const result = await app.predict("/predict", [
         imageArray, // binary input for the image
@@ -108,4 +128,23 @@ async function processImage2Img(imgBuffer, prompt) {
       reject(e.message);
     }
   });
+}
+
+async function chatgpt4(prompt) {
+  try {
+    const client = await Client.connect("KingNish/OpenGPT-4o");
+    const result = await client.predict("/chat", { 		
+      user_prompt: {"text": prompt,"files":[]}, 		
+	  model_selector: "idefics2-8b-chatty", 		
+	  decoding_strategy: "Greedy", 		
+	  temperature: 0, 		
+	  max_new_tokens: 2048, 		
+	  repetition_penalty: 0.01, 		
+	  top_p: 0.01, 		
+	  web_search: true, 
+    });
+    return result.data;
+  } catch(e) {
+    return e.message;
+  }
 }