Update lib/enchance.js

lib/enchance.js

@@ -1,6 +1,7 @@
 import WebSocket from "ws"
 import Jimp from "jimp"
 import FormData from "form-data"
+import axios from "axios"
 
 async function upscaler(images) {
   return new Promise((resolve, reject) => {
@@ -50,49 +51,37 @@ async function upscaler(images) {
   })
 
 }
-async function processImageUpscaler({ buffer, method, filenames, mimetype }) {
-    return new Promise(async (resolve, reject) => {
-        let Methods = ["enhance"];
-        Methods.includes(method) ? (method = method) : (method = Methods[0]);
 
-        let buffer,
-            Form = new FormData(),
-            scheme = "https" + "://" + "inferenceengine" + ".vyro" + ".ai/" + method;
+async function processImageUpscaler({ urlPath, method, filenames, mimetype }) {
+  return new Promise(async (resolve, reject) => {
+    let Methods = ["enhance"];
+    Methods.includes(method)? (method = method) : (method = Methods[0]);
 
-        Form.append("model_version", 1);
+    let buffer,
+        formData = new FormData(),
+        scheme = "https://" + "inferenceengine" + ".vyro" + ".ai/" + method;
 
-        Form.append("image", buffer, {
-            filename: filenames,
-            contentType: mimetype,
-        });
+    formData.append("model_version", 1);
 
-        Form.submit({
-                url: scheme,
-                host: "inferenceengine" + ".vyro" + ".ai",
-                path: "/" + method,
-                protocol: "https:",
-                headers: {
-                    "User-Agent": "okhttp/4.9.3",
-                    Connection: "Keep-Alive",
-                    "Accept-Encoding": "gzip",
-                },
-            },
-            function(err, res) {
-                if (err) reject(err);
-                let data = [];
-                res
-                    .on("data", function(chunk) {
-                        data.push(chunk);
-                    })
-                    .on("end", () => {
-                        resolve(Buffer.concat(data));
-                    });
-                res.on("error", (e) => {
-                    reject(e);
-                });
-            }
-        );
+    formData.append("image", await fs.promises.readFile(urlPath), {
+      filename: filenames,
+      contentType: mimetype,
     });
-};
+
+    const headers = {
+      "User-Agent": "okhttp/4.9.3",
+      Connection: "Keep-Alive",
+      "Accept-Encoding": "gzip",
+    };
+
+    axios.post(scheme, formData, { headers })
+     .then(response => {
+        resolve(response.data);
+      })
+     .catch(error => {
+        reject(error);
+      });
+  });
+}
 
 export { upscaler, processImageUpscaler }