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

fix: handle HEIC images in blur-faces and red-eye-removal, show warning when no faces detected

Browse Source
This commit is contained in:
ashim-hq
2026-04-19 19:52:23 +08:00
parent 10bdc24a4a
commit 8c83d7efcd
8 changed files with 147 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/ai/src/face-detection.ts
+5 -2
View File
@@ -1,6 +1,7 @@
import { readFile, unlink, writeFile } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";
import sharp from "sharp";
import { type ProgressCallback, parseStdoutJson, runPythonWithProgress } from "./bridge.js";
export interface BlurFacesOptions {
@@ -39,7 +40,8 @@ export async function blurFaces(
const inputPath = join(outputDir, "input_faces.png");
const outputPath = join(outputDir, "output_faces.png");
await writeFile(inputPath, inputBuffer);
const pngBuffer = await sharp(inputBuffer).png().toBuffer();
await writeFile(inputPath, pngBuffer);
const { stdout } = await runPythonWithProgress(
"detect_faces.py",
[inputPath, outputPath, JSON.stringify(options)],
@@ -67,7 +69,8 @@ export async function detectFaces(
const inputPath = join(tmpdir(), `detect_faces_${Date.now()}.png`);
try {
await writeFile(inputPath, inputBuffer);
const pngBuffer = await sharp(inputBuffer).png().toBuffer();
await writeFile(inputPath, pngBuffer);
const { stdout } = await runPythonWithProgress(
"detect_faces.py",
[inputPath, "unused", JSON.stringify({ ...options, detectOnly: true })],
packages/ai/src/red-eye-removal.ts
+3 -1
View File
@@ -1,5 +1,6 @@
import { readFile, writeFile } from "node:fs/promises";
import { join } from "node:path";
import sharp from "sharp";
import { type ProgressCallback, parseStdoutJson, runPythonWithProgress } from "./bridge.js";
export interface RedEyeRemovalOptions {
@@ -27,7 +28,8 @@ export async function removeRedEye(
const inputPath = join(outputDir, "input_redeye.png");
const outputPath = join(outputDir, "output_redeye.png");
await writeFile(inputPath, inputBuffer);
const pngBuffer = await sharp(inputBuffer).png().toBuffer();
await writeFile(inputPath, pngBuffer);
const { stdout } = await runPythonWithProgress(
"red_eye_removal.py",
[inputPath, outputPath, JSON.stringify(options)],