Update index.mjs

index.mjs

@@ -1,63 +1,60 @@
-// 🌍 Importing Necessary Modules and Libraries 🌍
 import express from "express";
-import 'dotenv/config.js'
+import 'dotenv/config.js';
 import { createServer } from "http";
-import { Server as SocketIOServer } from "socket.io";
-import { client as GradioClient } from "@gradio/client";
-import { createRequire } from 'node:module'
+import { Server } from "socket.io";
+import { client as gradioClient } from "@gradio/client";
+import { createRequire } from 'node:module';
 
-// 🚀 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);
 
-// 🔌 Socket.io Setup 🔌
-const io = new SocketIOServer(httpServer, { /* options */ });
+// 🌐 Server Initialization
+httpServer.listen(7860, () => {
+    console.log("App running on localhost:7860");
+});
 
-// 🛰️ Socket.io Connection Handler 🛰️
+// 🔌 Socket Connection Handler
 io.on("connection", (socket) => {
-    
-    console.log("🌐 New Socket Connection Established");
-    
+    console.log("New socket connection");
+
+    // 📡 API Request Listener
     socket.on("ask_api", (clientData) => {
-        console.log(clientData)
-        console.log("🔍 Trying to Reach API");   
-        callGradioAPIAsync(clientData, socket)
+        console.log("Received client data:", clientData);
+        console.log("Contacting API...");
+        handleAPICall(clientData, socket);
     });
 });
 
-// 🤖 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..."))
-  }    
+// 🚀 API Call Function
+async function handleAPICall(data, socket) {
+    const grapi = await gradioClient("fffiloni/mndrm-call");
+
+    try {
+        const apiResult = await grapi.predict("/infer", [data[0], data[1]]); 
+        console.log("API Result:", apiResult);
+        socket.emit("api_response", apiResult.data);
+    } catch (error) {
+        console.error("API Error:", error);
+        socket.emit("api_error", "ERROR ON API SIDE, SORRY...");
+    }
 }
 
-// 🏃 Server Startup 🏃
-httpServer.listen(7860);
-console.log("🚀 App Running on http://localhost:7860");
+// 🧪 Optional Server Test Function
+// Uncomment to test server connections
+/*
+async function testServers() {
+    try {
+        const grapiTest = await gradioClient("https://gradio-hello-world.hf.space");
+        const apiTestResult = await grapiTest.predict("/predict", ["John"]);
+        console.log("Test API Result:", apiTestResult);
+    } catch (error) {
+        console.error("Test Error:", error);
+    }
+}
+// testServers();
+*/