Files
ai-toy-patent-workflow/src/app/api/character/lock/route.ts

51 lines
1.7 KiB
TypeScript

import { NextResponse } from 'next/server';
import { buildCharacterSpec, cleanupCharacterAnchor } from '@/lib/packGenerator';
import { detectProvider } from '@/lib/providers';
import { loadSession, saveSession } from '@/lib/storage';
import type { LockCharacterRequest, LockCharacterResponse } from '@/lib/types';
export const runtime = 'nodejs';
export const dynamic = 'force-dynamic';
export async function POST(req: Request) {
const { sessionId, imageId, force = false } = (await req.json()) as LockCharacterRequest;
if (!sessionId || !imageId) {
return NextResponse.json({ error: 'sessionId and imageId required' }, { status: 400 });
}
const session = await loadSession(sessionId);
if (!session) return NextResponse.json({ error: 'session not found' }, { status: 404 });
const sourceImage = session.images.find(image => image.id === imageId);
if (!sourceImage) return NextResponse.json({ error: 'image not found' }, { status: 404 });
if (!force && session.characterSpec?.sourceImageId === imageId) {
const response: LockCharacterResponse = {
characterSpec: session.characterSpec,
provider: detectProvider(),
};
return NextResponse.json(response);
}
try {
const characterSpec = await buildCharacterSpec(session, sourceImage);
const cleaned = await cleanupCharacterAnchor({
session,
sourceImage,
characterSpec,
force,
});
session.characterSpec = cleaned.characterSpec;
await saveSession(session);
const response: LockCharacterResponse = {
characterSpec: cleaned.characterSpec,
provider: detectProvider(),
};
return NextResponse.json(response);
} catch (error) {
return NextResponse.json({ error: String(error) }, { status: 500 });
}
}