External/SnapOtter
1
0
Fork 0
mirror of https://github.com/snapotter-hq/SnapOtter.git synced 2026-08-03 07:46:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #48 from stirling-image/fix/upscale-bugs-and-features

Browse Source 
feat: overhaul upscale with bug fixes and advanced features
This commit is contained in:
stirling-image
2026-04-12 19:09:22 +08:00
committed by GitHub
parent d179021203 fe376aebd2
commit ed5f71e2fc
5 changed files with 412 additions and 100 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/api/src/routes/tools/upscale.ts
+10 -4
View File
@@ -56,8 +56,13 @@ export function registerUpscale(app: FastifyInstance) {
try {
const settings = settingsRaw ? JSON.parse(settingsRaw) : {};
const scale = Number(settings.scale) || 2;
const model = settings.model || "auto";
const faceEnhance = Boolean(settings.faceEnhance);
const denoise = Number(settings.denoise) || 0;
const format = settings.format || "png";
const outputQuality = Number(settings.quality) || 95;
request.log.info(
{ toolId: "upscale", imageSize: fileBuffer.length, scale },
{ toolId: "upscale", imageSize: fileBuffer.length, scale, model, format },
"Starting upscale",
);
@@ -87,12 +92,13 @@ export function registerUpscale(app: FastifyInstance) {
const result = await upscale(
fileBuffer,
join(workspacePath, "output"),
{ scale },
{ scale, model, faceEnhance, denoise, format, quality: outputQuality },
onProgress,
);
// Save output
const outputFilename = `${filename.replace(/\.[^.]+$/, "")}_${scale}x.png`;
// Save output with correct extension for the chosen format
const ext = result.format === "jpeg" ? "jpg" : result.format === "webp" ? "webp" : "png";
const outputFilename = `${filename.replace(/\.[^.]+$/, "")}_${scale}x.${ext}`;
const outputPath = join(workspacePath, "output", outputFilename);
await writeFile(outputPath, result.buffer);