From cfc5db45cd0c8d6403239bc4511af102adf5045e Mon Sep 17 00:00:00 2001
From: yhampe <yannik.hampe@knecon.com>
Date: Wed, 14 Feb 2024 12:24:32 +0100
Subject: [PATCH] RED-8481: Use visual layout parsing to detect signatures

fixed failing tests because of null pointer
---
 .../adapter/VisualLayoutParsingAdapter.java         | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/python_api/adapter/VisualLayoutParsingAdapter.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/python_api/adapter/VisualLayoutParsingAdapter.java
index 3501b36..3eca613 100644
--- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/python_api/adapter/VisualLayoutParsingAdapter.java
+++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/python_api/adapter/VisualLayoutParsingAdapter.java
@@ -33,13 +33,16 @@ public class VisualLayoutParsingAdapter {
     }
 
     public Map<Integer, List<ClassifiedImage>> buildExtractedSignaturesPerPage(VisualLayoutParsingResponse visualLayoutParsingResponse) {
+        if(visualLayoutParsingResponse != null) {
 
-        Map<Integer, List<ClassifiedImage>> tableCells = new HashMap<>();
-        visualLayoutParsingResponse.getData()
-                .forEach(tableData -> tableCells.computeIfAbsent(tableData.getPage_idx(), tableCell -> new ArrayList<>()).addAll(convertSignatures(tableData.getPage_idx(),tableData.getBoxes())));
-
-        return tableCells;
+            Map<Integer, List<ClassifiedImage>> signatures = new HashMap<>();
+            visualLayoutParsingResponse.getData()
+                    .forEach(tableData -> signatures.computeIfAbsent(tableData.getPage_idx(), tableCell -> new ArrayList<>())
+                            .addAll(convertSignatures(tableData.getPage_idx(), tableData.getBoxes())));
 
+            return signatures;
+        }
+        return null;
     }