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

'commit'

Browse Source
main
HuangHai 2 months ago
parent dcedea5d1a
commit 28c58fc9f9
3 changed files with 121 additions and 47 deletions
Show Stats Download Patch File Download Diff File

BIN
dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Midjourney/Example/kristina.burlakova_A_cute_cat_playing_with_a_ball_of_yarn_digit.png
View File

Binary file not shown.
Side by Side

After

Width:  |  Height:  |  Size: 6.8 MiB

104
dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Midjourney/Kit/MjCommon.java
Unescape View File

@ -0,0 +1,104 @@
package com.dsideal.aiSupport.Util.Midjourney.Kit;
import cn.hutool.core.io.FileUtil;
import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpResponse;
import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.auth0.jwt.JWT;
import com.auth0.jwt.algorithms.Algorithm;
import com.dsideal.aiSupport.Plugin.YamlProp;
import com.dsideal.aiSupport.Util.KeLing.Kit.KlErrorCode;
import com.jfinal.kit.Prop;
import lombok.SneakyThrows;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import static com.dsideal.aiSupport.AiSupportApplication.getEnvPrefix;
public class MjCommon {
private static final Logger log = LoggerFactory.getLogger(MjCommon.class);
// 获取项目根目录路径
protected static String projectRoot = System.getProperty("user.dir").replace("\\","/")+"/dsAiSupport";
// 拼接相对路径
protected static String basePath = projectRoot + "/src/main/java/com/dsideal/aiSupport/Util/Midjourney/Example/";
protected static String ak; // 填写access key
public static Prop PropKit; // 配置文件工具
protected static final String BASE_URL = "https://goapi.gptnb.ai";
static {
//加载配置文件
String configFile = "application_{?}.yaml".replace("{?}", getEnvPrefix());
PropKit = new YamlProp(configFile);
ak = PropKit.get("GPTNB.sk");
}
/**
* 从URL下载文件到指定路径
*
* @param fileUrl 文件URL
* @param saveFilePath 保存路径
* @throws Exception 下载过程中的异常
*/
public static void downloadFile(String fileUrl, String saveFilePath) throws Exception {
try {
// 使用Hutool下载文件
long fileSize = HttpUtil.downloadFile(fileUrl, FileUtil.file(saveFilePath));
log.info("文件下载成功，保存路径: {}, 文件大小: {}字节", saveFilePath, fileSize);
} catch (Exception e) {
log.error("文件下载失败: {}", e.getMessage(), e);
throw new Exception("文件下载失败: " + e.getMessage(), e);
}
}
/**
* 查询任务状态
*
* @param taskId 任务ID
* @return 任务结果
* @throws Exception 异常信息
*/
@SneakyThrows
public static JSONObject queryTaskStatus(String taskId) {
// 创建OkHttpClient
OkHttpClient client = new OkHttpClient().newBuilder()
.connectTimeout(30, TimeUnit.SECONDS)
.readTimeout(30, TimeUnit.SECONDS)
.build();
// 创建请求
Request request = new Request.Builder()
.url(BASE_URL + "/mj/task/" + taskId + "/fetch")
.method("GET", null)
.addHeader("Authorization", "Bearer " + ak)
.build();
// 发送请求并获取响应
log.info("查询Midjourney任务状态: {}", taskId);
Response response = client.newCall(request).execute();
// 检查响应状态
if (!response.isSuccessful()) {
String errorMsg = "Midjourney API请求失败，状态码: " + response.code();
log.error(errorMsg);
throw new Exception(errorMsg);
}
// 解析响应
String responseBody = response.body().string();
log.info("查询Midjourney任务状态响应: {}", responseBody);
return JSON.parseObject(responseBody);
}
}

64
dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Midjourney/Midjourney.java → dsAiSupport/src/main/java/com/dsideal/aiSupport/Util/Midjourney/Txt2Img.java
Unescape View File

