Update index.mjs

index.mjs

@@ -1,60 +1,64 @@
 import express from "express";
-import 'dotenv/config.js';
+import 'dotenv/config.js'
 import { createServer } from "http";
 import { Server } from "socket.io";
-import { client as gradioClient } from "@gradio/client";
-import { createRequire } from 'node:module';
-
+import { client } from "@gradio/client";
+import { createRequire } from 'node:module'
 const require = createRequire(import.meta.url);
 global.EventSource = require('eventsource');
 
 const app = express();
 const httpServer = createServer(app);
 app.use(express.static('public'));
-const io = new Server(httpServer);
-
-// 🌐 Server Initialization
-httpServer.listen(7860, () => {
-    console.log("App running on localhost:7860");
-});
+const io = new Server(httpServer, { /* options */ });
 
-// 🔌 Socket Connection Handler
 io.on("connection", (socket) => {
-    console.log("New socket connection");
-
-    // 📡 API Request Listener
-    socket.on("ask_api", (clientData) => {
-        console.log("Received client data:", clientData);
-        console.log("Contacting API...");
-        handleAPICall(clientData, 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)
     });
+
 });
 
-// 🚀 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...");
-    }
+async function test_servers(){
+  try{
+    const grapi_test = await client("https://gradio-hello-world.hf.space");
+    const apitest_result = await grapi_test.predict("/predict", [
+      "John",
+    ]);
+    console.log(apitest_result);
+  }
+  catch(e){
+    console.log(e)
+  }
+  
 }
 
-// 🧪 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);
-    }
+//test_servers()
+
+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)
+      socket.emit("api_response", (api_result.data))
+  }
+  catch(e){
+      console.log(e)
+      socket.emit("api_error", ("ERROR ON API SIDE, SORRY..."))
+  }    
+ 
 }
-// testServers();
-*/
+
+
+
+httpServer.listen(7860);
+console.log("App running on localhost:7860")