Cool_Tools/upscayl
1
0
Fork 0
mirror of https://github.com/upscayl/upscayl.git synced 2024-11-24 07:30:19 +01:00
Code Releases Wiki Activity

implemented working batch mode frontend

Browse Source
This commit is contained in:
TGS963 2022-09-18 17:47:49 +05:30
parent 313a7db4cb
commit d2abeb1f13
2 changed files with 26 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
main/index.js
View File

@ -223,7 +223,6 @@ ipcMain.on(commands.UPSCAYL, async (event, payload) => {
const scale = payload.scaleFactor;
let inputDir = payload.imagePath.match(/(.*)[\/\\]/)[1] || "";
let outputDir = payload.outputPath;
console.log(outputDir);
// COPY IMAGE TO TMP FOLDER
const platform = getPlatform();
@ -293,10 +292,12 @@ ipcMain.on(commands.UPSCAYL, async (event, payload) => {
//------------------------Upscayl Folder-----------------------------//
ipcMain.on(commands.FOLDER_UPSCAYL, async (event, payload) => {
const model = payload.model;
let inputDir = payload.imagePath;
let inputDir = payload.batchFolderPath;
let outputDir = payload.outputPath;
console.log(outputDir);
if (!fs.existsSync(outputDir)) {
fs.mkdirSync(outputDir, { recursive: true });
}
// UPSCALE
let upscayl = spawn(
execPath("realesrgan"),
@ -336,7 +337,7 @@ ipcMain.on(commands.FOLDER_UPSCAYL, async (event, payload) => {
upscayl.on("close", (code) => {
if (failed !== true) {
console.log("Done upscaling");
mainWindow.webContents.send(commands.FOLDER_UPSCAYL_DONE, outFile);
mainWindow.webContents.send(commands.FOLDER_UPSCAYL_DONE, outputDir);
}
});
});

25
renderer/pages/index.jsx
View File

@ -21,6 +21,7 @@ const Home = () => {
const [loaded, setLoaded] = useState(false);
const [version, setVersion] = useState("");
const [batchMode, setBatchMode] = useState(false);
const [batchFolderPath, setBatchFolderPath] = useState("");
const [doubleUpscayl, setDoubleUpscayl] = useState(false);
const resetImagePaths = () => {
@ -90,6 +91,14 @@ const Home = () => {
}
};
const selectFolderHandler = async () => {
var path = await window.electron.invoke(commands.SELECT_FOLDER);
if (path !== "cancelled") {
setBatchFolderPath(path);
SetOutputPath(path + "_upscayled");
}
};
const handleModelChange = (e) => {
setModel(e.target.value);
};
@ -164,7 +173,7 @@ const Home = () => {
const upscaylHandler = async () => {
setUpscaledImagePath("");
if (imagePath !== "") {
if (imagePath !== "" || batchFolderPath !== "") {
setProgress("Hold on...");
if (model === "models-DF2K") {
setDoubleUpscayl(false);
@ -176,6 +185,13 @@ const Home = () => {
outputPath,
model,
});
} else if (batchMode) {
await window.electron.send(commands.FOLDER_UPSCAYL, {
scaleFactor,
batchFolderPath,
outputPath,
model,
});
} else {
await window.electron.send(commands.UPSCAYL, {
scaleFactor,
@ -202,6 +218,7 @@ const Home = () => {
<LeftPaneSteps
progress={progress}
selectImageHandler={selectImageHandler}
selectFolderHandler={selectFolderHandler}
handleModelChange={handleModelChange}
handleDrop={handleDrop}
outputHandler={outputHandler}
@ -231,7 +248,7 @@ const Home = () => {
<ProgressBar progress={progress} />
)}
{imagePath.length === 0 ? (
{imagePath.length === 0 && batchFolderPath.length === 0 ? (
<RightPaneInfo version={version} />
) : upscaledImagePath.length === 0 ? (
<img
@ -242,7 +259,7 @@ const Home = () => {
draggable="false"
alt=""
/>
) : (
) : !batchMode ? (
<ReactCompareSlider
itemOne={
<ReactCompareSliderImage
@ -264,7 +281,7 @@ const Home = () => {
}
className="h-screen"
/>
)}
) : null}
</div>
</div>
);