huanghai
/
dsProject
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

'commit'

Browse Source
main
HuangHai 2 months ago
parent 1fc1588b12
commit b2af724e01
2 changed files with 48 additions and 118 deletions
Show Stats Download Patch File Download Diff File

31
dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Liblib/Kit/SignUtil.java → dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Liblib/Kit/LibLibUtil.java
Unescape View File

@ -2,6 +2,7 @@ package com.dsideal.aiSupport.Util.Liblib.Kit;
import java.security.InvalidKeyException; import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException; import java.security.NoSuchAlgorithmException;
import java.util.concurrent.TimeUnit;
import javax.crypto.Mac; import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec; import javax.crypto.spec.SecretKeySpec;
@ -9,24 +10,26 @@ import javax.crypto.spec.SecretKeySpec;
import com.dsideal.aiSupport.Plugin.YamlProp; import com.dsideal.aiSupport.Plugin.YamlProp;
import com.jfinal.kit.Prop; import com.jfinal.kit.Prop;
import lombok.Getter; import lombok.Getter;
import okhttp3.OkHttpClient;
import org.apache.commons.codec.binary.Base64; import org.apache.commons.codec.binary.Base64;
import org.apache.commons.lang3.RandomStringUtils; import org.apache.commons.lang3.RandomStringUtils;
import static com.dsideal.aiSupport.AiSupportApplication.getEnvPrefix; import static com.dsideal.aiSupport.AiSupportApplication.getEnvPrefix;
public class SignUtil { public class LibLibUtil {
// API基础URL
protected static final String API_BASE_URL = "https://openapi.liblibai.cloud";
// API访问凭证 // API访问凭证
protected static final String ak; // Access Key protected static final String accessKey; // Access Key
protected static final String sk; // Secret Key protected static final String secretKey; // Secret Key
public static Prop PropKit; // 配置文件工具 public static Prop PropKit; // 配置文件工具
static { static {
//加载配置文件 //加载配置文件
String configFile = "application_{?}.yaml".replace("{?}", getEnvPrefix()); String configFile = "application_{?}.yaml".replace("{?}", getEnvPrefix());
PropKit = new YamlProp(configFile); PropKit = new YamlProp(configFile);
ak = PropKit.get("LIBLIB.accessKey"); // 从配置文件获取Access Key accessKey = PropKit.get("LIBLIB.accessKey"); // 从配置文件获取Access Key
sk = PropKit.get("LIBLIB.secretKey"); // 从配置文件获取Secret Key secretKey = PropKit.get("LIBLIB.secretKey"); // 从配置文件获取Secret Key
} }
/** /**
@ -45,13 +48,23 @@ public class SignUtil {
} }
} }
/**
* 创建OkHttpClient实例
* @return OkHttpClient实例
*/
protected static OkHttpClient createHttpClient() {
return new OkHttpClient.Builder()
.connectTimeout(30, TimeUnit.SECONDS)
.readTimeout(60, TimeUnit.SECONDS)
.writeTimeout(60, TimeUnit.SECONDS)
.build();
}
/** /**
* 生成完整的签名信息，包括签名、时间戳和随机字符串 * 生成完整的签名信息，包括签名、时间戳和随机字符串
* @param uri API接口的uri地址 * @param uri API接口的uri地址
* @return 签名信息对象 * @return 签名信息对象
*/ */
public static SignatureInfo makeSignInfo(String uri) { public static SignatureInfo sign(String uri) {
// 当前毫秒时间戳 // 当前毫秒时间戳
long timestamp = System.currentTimeMillis(); long timestamp = System.currentTimeMillis();
// 随机字符串 // 随机字符串
@ -60,7 +73,7 @@ public class SignUtil {
String content = uri + "&" + timestamp + "&" + signatureNonce; String content = uri + "&" + timestamp + "&" + signatureNonce;
try { try {
// 生成签名 // 生成签名
SecretKeySpec secret = new SecretKeySpec(sk.getBytes(), "HmacSHA1"); SecretKeySpec secret = new SecretKeySpec(secretKey.getBytes(), "HmacSHA1");
Mac mac = Mac.getInstance("HmacSHA1"); Mac mac = Mac.getInstance("HmacSHA1");
mac.init(secret); mac.init(secret);
String signature = Base64.encodeBase64URLSafeString(mac.doFinal(content.getBytes())); String signature = Base64.encodeBase64URLSafeString(mac.doFinal(content.getBytes()));

99
dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Liblib/ImageToImageUltra.java → dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Liblib/UltraImg2Img.java
Unescape View File

@ -4,7 +4,7 @@ import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject; import com.alibaba.fastjson.JSONObject;
import com.dsideal.aiSupport.Plugin.YamlProp; import com.dsideal.aiSupport.Plugin.YamlProp;
import com.dsideal.aiSupport.Util.Liblib.Kit.SignUtil; import com.dsideal.aiSupport.Util.Liblib.Kit.LibLibUtil;
import com.jfinal.kit.Prop; import com.jfinal.kit.Prop;
import okhttp3.*; import okhttp3.*;
import org.slf4j.Logger; import org.slf4j.Logger;
@ -15,45 +15,16 @@ import java.util.concurrent.TimeUnit;
import static com.dsideal.aiSupport.AiSupportApplication.getEnvPrefix; import static com.dsideal.aiSupport.AiSupportApplication.getEnvPrefix;
/** public class UltraImg2Img extends LibLibUtil {
* LibLib 图生图API工具类 //日志
* 使用ControlNet技术将一个源图像作为参考来生成新图像 private static final Logger log = LoggerFactory.getLogger(UltraImg2Img.class);
*/
public class ImageToImageUltra {
private static final Logger log = LoggerFactory.getLogger(ImageToImageUltra.class);
// API基础URL
private static final String API_BASE_URL = "https://openapi.liblibai.cloud";
// 图生图API路径 // 图生图API路径
private static final String IMG_TO_IMG_PATH = "/api/generate/webui/img2img/ultra"; private static final String IMG_TO_IMG_PATH = "/api/generate/webui/img2img/ultra";
// API访问凭证
protected static String accessKey;
protected static String secretKey;
public static Prop PropKit;
static {
//加载配置文件
String configFile = "application_{?}.yaml".replace("{?}", getEnvPrefix());
PropKit = new YamlProp(configFile);
accessKey = PropKit.get("LIBLIB.accessKey");
secretKey = PropKit.get("LIBLIB.secretKey");
}
/**
* 创建OkHttpClient实例
* @return OkHttpClient实例
*/
private static OkHttpClient createHttpClient() {
return new OkHttpClient.Builder()
.connectTimeout(30, TimeUnit.SECONDS)
.readTimeout(60, TimeUnit.SECONDS)
.writeTimeout(60, TimeUnit.SECONDS)
.build();
}
/** /**
* 提交图生图任务 * 提交图生图任务
*
* @param templateUuid 模板UUID * @param templateUuid 模板UUID
* @param prompt 提示词 * @param prompt 提示词
* @param sourceImageUrl 源图片URL * @param sourceImageUrl 源图片URL
@ -68,26 +39,15 @@ public class ImageToImageUltra {
* @throws IOException 异常信息 * @throws IOException 异常信息
*/ */
public static String submitImageToImageTask( public static String submitImageToImageTask(
String templateUuid, String templateUuid, String prompt, String sourceImageUrl, int width,
String prompt, int height, int steps, double cfgScale, long seed, String controlNetModel,
String sourceImageUrl,
int width,
int height,
int steps,
double cfgScale,
long seed,
String controlNetModel,
double controlWeight) throws IOException { double controlWeight) throws IOException {
// 创建OkHttpClient // 创建OkHttpClient
OkHttpClient client = createHttpClient(); OkHttpClient client = createHttpClient();
// 构建请求体 // 构建请求体
JSONObject requestBody = new JSONObject(); JSONObject requestBody = new JSONObject();
// 添加模板UUID // 添加模板UUID
requestBody.put("templateUuid", templateUuid); requestBody.put("templateUuid", templateUuid);
// 构建生成参数 // 构建生成参数
JSONObject generateParams = new JSONObject(); JSONObject generateParams = new JSONObject();
generateParams.put("prompt", prompt); generateParams.put("prompt", prompt);
@ -102,59 +62,18 @@ public class ImageToImageUltra {
generateParams.put("steps", steps); generateParams.put("steps", steps);
generateParams.put("restoreFaces", 0); generateParams.put("restoreFaces", 0);
generateParams.put("sourceImage", sourceImageUrl); generateParams.put("sourceImage", sourceImageUrl);
// 构建ControlNet参数
JSONArray controlNetArray = new JSONArray();
JSONObject controlNet = new JSONObject();
controlNet.put("unitOrder", 0);
/*
1. controlType：
1. line：线稿轮廓
2. depth：空间关系
3. pose：人物姿态
4. IPAdapter：风格迁移
2. controlImage：参考图可公网访问的完整URL
*/
controlNet.put("controlImage", sourceImageUrl);
controlNet.put("width", width);
controlNet.put("height", height);
controlNet.put("preprocessor", 68);
// 构建注释参数
JSONObject annotationParameters = new JSONObject();
annotationParameters.put("entityControl", new JSONObject());
controlNet.put("annotationParameters", annotationParameters);
// 设置ControlNet模型和权重
controlNet.put("model", controlNetModel);
controlNet.put("controlWeight", controlWeight);
controlNet.put("startingControlStep", 0);
controlNet.put("endingControlStep", 1);
controlNet.put("pixelPerfect", 1);
controlNet.put("controlMode", 0);
controlNet.put("resizeMode", 1);
// 添加ControlNet到数组
// 暂不加载ControlNet
//controlNetArray.add(controlNet);
//generateParams.put("controlNet", controlNetArray);
// 将生成参数添加到请求体 // 将生成参数添加到请求体
requestBody.put("generateParams", generateParams); requestBody.put("generateParams", generateParams);
// 获取API路径 // 获取API路径
String uri = IMG_TO_IMG_PATH; String uri = IMG_TO_IMG_PATH;
// 生成签名信息 // 生成签名信息
SignUtil.SignatureInfo signInfo = SignUtil.makeSignInfo(uri); LibLibUtil.SignatureInfo signInfo = LibLibUtil.sign(uri);
// 构建带签名的URL // 构建带签名的URL
HttpUrl.Builder urlBuilder = HttpUrl.parse(API_BASE_URL + uri).newBuilder() HttpUrl.Builder urlBuilder = HttpUrl.parse(API_BASE_URL + uri).newBuilder()
.addQueryParameter("AccessKey", accessKey) .addQueryParameter("AccessKey", accessKey)
.addQueryParameter("Signature", signInfo.getSignature()) .addQueryParameter("Signature", signInfo.getSignature())
.addQueryParameter("Timestamp", String.valueOf(signInfo.getTimestamp())) .addQueryParameter("Timestamp", String.valueOf(signInfo.getTimestamp()))
.addQueryParameter("SignatureNonce", signInfo.getSignatureNonce()); .addQueryParameter("SignatureNonce", signInfo.getSignatureNonce());
// 创建请求 // 创建请求
MediaType mediaType = MediaType.parse("application/json"); MediaType mediaType = MediaType.parse("application/json");
RequestBody body = RequestBody.create(mediaType, requestBody.toJSONString()); RequestBody body = RequestBody.create(mediaType, requestBody.toJSONString());
@ -188,11 +107,9 @@ public class ImageToImageUltra {
log.error(errorMsg); log.error(errorMsg);
throw new IOException(errorMsg); throw new IOException(errorMsg);
} }
// 获取生成任务UUID // 获取生成任务UUID
String generateUuid = responseJson.getJSONObject("data").getString("generateUuid"); String generateUuid = responseJson.getJSONObject("data").getString("generateUuid");
log.info("图生图任务已提交，任务UUID: {}", generateUuid); log.info("图生图任务已提交，任务UUID: {}", generateUuid);
return generateUuid; return generateUuid;
} }
Write Preview
Loading…
Cancel
Save