@ -1,5 +1,6 @@
package com.dsideal.aiSupport.Util.Midjourney;
import com.dsideal.aiSupport.Util.Midjourney.Kit.MjCommon;
import lombok.SneakyThrows;
import okhttp3.*;
import com.alibaba.fastjson.JSON;
@ -14,10 +15,8 @@ import java.util.concurrent.TimeUnit;
* Midjourney API 工具类
* 用于调用 Midjourney 的 imagine 接口生成图片
*/
public class Midjourney {
private static final Logger log = LoggerFactory.getLogger(Midjourney.class);
private static final String BASE_URL = "https://goapi.gptnb.ai";
private static final String API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662";
public class Txt2Img extends MjCommon {
private static final Logger log = LoggerFactory.getLogger(Txt2Img.class);
/**
* 提交 imagine 请求
@ -73,7 +72,7 @@ public class Midjourney {
.url(BASE_URL + "/mj/submit/imagine")
.method("POST", body)
.addHeader("Content-Type", "application/json")
.addHeader("Authorization", "Bearer " + API_KEY)
.addHeader("Authorization", "Bearer " + ak)
.build();
// 发送请求并获取响应
@ -109,46 +108,6 @@ public class Midjourney {
return taskId;
}
/**
* 查询任务状态
*
* @param taskId 任务ID
* @return 任务结果
* @throws Exception 异常信息
*/
@SneakyThrows
public static JSONObject queryTaskStatus(String taskId) {
// 创建OkHttpClient
OkHttpClient client = new OkHttpClient().newBuilder()
.connectTimeout(30, TimeUnit.SECONDS)
.readTimeout(30, TimeUnit.SECONDS)
.build();
// 创建请求
Request request = new Request.Builder()
.url(BASE_URL + "/mj/task/" + taskId + "/fetch")
.method("GET", null)
.addHeader("Authorization", "Bearer " + API_KEY)
.build();
// 发送请求并获取响应
log.info("查询Midjourney任务状态: {}", taskId);
Response response = client.newCall(request).execute();
// 检查响应状态
if (!response.isSuccessful()) {
String errorMsg = "Midjourney API请求失败，状态码: " + response.code();
log.error(errorMsg);
throw new Exception(errorMsg);
}
// 解析响应
String responseBody = response.body().string();
log.info("查询Midjourney任务状态响应: {}", responseBody);
return JSON.parseObject(responseBody);
}
@SneakyThrows
public static void main(String[] args) {
// 提示词
@ -161,6 +120,7 @@ public class Midjourney {
int maxRetries = 1000;
int retryCount = 0;
int retryInterval = 5000; // 5秒
String imageUrl = null;
while (retryCount < maxRetries) {
JSONObject result = queryTaskStatus(taskId);
@ -177,7 +137,7 @@ public class Midjourney {
if (status != null && !status.isEmpty()) {
if ("SUCCESS".equals(status)) {
// 任务成功，获取图片URL
String imageUrl = result.getString("imageUrl");
imageUrl = result.getString("imageUrl");
log.info("生成的图片URL: {}", imageUrl);
break;
} else if ("FAILED".equals(status)) {
@ -190,7 +150,7 @@ public class Midjourney {
String description = result.getString("description");
if (description != null && description.contains("成功") && !"0%".equals(progress)) {
// 如果描述包含"成功"且进度不为0%，可能任务已完成
String imageUrl = result.getString("imageUrl");
imageUrl = result.getString("imageUrl");
if (imageUrl != null && !imageUrl.isEmpty()) {
log.info("生成的图片URL: {}", imageUrl);
break;
@ -207,5 +167,15 @@ public class Midjourney {
if (retryCount >= maxRetries) {
log.error("查询任务状态超时，已达到最大重试次数: {}", maxRetries);
}
// 如果获取到了图片URL，则下载保存
if (imageUrl != null && !imageUrl.isEmpty()) {
// 生成文件名（使用时间戳和任务ID）
String fileName = "mj_" + System.currentTimeMillis() + "_" + taskId + ".png";
// 完整保存路径
String savePath = basePath + "/" + fileName;
// 下载图片
downloadFile(imageUrl, savePath);
}
}
}
Write Preview
Loading…
Cancel
Save