index.mjs

import express from "express";
import 'dotenv/config.js';
import { createServer } from "http";
import { Server } from "socket.io";
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, { /* options */ });

io.on("connection", (socket) => {
    console.log("new socket connection 🎉");
    
    socket.on("ask_api", (client_data) => {
        console.log("Received data from client 📸", client_data);
        console.log("Attempting to reach API... 🚀");   
        asyncAPICall(client_data, 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 response received ✅", api_result);
      socket.emit("api_response", (api_result.data));
  } catch (e) {
      console.error("API call failed ❌", e);
      socket.emit("api_error", ("ERROR ON API SIDE, SORRY..."));
  }    
}

httpServer.listen(7860, () => console.log("App running on localhost:7860 🌍"));