helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-11-14 17:36:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

refactor(ocr): update OCR schema and response structure to use individual coordinates instead of bounding box, and adjust related service and repository files

Browse source
This commit is contained in:
CoderKang 2025-06-02 22:26:49 +08:00 • committed by mertalev
parent 46ef02342d
commit df36a09cd3
No known key found for this signature in database
GPG key ID: DF6ABC77AAD98C95
7 changed files with 66 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

25
server/src/services/ocr.service.ts
View file

@ -66,14 +66,7 @@ export class OcrService extends BaseService {
machineLearning.ocr
);
const resultsArray = Array.isArray(ocrResults) ? ocrResults : [ocrResults];
const validResults = resultsArray.filter(result =>
result &&
result.text &&
result.text.trim().length > 0
);
if (validResults.length === 0) {
if (ocrResults.length === 0) {
this.logger.warn(`No valid OCR results for document ${id}`);
await this.assetRepository.upsertJobStatus({
assetId: asset.id,
@ -83,12 +76,16 @@ export class OcrService extends BaseService {
}
try {
const ocrDataList = validResults.map(result => ({
const ocrDataList = ocrResults.map(result => ({
assetId: id,
boundingBoxX1: result.boundingBox.x1,
boundingBoxY1: result.boundingBox.y1,
boundingBoxX2: result.boundingBox.x2,
boundingBoxY2: result.boundingBox.y2,
x1: result.x1,
y1: result.y1,
x2: result.x2,
y2: result.y2,
x3: result.x3,
y3: result.y3,
x4: result.x4,
y4: result.y4,
text: result.text.trim(),
}));
@ -99,7 +96,7 @@ export class OcrService extends BaseService {
ocrAt: new Date(),
});
this.logger.debug(`Processed ${validResults.length} OCR result(s) for ${id}`);
this.logger.debug(`Processed ${ocrResults.length} OCR result(s) for ${id}`);
return JobStatus.SUCCESS;
} catch (error) {
this.logger.error(`Failed to insert OCR results for ${id}:`, error);