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(client_data)
        console.log("trying to reach api");   
        asyncAPICall(client_data, socket)
    });

});

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)
  }
  
}

//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..."))
  }    
 
}



httpServer.listen(7860);
console.log("App running on localhost:7860")