Update index.mjs

index.mjs

@@ -1,73 +1,63 @@
+// 🌍 Importing Necessary Modules and Libraries 🌍
 import express from "express";
 import 'dotenv/config.js'
 import { createServer } from "http";
-import { Server } from "socket.io";
-import { client } from "@gradio/client";
+import { Server as SocketIOServer } from "socket.io";
+import { client as GradioClient } from "@gradio/client";
 import { createRequire } from 'node:module'
-import fs from 'fs';
-import path from 'path';
+
+// 🚀 Global Configurations 🚀
 const require = createRequire(import.meta.url);
 global.EventSource = require('eventsource');
 
+// 📱 Express App Setup 📱
 const app = express();
 const httpServer = createServer(app);
 app.use(express.static('public'));
-const io = new Server(httpServer, { /* options */ });
 
+// 🔌 Socket.io Setup 🔌
+const io = new SocketIOServer(httpServer, { /* options */ });
+
+// 🛰️ Socket.io Connection Handler 🛰️
 io.on("connection", (socket) => {
-    console.log("new socket connection");
     
-    socket.on("ask_api", (client_data) => {
-        console.log(client_data)
-        console.log("trying to reach api");   
-        asyncAPICall(client_data, socket)
+    console.log("🌐 New Socket Connection Established");
+    
+    socket.on("ask_api", (clientData) => {
+        console.log(clientData)
+        console.log("🔍 Trying to Reach API");   
+        callGradioAPIAsync(clientData, socket)
     });
 });
 
-async function asyncAPICall(data, socket) {
-    const grapi = await client("fffiloni/mndrm-call");
-    try {
-        const api_result = await grapi.predict("/infer", [
-            data[0], // blob in 'image' Image component		
-            data[1], // string in 'Question' Textbox component
-        ]); 
-        console.log(api_result)
-
-        // Save the image and provide a download link
-        if(api_result && api_result.data){
-            const filename = createFileName(api_result.data.text);
-            saveImage(data[0], filename);
-            socket.emit("api_response", { ...api_result.data, downloadLink: `/download/${filename}` });
-        }
-    } catch(e) {
-        console.log(e)
-        socket.emit("api_error", ("ERROR ON API SIDE, SORRY..."))
-    }    
-}
-
-// Function to create a filename from the caption and current datetime
-function createFileName(caption) {
-    const datetime = new Date().toISOString().replace(/:/g, '-');
-    return `${caption}-${datetime}.png`;
+// 🤖 Gradio API Testing Function (Uncomment to test) 🤖
+// async function testGradioServers(){
+//   try {
+//     const gradioTestClient = await GradioClient("https://gradio-hello-world.hf.space");
+//     const apiTestResult = await gradioTestClient.predict("/predict", ["John"]);
+//     console.log(apiTestResult);
+//   } catch (error) {
+//     console.log(error);
+//   }
+// }
+// testGradioServers();
+
+// 🌟 Async Function to Call Gradio API 🌟
+async function callGradioAPIAsync(clientData, socket) {
+  const gradioClient = await GradioClient("fffiloni/mndrm-call");
+  try {
+      const apiResult = await gradioClient.predict("/infer", [
+          clientData[0], // blob in 'image' Image component		
+          clientData[1], // string in 'Question' Textbox component
+      ]); 
+      console.log(apiResult)
+      socket.emit("api_response", (apiResult.data))
+  } catch (error) {
+      console.log(error)
+      socket.emit("api_error", ("ERROR ON API SIDE, SORRY..."))
+  }    
 }
 
-// Function to save image data to a file
-function saveImage(imageData, filename) {
-    const buffer = Buffer.from(imageData.split(",")[1], 'base64');
-    fs.writeFileSync(path.join(__dirname, 'public', filename), buffer);
-}
-
-// Express route to download the image
-app.get('/download/:filename', (req, res) => {
-    const filename = req.params.filename;
-    const filepath = path.join(__dirname, 'public', filename);
-    res.download(filepath, filename, (err) => {
-        if (err) {
-            console.error(err);
-            res.status(500).send("Error occurred while downloading the file.");
-        }
-    });
-});
-
+// 🏃 Server Startup 🏃
 httpServer.listen(7860);
-console.log("App running on localhost:7860")
+console.log("🚀 App Running on http://localhost:7860");