diff --git a/bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java b/bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java new file mode 100644 index 0000000..db318ba --- /dev/null +++ b/bnyer-common/bnyer-common-core/src/main/java/com/bnyer/common/core/dto/TextToImgLoveDto.java @@ -0,0 +1,37 @@ +package com.bnyer.common.core.dto; + +import io.swagger.annotations.ApiModel; +import io.swagger.annotations.ApiModelProperty; +import lombok.Getter; +import lombok.Setter; + +import java.io.Serializable; + + +@Getter +@Setter +@ApiModel("情侣头像接收类") +public class TextToImgLoveDto implements Serializable { + + + @ApiModelProperty(value="内容提示词") + private String prompt; + + @ApiModelProperty(value="场景提示词") + private String scenePrompt; + + @ApiModelProperty(value="风格名称") + private String styleName; + + @ApiModelProperty(value="绘图者id") + private Long painterId; + + @ApiModelProperty(value="绘图者昵称") + private String painterName; + + @ApiModelProperty(value="应用") + private String appType; + + @ApiModelProperty(value="平台") + private String platform; +} diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java index d8d9025..85e6a6a 100644 --- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java +++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/StableDiffusionService.java @@ -1,6 +1,7 @@ package com.bnyer.img.service; import com.bnyer.common.core.dto.TextToImgDto; +import com.bnyer.common.core.dto.TextToImgLoveDto; import com.bnyer.common.core.vo.TextToImgVo; public interface StableDiffusionService { @@ -13,6 +14,13 @@ public interface StableDiffusionService { */ TextToImgVo textToImg(TextToImgDto param); + /** + * 情侣头像 + * @param param + * @return + */ + TextToImgVo textToImgLove(TextToImgLoveDto param); + void imgToImg(); /** diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java index b4c2383..0913795 100644 --- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java +++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java @@ -6,6 +6,7 @@ import com.alibaba.fastjson.JSONObject; import com.bnyer.common.core.constant.RedisKeyConstant; import com.bnyer.common.core.domain.AiPaint; import com.bnyer.common.core.dto.TextToImgDto; +import com.bnyer.common.core.dto.TextToImgLoveDto; import com.bnyer.common.core.exception.ServiceException; import com.bnyer.common.core.utils.StringUtils; import com.bnyer.common.core.utils.TranslateUtils; @@ -89,7 +90,6 @@ public class StableDiffusionServiceImpl implements StableDiffusionService { @Override public TextToImgVo textToImg(TextToImgDto param) { - //配置控制采用sd还是tx文生图 int button = redisService.getCacheObject(RedisKeyConstant.AI_PAINT_BUTTON); if(button == AiPaintButtonEnum.TX.getCode()){ @@ -286,6 +286,11 @@ public class StableDiffusionServiceImpl implements StableDiffusionService { return null; } + @Override + public TextToImgVo textToImgLove(TextToImgLoveDto param) { + return null; + } + /** * 写入ai绘画次数 * @param appType 应用