diff --git a/apps/red-ui/src/app/modules/file-preview/services/suggestions.service.ts b/apps/red-ui/src/app/modules/file-preview/services/suggestions.service.ts index 0ac907e01..eceb4b20a 100644 --- a/apps/red-ui/src/app/modules/file-preview/services/suggestions.service.ts +++ b/apps/red-ui/src/app/modules/file-preview/services/suggestions.service.ts @@ -26,13 +26,17 @@ export class SuggestionsService { hideSuggestionsInPreview(annotations: Annotation[]): void { if (this._readableRedactionsService.active) { if (this._userPreferenceService.getDisplaySuggestionsInPreview()) { - const suggestionsRemove = annotations.filter(a => bool(a.getCustomData('suggestionRemove'))); + const suggestionsRemove = annotations.filter( + a => bool(a.getCustomData('suggestionRemove')) || bool(a.getCustomData('suggestionRecategorizeImage')), + ); this._annotationManager.hide(suggestionsRemove); return; } } const suggestionsToHide = annotations.filter( - a => bool(a.getCustomData('suggestionAdd')) && !bool(a.getCustomData('suggestionAddToFalsePositive')), + a => + (bool(a.getCustomData('suggestionAdd')) && !bool(a.getCustomData('suggestionAddToFalsePositive'))) || + bool(a.getCustomData('suggestionRecategorizeImage')), ); annotations.forEach(a => { if (bool(a.getCustomData('suggestionRemove'))) { @@ -52,7 +56,7 @@ export class SuggestionsService { } } - annotations = annotations.filter(a => !a.isSuggestionAdd || a.isSuggestionAddToFalsePositive); + annotations = annotations.filter(a => (!a.isSuggestionAdd || a.isSuggestionAddToFalsePositive) && !a.isSuggestionRecategorizeImage); for (let i = annotations.length - 1; i >= 0; i--) { const foundRemovedRedaction = this.#removedRedactions.find(r => r.id === annotations[i].id); if (foundRemovedRedaction) { diff --git a/apps/red-ui/src/app/modules/pdf-viewer/services/annotation-draw.service.ts b/apps/red-ui/src/app/modules/pdf-viewer/services/annotation-draw.service.ts index 2d642d896..7211bbf13 100644 --- a/apps/red-ui/src/app/modules/pdf-viewer/services/annotation-draw.service.ts +++ b/apps/red-ui/src/app/modules/pdf-viewer/services/annotation-draw.service.ts @@ -159,6 +159,7 @@ export class AnnotationDrawService { annotation.setCustomData('suggestionAdd', String(annotationWrapper.isSuggestionAdd)); annotation.setCustomData('suggestionAddToFalsePositive', String(annotationWrapper.isSuggestionAddToFalsePositive)); annotation.setCustomData('suggestionRemove', String(annotationWrapper.isSuggestionRemove)); + annotation.setCustomData('suggestionRecategorizeImage', String(annotationWrapper.isSuggestionRecategorizeImage)); annotation.setCustomData('skipped', String(annotationWrapper.isSkipped)); annotation.setCustomData('changeLog', String(annotationWrapper.isChangeLogEntry)); annotation.setCustomData('changeLogRemoved', String(annotationWrapper.isChangeLogRemoved)); diff --git a/apps/red-ui/src/app/modules/pdf-viewer/services/readable-redactions.service.ts b/apps/red-ui/src/app/modules/pdf-viewer/services/readable-redactions.service.ts index 6b453ee04..c9417bee2 100644 --- a/apps/red-ui/src/app/modules/pdf-viewer/services/readable-redactions.service.ts +++ b/apps/red-ui/src/app/modules/pdf-viewer/services/readable-redactions.service.ts @@ -79,7 +79,8 @@ export class ReadableRedactionsService { setAnnotationsOpacity(annotations: Annotation[], restoreToOriginal = false) { annotations.forEach(annotation => { - annotation['Opacity'] = restoreToOriginal ? parseFloat(annotation.getCustomData('opacity')) : 0.5; + const isSuggestion = annotation.getCustomData('suggestion'); + annotation['Opacity'] = restoreToOriginal || isSuggestion ? parseFloat(annotation.getCustomData('opacity')) : 0.5; }); }