Skip to content

Commit 477407e

Browse files
committed
Use the highest priority as the video priority. VERY_HIGH>HIGH>MEDIUM>LOW>VERY_LOW
1 parent 18e3c68 commit 477407e

1 file changed

Lines changed: 32 additions & 1 deletion

File tree

iped-engine/src/main/java/iped/engine/task/RemoteImageExternalClassifierTask.java

Lines changed: 32 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,36 @@ public boolean hasAnyResult() {
177177
}
178178
}
179179

180+
static boolean shouldReplaceReviewPriority(String currentPriority, String newPriority) {
181+
if (newPriority == null || newPriority.isBlank()) {
182+
return false;
183+
}
184+
if (currentPriority == null || currentPriority.isBlank()) {
185+
return true;
186+
}
187+
return getReviewPriorityRank(newPriority) > getReviewPriorityRank(currentPriority);
188+
}
189+
190+
static int getReviewPriorityRank(String reviewPriority) {
191+
if (reviewPriority == null) {
192+
return Integer.MIN_VALUE;
193+
}
194+
switch (reviewPriority) {
195+
case "VERY_HIGH":
196+
return 5;
197+
case "HIGH":
198+
return 4;
199+
case "MEDIUM":
200+
return 3;
201+
case "LOW":
202+
return 2;
203+
case "VERY_LOW":
204+
return 1;
205+
default:
206+
return Integer.MIN_VALUE;
207+
}
208+
}
209+
180210
/**
181211
* Represents a Zip archive holding files to be sent for classification.
182212
*/
@@ -432,7 +462,8 @@ private void processResult(InputStream responseStream) throws IOException {
432462

433463
// New API payload
434464
JsonNode reviewPriority = item.get("reviewPriority");
435-
if (reviewPriority != null && !reviewPriority.isNull() && !reviewPriority.asText().isBlank()) {
465+
if (reviewPriority != null && !reviewPriority.isNull() && !reviewPriority.asText().isBlank()
466+
&& shouldReplaceReviewPriority(res.getReviewPriority(), reviewPriority.asText())) {
436467
res.setReviewPriority(reviewPriority.asText());
437468
}
438469
JsonNode embeddings = item.get("embeddings");

0 commit comments

Comments
 (0)