Add jimp and scaled image

electron/index.ts

@@ -12,6 +12,7 @@ import { join, parse } from "path";
 import log from "electron-log";
 import { format } from "url";
 import fs from "fs";
+import Jimp from "jimp";
 
 import { execPath, modelsPath } from "./binaries";
 // Packages
@@ -22,6 +23,7 @@ import {
   dialog,
   shell,
   MessageBoxOptions,
+  protocol,
 } from "electron";
 
 import { spawnUpscayl } from "./upscayl";
@@ -71,6 +73,7 @@ app.on("ready", async () => {
     backgroundColor: "#171717",
     webPreferences: {
       nodeIntegration: true,
+      nodeIntegrationInWorker: true,
       webSecurity: false,
       preload: join(__dirname, "preload.js"),
     },
@@ -96,6 +99,13 @@ app.on("ready", async () => {
     mainWindow.webContents.setZoomFactor(1);
   });
 
+  app.whenReady().then(() => {
+    protocol.registerFileProtocol("file", (request, callback) => {
+      const pathname = decodeURI(request.url.replace("file:///", ""));
+      callback(pathname);
+    });
+  });
+
   if (!isDev) {
     autoUpdater.checkForUpdates();
   }
@@ -473,7 +483,6 @@ ipcMain.on(commands.DOUBLE_UPSCAYL, async (event, payload) => {
 //------------------------Image Upscayl-----------------------------//
 ipcMain.on(commands.UPSCAYL, async (event, payload) => {
   const model = payload.model as string;
-  const scale = payload.scale as string;
   const gpuId = payload.gpuId as string;
   const saveImageAs = payload.saveImageAs as string;
 
@@ -490,12 +499,21 @@ ipcMain.on(commands.UPSCAYL, async (event, payload) => {
   const fileName = parse(fullfileName).name;
   const fileExt = parse(fullfileName).ext;
 
+  let scale = "4";
+  if (model.includes("x2")) {
+    scale = "2";
+  } else if (model.includes("x3")) {
+    scale = "3";
+  } else {
+    scale = "4";
+  }
+
   const outFile =
     outputDir +
     slash +
     fileName +
     "_upscayl_" +
-    scale +
+    payload.scale +
     "x_" +
     model +
     "." +
@@ -522,19 +540,6 @@ ipcMain.on(commands.UPSCAYL, async (event, payload) => {
       logit
     );
 
-    console.log(
-      "BRUH: " +
-        getSingleImageArguments(
-          inputDir,
-          fullfileName,
-          outFile,
-          isDefaultModel ? modelsPath : customModelsFolderPath ?? modelsPath,
-          model,
-          scale,
-          gpuId,
-          saveImageAs
-        )
-    );
     childProcesses.push(upscayl);
 
     stopped = false;
@@ -563,12 +568,26 @@ ipcMain.on(commands.UPSCAYL, async (event, payload) => {
     const onClose = () => {
       if (!failed && !stopped) {
         logit("💯 Done upscaling");
+        logit("♻ Scaling and converting now...");
+        Jimp.read(
+          isAlpha ? outFile + ".png" : outFile,
+          (err: any, image: any) => {
+            if (err) {
+              logit("❌ Error converting to PNG: ", err);
+              onError(err);
+              return;
+            }
+            image
+              .scale(parseInt(payload.scale as string))
+              .write(isAlpha ? outFile + ".png" : outFile);
             mainWindow.setProgressBar(-1);
             mainWindow.webContents.send(
               commands.UPSCAYL_DONE,
               isAlpha ? outFile + ".png" : outFile
             );
           }
+        );
+      }
     };
 
     upscayl.process.stderr.on("data", onData);

package.json

@@ -179,6 +179,7 @@
     "electron-next": "^3.1.5",
     "electron-updater": "^5.3.0",
     "image-size": "^1.0.2",
+    "jimp": "^0.22.8",
     "jotai": "^2.0.4",
     "react-compare-slider": "^2.2.0",
     "react-dropzone": "^14.2.3",

renderer/components/upscayl-tab/config/LeftPaneImageSteps.tsx

@@ -1,4 +1,4 @@
-import { useAtom, useAtomValue } from "jotai";
+import { useAtomValue } from "jotai";
 import React, { useEffect, useState } from "react";
 import Select from "react-select";
 import ReactTooltip from "react-tooltip";

renderer/pages/index.tsx

@@ -125,7 +125,7 @@ const Home = () => {
     window.electron.on(commands.UPSCAYL_DONE, (_, data: string) => {
       setProgress("");
       setUpscaledImagePath(data);
-      logit("upscaledImagePath: ", upscaledImagePath);
+      logit("upscaledImagePath: ", data);
       logit(`💯 UPSCAYL_DONE: `, data);
     });