update: export from starry-refactor 2026-02-20 17:38

backend/omr-service/src/routes/predictor.ts

@@ -10,6 +10,25 @@ interface PredictorBody {
 	images: string[];
 }
 
+/** Recursively convert Buffer objects in 'image' fields to base64 data URLs. */
+function convertBufferImages(obj: any): any {
+	if (obj == null || typeof obj !== 'object') return obj;
+	if (Buffer.isBuffer(obj)) return obj;
+	if (Array.isArray(obj)) return obj.map(convertBufferImages);
+
+	const result: any = {};
+	for (const [key, value] of Object.entries(obj)) {
+		if (key === 'image' && Buffer.isBuffer(value)) {
+			result[key] = `data:image/png;base64,${Buffer.from(value).toString('base64')}`;
+		} else if (typeof value === 'object' && value !== null) {
+			result[key] = convertBufferImages(value);
+		} else {
+			result[key] = value;
+		}
+	}
+	return result;
+}
+
 export default async function predictorRoutes(fastify: FastifyInstance) {
 	// Layout predictor - returns detection results
 	fastify.post<{ Body: PredictorBody }>('/predict/layout', async (request, reply) => {
@@ -25,10 +44,7 @@ export default async function predictorRoutes(fastify: FastifyInstance) {
 
 		return {
 			code: 0,
-			data: results.map((result) => ({
-				...result,
-				image: result.image ? `data:image/png;base64,${Buffer.from(result.image).toString('base64')}` : undefined,
-			})),
+			data: results.map((result) => convertBufferImages(result)),
 		};
 	});
 
@@ -46,10 +62,7 @@ export default async function predictorRoutes(fastify: FastifyInstance) {
 
 		return {
 			code: 0,
-			data: results.map((result) => ({
-				...result,
-				image: result.image ? `data:image/png;base64,${Buffer.from(result.image).toString('base64')}` : undefined,
-			})),
+			data: results.map((result) => convertBufferImages(result)),
 		};
 	});
 
@@ -67,10 +80,7 @@ export default async function predictorRoutes(fastify: FastifyInstance) {
 
 		return {
 			code: 0,
-			data: results.map((result) => ({
-				...result,
-				image: result.image ? `data:image/png;base64,${Buffer.from(result.image).toString('base64')}` : undefined,
-			})),
+			data: results.map((result) => convertBufferImages(result)),
 		};
 	});