diff --git a/layoutparser-service/layoutparser-service-internal-api/src/main/java/com/knecon/fforesight/service/layoutparser/internal/api/data/redaction/SimplifiedSectionText.java b/layoutparser-service/layoutparser-service-internal-api/src/main/java/com/knecon/fforesight/service/layoutparser/internal/api/data/redaction/SimplifiedSectionText.java index f8cc10b..9707325 100644 --- a/layoutparser-service/layoutparser-service-internal-api/src/main/java/com/knecon/fforesight/service/layoutparser/internal/api/data/redaction/SimplifiedSectionText.java +++ b/layoutparser-service/layoutparser-service-internal-api/src/main/java/com/knecon/fforesight/service/layoutparser/internal/api/data/redaction/SimplifiedSectionText.java @@ -14,7 +14,7 @@ import lombok.NoArgsConstructor; public class SimplifiedSectionText { @Schema(description = "The number of this Section. This is used to map the simplified section text back to the original Section.") - private int sectionNumber; + private String sectionNumber; @Schema(description = "The text in this Section.") private String text; diff --git a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/SimplifiedSectionTextService.java b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/SimplifiedSectionTextService.java index ecde09b..3a06902 100644 --- a/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/SimplifiedSectionTextService.java +++ b/layoutparser-service/layoutparser-service-processor/src/main/java/com/knecon/fforesight/service/layoutparser/processor/services/SimplifiedSectionTextService.java @@ -42,7 +42,9 @@ public class SimplifiedSectionTextService { return SimplifiedSectionText.builder() .sectionNumber(section.getTreeId() - .get(0)) + .stream() + .map(String::valueOf) + .collect(Collectors.joining("."))) .text(section.getTextBlock().getSearchText()) .build(); }