mirror of
https://github.com/samsonjs/immich.git
synced 2026-03-25 09:15:56 +00:00
02b29046b3
* feat: add OCR functionality and related configurations * chore: update labeler configuration for machine learning files * feat(i18n): enhance OCR model descriptions and add orientation classification and unwarping features * chore: update Dockerfile to include ccache for improved build performance * feat(ocr): enhance OCR model configuration with orientation classification and unwarping options, update PaddleOCR integration, and improve response structure * refactor(ocr): remove OCR_CLEANUP job from enum and type definitions * refactor(ocr): remove obsolete OCR entity and migration files, and update asset job status and schema to accommodate new OCR table structure * refactor(ocr): update OCR schema and response structure to use individual coordinates instead of bounding box, and adjust related service and repository files * feat: enhance OCR configuration and functionality - Updated OCR settings to include minimum detection box score, minimum detection score, and minimum recognition score. - Refactored PaddleOCRecognizer to utilize new scoring parameters. - Introduced new database tables for asset OCR data and search functionality. - Modified related services and repositories to support the new OCR features. - Updated translations for improved clarity in settings UI. * sql changes * use rapidocr * change dto * update web * update lock * update api * store positions as normalized floats * match column order in db * update admin ui settings descriptions fix max resolution key set min threshold to 0.1 fix bind * apply config correctly, adjust defaults * unnecessary model type * unnecessary sources * fix(ocr): switch RapidOCR lang type from LangDet to LangRec * fix(ocr): expose lang_type (LangRec.CH) and font_path on OcrOptions for RapidOCR * fix(ocr): make OCR text search case- and accent-insensitive using ILIKE + unaccent * fix(ocr): add OCR search fields * fix: Add OCR database migration and update ML prediction logic. * trigrams are already case insensitive * add tests * format * update migrations * wrong uuid function * linting * maybe fix medium tests * formatting * fix weblate check * openapi * sql * minor fixes * maybe fix medium tests part 2 * passing medium tests * format web * readd sql * format dart * disabled in e2e * chore: translation ordering --------- Co-authored-by: mertalev <101130780+mertalev@users.noreply.github.com> Co-authored-by: Alex Tran <alex.tran1502@gmail.com>
71 lines
2.7 KiB
Dart
71 lines
2.7 KiB
Dart
import 'package:immich_mobile/domain/models/asset/base_asset.model.dart' hide AssetVisibility;
|
|
import 'package:immich_mobile/infrastructure/repositories/api.repository.dart';
|
|
import 'package:immich_mobile/models/search/search_filter.model.dart';
|
|
import 'package:openapi/api.dart';
|
|
|
|
class SearchApiRepository extends ApiRepository {
|
|
final SearchApi _api;
|
|
const SearchApiRepository(this._api);
|
|
|
|
Future<SearchResponseDto?> search(SearchFilter filter, int page) {
|
|
AssetTypeEnum? type;
|
|
if (filter.mediaType.index == AssetType.image.index) {
|
|
type = AssetTypeEnum.IMAGE;
|
|
} else if (filter.mediaType.index == AssetType.video.index) {
|
|
type = AssetTypeEnum.VIDEO;
|
|
}
|
|
|
|
if (filter.context != null && filter.context!.isNotEmpty) {
|
|
return _api.searchSmart(
|
|
SmartSearchDto(
|
|
query: filter.context!,
|
|
language: filter.language,
|
|
country: filter.location.country,
|
|
state: filter.location.state,
|
|
city: filter.location.city,
|
|
make: filter.camera.make,
|
|
model: filter.camera.model,
|
|
takenAfter: filter.date.takenAfter,
|
|
takenBefore: filter.date.takenBefore,
|
|
visibility: filter.display.isArchive ? AssetVisibility.archive : AssetVisibility.timeline,
|
|
isFavorite: filter.display.isFavorite ? true : null,
|
|
isNotInAlbum: filter.display.isNotInAlbum ? true : null,
|
|
personIds: filter.people.map((e) => e.id).toList(),
|
|
type: type,
|
|
page: page,
|
|
size: 100,
|
|
),
|
|
);
|
|
}
|
|
|
|
return _api.searchAssets(
|
|
MetadataSearchDto(
|
|
originalFileName: filter.filename != null && filter.filename!.isNotEmpty ? filter.filename : null,
|
|
country: filter.location.country,
|
|
description: filter.description != null && filter.description!.isNotEmpty ? filter.description : null,
|
|
ocr: filter.ocr != null && filter.ocr!.isNotEmpty ? filter.ocr : null,
|
|
state: filter.location.state,
|
|
city: filter.location.city,
|
|
make: filter.camera.make,
|
|
model: filter.camera.model,
|
|
takenAfter: filter.date.takenAfter,
|
|
takenBefore: filter.date.takenBefore,
|
|
visibility: filter.display.isArchive ? AssetVisibility.archive : AssetVisibility.timeline,
|
|
isFavorite: filter.display.isFavorite ? true : null,
|
|
isNotInAlbum: filter.display.isNotInAlbum ? true : null,
|
|
personIds: filter.people.map((e) => e.id).toList(),
|
|
type: type,
|
|
page: page,
|
|
size: 1000,
|
|
),
|
|
);
|
|
}
|
|
|
|
Future<List<String>?> getSearchSuggestions(
|
|
SearchSuggestionType type, {
|
|
String? country,
|
|
String? state,
|
|
String? make,
|
|
String? model,
|
|
}) => _api.getSearchSuggestions(type, country: country, state: state, make: make, model: model);
|
|
}
|