#75 Refactor: NodeRequestDTO에 seed 추가

Author: jeongyeon0208

src/main/java/com/example/dto/request/NodeRequestDTO.java

@@ -16,4 +16,6 @@ public class NodeRequestDTO {
     private Long parentNodeId;
 
     private Long projectId;
+
+    private Boolean seed;
 }

src/main/java/com/example/service/ImageService.java

@@ -10,7 +10,7 @@ public interface ImageService {
 
     public String generateTextToImage(String prompt);
 
-    public String generateImageToImage(String prompt, String imageURL);
+    public String generateImageToImage(String prompt, String imageURL, boolean seed);
 
     public void deleteS3Image(String imageURL);
 

src/main/java/com/example/service/ImageServiceImpl.java

@@ -54,14 +54,14 @@ public String generateTextToImage(String prompt) {
     }
 
     @Override
-    public String generateImageToImage(String prompt, String imageURL) {
+    public String generateImageToImage(String prompt, String imageURL, boolean seed) {
 
         HttpHeaders headers = new HttpHeaders();
         headers.setContentType(MediaType.APPLICATION_JSON);
 
         String url = "https://1dc6-124-55-57-87.ngrok-free.app//modify-image";  // FastAPI 서버의 URL
 
-        String requestBody = "{\"prompt\": \"" + prompt + "\", \"imageURL\": " + imageURL + "}";
+        String requestBody = "{\"prompt\": \"" + prompt + "\", \"imageURL\": " + imageURL  + "\", \"seed\": " + seed + "}";
 
         HttpEntity<String> requestEntity = new HttpEntity<>(requestBody, headers);
 

src/main/java/com/example/service/NodeServiceImpl.java

@@ -42,7 +42,7 @@ public NodeResponseDTO saveNode(NodeRequestDTO nodeRequestDto, User user, Projec
         Long parentNodeId = nodeRequestDto.getParentNodeId();
         if (parentNodeId != null) {
             parentNode = findNodeById(nodeRequestDto.getParentNodeId());
-            imageURL = imageService.generateImageToImage(nodeRequestDto.getPrompt(), parentNode.getImageURL());
+            imageURL = imageService.generateImageToImage(nodeRequestDto.getPrompt(), parentNode.getImageURL(), nodeRequestDto.getSeed());
 
             // 로컬에서 위에 generateImage 없이 돌릴때 사용할 용도
             //imageURL = "https://hongik-s3.s3.amazonaws.com/42_10_7.5.png";