mirror of
https://github.com/snapotter-hq/SnapOtter.git
synced 2026-08-03 07:46:42 +02:00
SnapOtter
dee9452c48
Closes #17, #18, #19, #31, #32, #33, #34 Format preservation (#17, #18, #19): - Add resolveOutputFormat to rotate, resize, text-overlay, watermark-text, border, replace-color, blur-faces, upscale, erase-object, restore-photo - Alpha-aware fallback: border with corner radius/shadow and replace-color with makeTransparent fall back to PNG for non-alpha formats (JPEG) - Python sidecar tools (blur-faces, upscale, erase-object) now convert PNG output back to input format, matching restore-photo/colorize pattern - Upscale and erase-object default to "auto" format detection instead of PNG Dispatcher stability (#31, #32): - Add gc.collect() and torch.cuda.empty_cache() after each dispatcher request - Add configurable max_requests (default 50) for periodic dispatcher restart - Add exponential backoff to dispatcher crash recovery in bridge.ts - Circuit breaker: 5 crashes within 60s permanently disables dispatcher - Reset crash counter on successful dispatcher startup Health & security (#33, #34): - Export getDispatcherStatus() from @snapotter/ai with running/ready/failed/ gpu/pid/consecutiveCrashes fields - Admin health endpoint now includes full dispatcher status - Add pip-audit job to CI workflow for Python dependency scanning
92 lines
2.9 KiB
TypeScript
92 lines
2.9 KiB
TypeScript
import type { FastifyInstance } from "fastify";
|
|
import sharp from "sharp";
|
|
import { z } from "zod";
|
|
import { resolveOutputFormat } from "../../lib/output-format.js";
|
|
import { createToolRoute } from "../tool-factory.js";
|
|
|
|
const settingsSchema = z.object({
|
|
text: z.string().min(1).max(500),
|
|
fontSize: z.number().min(8).max(200).default(48),
|
|
color: z
|
|
.string()
|
|
.regex(/^#[0-9a-fA-F]{6}$/)
|
|
.default("#FFFFFF"),
|
|
position: z.enum(["top", "center", "bottom"]).default("bottom"),
|
|
backgroundBox: z.boolean().default(false),
|
|
backgroundColor: z
|
|
.string()
|
|
.regex(/^#[0-9a-fA-F]{6}$/)
|
|
.default("#000000"),
|
|
shadow: z.boolean().default(true),
|
|
});
|
|
|
|
function escapeXml(text: string): string {
|
|
return text
|
|
.replace(/&/g, "&")
|
|
.replace(/</g, "<")
|
|
.replace(/>/g, ">")
|
|
.replace(/"/g, """)
|
|
.replace(/'/g, "'");
|
|
}
|
|
|
|
export function registerTextOverlay(app: FastifyInstance) {
|
|
createToolRoute(app, {
|
|
toolId: "text-overlay",
|
|
settingsSchema,
|
|
process: async (inputBuffer, settings, filename) => {
|
|
const outputFormat = await resolveOutputFormat(inputBuffer, filename);
|
|
const image = sharp(inputBuffer);
|
|
const metadata = await image.metadata();
|
|
const width = metadata.width ?? 800;
|
|
const height = metadata.height ?? 600;
|
|
const escapedText = escapeXml(settings.text);
|
|
|
|
let y: number;
|
|
const pad = settings.fontSize;
|
|
|
|
switch (settings.position) {
|
|
case "top":
|
|
y = pad + settings.fontSize;
|
|
break;
|
|
case "center":
|
|
y = height / 2;
|
|
break;
|
|
default:
|
|
y = height - pad;
|
|
break;
|
|
}
|
|
|
|
const x = width / 2;
|
|
|
|
// Build SVG text element with optional effects
|
|
let filter = "";
|
|
let filterRef = "";
|
|
if (settings.shadow) {
|
|
filter = `<defs><filter id="shadow"><feDropShadow dx="2" dy="2" stdDeviation="3" flood-color="rgba(0,0,0,0.7)"/></filter></defs>`;
|
|
filterRef = ' filter="url(#shadow)"';
|
|
}
|
|
|
|
let bgRect = "";
|
|
if (settings.backgroundBox) {
|
|
const boxH = settings.fontSize * 1.8;
|
|
const boxY = y - settings.fontSize * 0.9;
|
|
bgRect = `<rect x="0" y="${boxY}" width="${width}" height="${boxH}" fill="${settings.backgroundColor}" opacity="0.7"/>`;
|
|
}
|
|
|
|
const svgOverlay = `<svg width="${width}" height="${height}">
|
|
${filter}
|
|
${bgRect}
|
|
<text x="${x}" y="${y}" font-size="${settings.fontSize}" fill="${settings.color}" font-family="sans-serif" text-anchor="middle" dominant-baseline="middle"${filterRef}>${escapedText}</text>
|
|
</svg>`;
|
|
|
|
const svgBuffer = Buffer.from(svgOverlay);
|
|
const result = await image.composite([{ input: svgBuffer, top: 0, left: 0 }]);
|
|
const buffer = await result
|
|
.toFormat(outputFormat.format, { quality: outputFormat.quality })
|
|
.toBuffer();
|
|
|
|
return { buffer, filename, contentType: outputFormat.contentType };
|
|
},
|
|
});
|
|
}
|