Update lib/enchance.js

lib/enchance.js

@@ -1,4 +1,6 @@
 import WebSocket from "ws"
+import Jimp from "jimp"
+import FormData from "form-data"
 
 async function upscaler(images) {
   return new Promise((resolve, reject) => {
@@ -48,5 +50,49 @@ 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]);
 
-export { upscaler }
+        let buffer,
+            Form = new FormData(),
+            scheme = "https" + "://" + "inferenceengine" + ".vyro" + ".ai/" + method;
+
+        Form.append("model_version", 1);
+
+        Form.append("image", buffer, {
+            filename: filenames,
+            contentType: mimetype,
+        });
+
+        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);
+                });
+            }
+        );
+    });
+};
+
+export { upscaler, processImageUpscaler }