⚡️ OCR 파일 템플릿 확장 (#172)

Author: Juye0nLee

src/main/java/com/going/server/domain/ocr/PdfOcrService.java

@@ -20,17 +20,22 @@ public Map<String, String> parse(String response) {
 
             //첫번째 파일에서 데이터 추출
             JsonNode file = root.path("images").get(0);
-            if(file == null || file.isEmpty()) {
-                throw new IllegalArgumentException("OCR 응답에서 파일 데이터가 없습니다.");
+            if (file == null || file.isEmpty()) {
+                throw new IllegalArgumentException("OCR 응답에서 이미지 데이터가 없습니다.");
             }
 
-            //"fields" 배열에서 각 필드 추출
-            JsonNode fields = file.path("fields");
-            for (JsonNode field : fields) {
-                String name = field.path("name").asText();
-                String value = field.path("inferText").asText();
+            JsonNode titleNode = file.path("title");
+            if (!titleNode.isMissingNode()) {
+                String name = titleNode.path("name").asText();
+                String value = titleNode.path("inferText").asText();
                 result.put(name, value);
             }
+//            JsonNode fields = file.path("fields");
+//            for (JsonNode field : fields) {
+//                String name = field.path("name").asText();
+//                String value = field.path("inferText").asText();
+//                result.put(name, value);
+//            }
         }catch (Exception e){
             log.error("OCR 응답 파싱 중 오류 발생",e);
             throw new RuntimeException("OCR 응답 파싱 실패",e);

src/main/java/com/going/server/domain/upload/service/UploadServiceImpl.java

@@ -55,9 +55,10 @@ public class UploadServiceImpl implements UploadService {
     public UploadResponseDto uploadFile(UploadRequestDto dto) {
         try {
             String jsonResponse = ocrService.processOcr(dto.getFile(), apiUrl, secretKey);
+            System.out.println(jsonResponse);
             Map<String, String> paresData = pdfOcrService.parse(jsonResponse);
-            String text = paresData.get("6학년 읽기자료 내용");
-            //System.out.println("추출된 텍스트: " + text);
+            String text = paresData.get("읽기자료");
+            System.out.println("추출된 텍스트: " + text);
 
             //모델에 돌린 값을 받아옴
             String response = setModelData(text);