Update index.mjs

index.mjs

@@ -1,19 +1,9 @@
-// At the top of your index.mjs or wherever you're planning to use fetch
-import fetch from 'node-fetch';
-if (!globalThis.fetch) {
-    globalThis.fetch = fetch;
-}
 import express from "express";
-import 'dotenv/config.js';
+import 'dotenv/config.js'
 import { createServer } from "http";
 import { Server } from "socket.io";
 import { client } from "@gradio/client";
-import { createRequire } from 'node:module';
-import { writeFile, readFileSync, appendFileSync } from 'node:fs';
-import { join } from 'node:path';
-import { randomBytes } from 'node:crypto';
-import base64url from 'base64-url';
-
+import { createRequire } from 'node:module'
 const require = createRequire(import.meta.url);
 global.EventSource = require('eventsource');
 
@@ -22,80 +12,53 @@ const httpServer = createServer(app);
 app.use(express.static('public'));
 const io = new Server(httpServer, { /* options */ });
 
-const textHistoryPath = join(__dirname, 'public', 'history.txt');
-
 io.on("connection", (socket) => {
-    console.log("new socket connection 🎉");
+    
+    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);
+        console.log(client_data)
+        console.log("trying 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);
-      
-      // Save image and text
-      const imageName = saveImage(api_result.data.image); // Assuming api_result.data.image is the image blob or base64 string
-      const textName = saveText(api_result.data.text, imageName); // Assuming api_result.data.text is the text to be saved
-      
-      // Emit the response with download links
-      socket.emit("api_response", {
-        ...api_result.data,
-        imageDownloadLink: `/download/${imageName}`,
-        textDownloadLink: `/download/${textName}`
-      });
-      
-      updateHistory(api_result.data.text, imageName);
-      
-  } catch (e) {
-      console.error("API call failed ❌", e);
-      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)
+  }
+  
 }
 
-function saveImage(imageData) {
-    const imageName = `image_${randomBytes(8).toString('hex')}.png`;
-    writeFile(join(__dirname, 'public', imageName), imageData, 'base64', (err) => {
-        if (err) throw err;
-        console.log(`Image saved as ${imageName}`);
-    });
-    return imageName;
-}
+//test_servers()
 
-function saveText(text, imageName) {
-    const textName = `text_${randomBytes(8).toString('hex')}.txt`;
-    writeFile(join(__dirname, 'public', textName), `${text}\nImage File: ${imageName}\n`, (err) => {
-        if (err) throw err;
-        console.log(`Text saved as ${textName}`);
-    });
-    return textName;
-}
+async function asyncAPICall(data, socket) {
 
-function updateHistory(text, imageName) {
-    appendFileSync(textHistoryPath, `${text}\nImage File: ${imageName}\n`);
+  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..."))
+  }    
+ 
 }
 
-// Serve files for download
-app.get('/download/:name', (req, res) => {
-    const fileName = req.params.name;
-    const filePath = join(__dirname, 'public', fileName);
-    res.download(filePath);
-});
 
-// Serve history.txt encoded in Base64
-app.get('/history', (req, res) => {
-    const history = readFileSync(textHistoryPath, 'utf8');
-    const encodedHistory = base64url.encode(history);
-    res.send(`<a href="data:text/plain;base64,${encodedHistory}" download="history.txt">Download History</a>`);
-});
 
-httpServer.listen(7860, () => console.log("App running on localhost:7860 🌍"));
+httpServer.listen(7860);
+console.log("App running on localhost:7860")