90 lines
2.9 KiB
JavaScript
90 lines
2.9 KiB
JavaScript
|
import { NextResponse } from 'next/server';
|
|||
|
|
import { PrismaClient } from '@prisma/client';
|
|||
|
|
import { getImageById, getImageChunk } from '@/lib/db/images';
|
|||
|
|
import { createImageDataset } from '@/lib/db/imageDatasets';
|
|||
|
|
|
|||
|
|
const prisma = new PrismaClient();
|
|||
|
|
|
|||
|
|
// 创建标注
|
|||
|
|
export async function POST(request, { params }) {
|
|||
|
|
try {
|
|||
|
|
const { projectId } = params;
|
|||
|
|
const { imageId, questionId, question, answerType, answer, note } = await request.json();
|
|||
|
|
|
|||
|
|
// 验证必填字段
|
|||
|
|
if (!imageId || !question || !answerType || answer === undefined || answer === null) {
|
|||
|
|
return NextResponse.json({ error: '缺少必要参数：imageId, question, answerType, answer' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证图片存在
|
|||
|
|
const image = await getImageById(imageId);
|
|||
|
|
if (!image || image.projectId !== projectId) {
|
|||
|
|
return NextResponse.json({ error: '图片不存在' }, { status: 404 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证答案类型
|
|||
|
|
if (!['text', 'label', 'custom_format'].includes(answerType)) {
|
|||
|
|
return NextResponse.json({ error: '无效的答案类型' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证答案内容
|
|||
|
|
if (answerType === 'text' && typeof answer !== 'string') {
|
|||
|
|
return NextResponse.json({ error: '文本类型答案必须是字符串' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
if (answerType === 'label' && !Array.isArray(answer)) {
|
|||
|
|
return NextResponse.json({ error: '标签类型答案必须是数组' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 序列化答案
|
|||
|
|
let answerString = answer;
|
|||
|
|
if (answerType !== 'text' && typeof answerString !== 'string') {
|
|||
|
|
answerString = JSON.stringify(answer, null, 2);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 1. 获取问题记录（前端传递的 questionId 指向已有的问题）
|
|||
|
|
if (!questionId) {
|
|||
|
|
return NextResponse.json({ error: '缺少必要参数：questionId' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const questionRecord = await prisma.questions.findUnique({
|
|||
|
|
where: { id: questionId }
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
if (!questionRecord) {
|
|||
|
|
return NextResponse.json({ error: '问题不存在' }, { status: 404 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证问题属于该图片
|
|||
|
|
if (questionRecord.imageId !== imageId) {
|
|||
|
|
return NextResponse.json({ error: '问题不属于该图片' }, { status: 400 });
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 2. 更新问题为已回答
|
|||
|
|
await prisma.questions.update({
|
|||
|
|
where: { id: questionRecord.id },
|
|||
|
|
data: { answered: true }
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
// 3. 创建 ImageDataset 记录
|
|||
|
|
const dataset = await createImageDataset(projectId, {
|
|||
|
|
imageId: image.id,
|
|||
|
|
imageName: image.imageName,
|
|||
|
|
questionId: questionRecord.id,
|
|||
|
|
question,
|
|||
|
|
answer: answerString,
|
|||
|
|
answerType,
|
|||
|
|
model: 'manual',
|
|||
|
|
note: note || ''
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
return NextResponse.json({
|
|||
|
|
success: true,
|
|||
|
|
dataset,
|
|||
|
|
questionId: questionRecord.id
|
|||
|
|
});
|
|||
|
|
} catch (error) {
|
|||
|
|
console.error('Failed to create annotation:', error);
|
|||
|
|
return NextResponse.json({ error: error.message || 'Failed to create annotation' }, { status: 500 });
|
|||
|
|
}
|
|||
|
|
}
|