Merge pull request #54 from DropThe8bit/feature/story

[refactor] 이미지 생성을 위한 프롬프트를 scene내용 -> 프롬프트 직접 입력으로 변경

Author: hhongyeahh

src/main/java/everTale/everTale_be/domain/story/controller/StoryController.java

@@ -8,12 +8,12 @@
 import everTale.everTale_be.global.apiPayload.ApiResponse;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.Parameter;
-import io.swagger.v3.oas.annotations.media.Content;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.RequiredArgsConstructor;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.web.PageableDefault;
 import org.springframework.http.MediaType;
+import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
@@ -128,18 +128,17 @@ public ApiResponse<String> createNextSceneFromAnswer(
         String nextStory = storyService.generateNextSceneWithAnswer(storyId, pageNum, request.getAnswer());
         return ApiResponse.onSuccess(nextStory);
     }
-    @Operation(summary = "스케치 기반 이미지 생성 API", description = "아이의 스케치 이미지와 줄거리를 기반으로 이미지를 생성합니다.")
+    @Operation(summary = "스케치 기반 이미지 생성 API", description = "아이의 스케치 이미지와 장면 프롬프트를 기반으로 이미지를 생성합니다.")
     @PostMapping(
             value = "/{storyId}/scenes/{pageNum}/controlnet",
             consumes = MediaType.MULTIPART_FORM_DATA_VALUE
     )
     public ApiResponse<String> createImageFromSketch(
             @Parameter(description = "스토리 ID") @PathVariable Long storyId,
             @Parameter(description = "페이지 번호(1~8)") @PathVariable int pageNum,
-            @Parameter(description = "스케치 이미지 파일", content = @Content(mediaType = MediaType.APPLICATION_OCTET_STREAM_VALUE))
-            @RequestPart("sketch") MultipartFile sketch
-    ) {
-        String image = storyService.generateImageFromSketch(storyId, pageNum, sketch);
+            @Validated @ModelAttribute StoryRequestDTO.SketchImageRequestDTO request
+            ) {
+        String image = storyService.generateImageFromSketch(storyId, pageNum, request);
         return ApiResponse.onSuccess(image);
     }
 

src/main/java/everTale/everTale_be/domain/story/dto/StoryRequestDTO.java

@@ -3,6 +3,7 @@
 import everTale.everTale_be.domain.story.entity.enums.Genre;
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.*;
+import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
 
@@ -83,6 +84,16 @@ public static class StoryUpdateRequestDTO {
         @Schema(description = "수정된 줄거리 내용", example = "주인공은 용기를 내어 드래곤에게 다가갔다.")
         private String updatedContent;
     }
+    @Data
+    @AllArgsConstructor
+    public static class SketchImageRequestDTO {
+
+        @Schema(description = "장면 프롬프트", example = "A little girl holding a balloon")
+        private String prompt;
+
+        @Schema(type = "string", format = "binary", description = "스케치 이미지 파일")
+        private MultipartFile sketch;
+    }
 
     @Getter
     @Builder

src/main/java/everTale/everTale_be/domain/story/entity/Scene.java

@@ -16,6 +16,7 @@ public class Scene {
     @GeneratedValue(strategy = GenerationType.IDENTITY)
     private Long id;
 
+    @Column(columnDefinition = "TEXT")
     private String content;
 
     private int page;

src/main/java/everTale/everTale_be/domain/story/service/StoryService.java

@@ -295,14 +295,11 @@ public String generateNextSceneWithAnswer(Long storyId, int pageNum, String answ
         story.addScene(newScene);
         return nextContent;
     }
-    // 줄거리 및 아이그림 기반 그림 생성
+    // 장면 프롬프트 및 아이그림 기반 그림 생성
     @Transactional
-    public String generateImageFromSketch(Long storyId, int pageNum, MultipartFile sketch) {
+    public String generateImageFromSketch(Long storyId, int pageNum, StoryRequestDTO.SketchImageRequestDTO request) {
         Scene scene = findMyScene(storyId, pageNum);
-
-        String prompt = scene.getContent();
-        String imageUrl = storyApiClient.callFastApiForImageFromSketch(sketch, prompt, scene.getStory().getGenre().name());
-
+        String imageUrl = storyApiClient.callFastApiForImageFromSketch(request.getSketch(), request.getPrompt(), scene.getStory().getGenre().name());
         scene.updateImageUrl(imageUrl);
         return imageUrl;
     }