修改 imageToSketch

src/main/java/com/ai/da/model/dto/ImageToSketchDTO.java

@@ -0,0 +1,22 @@
+package com.ai.da.model.dto;
+
+import io.swagger.annotations.ApiModel;
+import io.swagger.annotations.ApiModelProperty;
+import lombok.Data;
+
+@ApiModel("image to sketch")
+@Data
+public class ImageToSketchDTO {
+
+    @ApiModelProperty("上传图片的collection_element_id")
+    private Long elementId;
+
+    @ApiModelProperty("sketch线条风格 1->粗， 2->中， 3->细，传数字，自定义风格时，传空 ")
+    private String style;
+
+    @ApiModelProperty("自定义线条风格，上传图片的collection_element_id")
+    private Long styleImageId;
+
+    @ApiModelProperty("性别")
+    private String gender;
+}

src/main/java/com/ai/da/model/enums/SketchStyle.java

@@ -0,0 +1,20 @@
+package com.ai.da.model.enums;
+
+public enum SketchStyle implements IEnumDisplay{
+
+    THICK("Thick"),
+
+    MEDIUM("Medium"),
+
+    THIN("Thin");
+
+    private String value;
+
+    SketchStyle(String value) {
+        this.value = value;
+    }
+    @Override
+    public String getValue() {
+        return value;
+    }
+}

src/main/java/com/ai/da/model/vo/GenerateResultVO.java

@@ -28,9 +28,10 @@ public class GenerateResultVO {
         this.status = status;
     }
 
-    public GenerateResultVO(Long id, String url, String status) {
+    public GenerateResultVO(Long id, String url, String status, String category) {
         this.id = id;
         this.url = url;
         this.status = status;
+        this.category = category;
     }
 }