Update index.mjs

index.mjs

@@ -4,6 +4,11 @@ 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';
+
 const require = createRequire(import.meta.url);
 global.EventSource = require('eventsource');
 
@@ -12,6 +17,8 @@ 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 🎉");
     
@@ -30,11 +37,60 @@ async function asyncAPICall(data, socket) {
           data[1], // string in 'Question' Textbox component
       ]);
       console.log("API response received ✅", api_result);
-      socket.emit("api_response", (api_result.data));
+      
+      // 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..."));
   }    
 }
 
+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;
+}
+
+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;
+}
+
+function updateHistory(text, imageName) {
+    appendFileSync(textHistoryPath, `${text}\nImage File: ${imageName}\n`);
+}
+
+// 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 🌍"));