diff --git a/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageEnhancer.java b/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageEnhancer.java index cc59d82..9990d02 100644 --- a/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageEnhancer.java +++ b/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageEnhancer.java @@ -33,7 +33,7 @@ public class BceImageEnhancer implements IEnhancer { public MultipartFile enhance(String url) { AipImageProcess client = getClient(); HashMap options = new HashMap<>(); - JSONObject jsonObject = client.imageQualityEnhance(url, options); + JSONObject jsonObject = client.imageDefinitionEnhance(url, options); return ImageUtils.base64ToMultipartFile(jsonObject.getString("image")); } diff --git a/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageSR.java b/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageSR.java new file mode 100644 index 0000000..db3d305 --- /dev/null +++ b/src/main/java/com/ycwl/basic/image/enhancer/adapter/BceImageSR.java @@ -0,0 +1,43 @@ +package com.ycwl.basic.image.enhancer.adapter; + +import com.baidu.aip.imageprocess.AipImageProcess; +import com.ycwl.basic.image.enhancer.entity.BceEnhancerConfig; +import com.ycwl.basic.utils.ImageUtils; +import org.json.JSONObject; +import org.springframework.web.multipart.MultipartFile; + +import java.util.HashMap; +import java.util.Map; + +public class BceImageSR implements IEnhancer { + private BceEnhancerConfig config; + + public boolean setConfig(BceEnhancerConfig config) { + this.config = config; + return true; + } + + @Override + public boolean loadConfig(Map _config) { + BceEnhancerConfig config = new BceEnhancerConfig(); + config.setAppId(_config.get("appId")); + config.setApiKey(_config.get("apiKey")); + config.setSecretKey(_config.get("secretKey")); + config.setQps(Float.parseFloat(_config.get("qps"))); + this.config = config; + return true; + } + + @Override + public MultipartFile enhance(String url) { + AipImageProcess client = getClient(); + HashMap options = new HashMap<>(); + JSONObject jsonObject = client.imageQualityEnhance(url, options); + return ImageUtils.base64ToMultipartFile(jsonObject.getString("image")); + } + + public AipImageProcess getClient() { + AipImageProcess client = new AipImageProcess(config.getAppId(), config.getApiKey(), config.getSecretKey()); + return client; + } +}