package com.baidu.ai.aip;
import com.baidu.ai.aip.utils.Base64Util;
import com.baidu.ai.aip.utils.FileUtil;
import com.baidu.ai.aip.utils.HttpUtil;
import java.io.FileOutputStream;
import java.net.URLEncoder;
import java.util.Base64;
public class Image_definition_enhance {
public static String image_definition_enhance(String filePath) {
if (filePath == null || filePath.isEmpty()) {
System.out.println("Error: 无效的图像路径!");
return "";
}
// 请求url
String url = "https://aip.baidubce.com/oauth/2.0/token";
try {
// 使用传入的文件路径
byte[] imgData = FileUtil.readFileByBytes(filePath);
String imgStr = Base64Util.encode(imgData);
String imgParam = URLEncoder.encode(imgStr, "UTF-8");
String param = "image=" + imgParam;
// 获取 access_token(一般情况下可以缓存)
String accessToken = "24.b83939e6d9bca2a235df588ce0d745b0.2592000.1736018850.282335-116551191";
String result = HttpUtil.post(url, accessToken, param);
System.out.println("API 返回结果: " + result); // 打印返回结果用于调试
// 提取 Base64 编码的图片数据
String base64Image = extractBase64Image(result);
// 调试:打印提取的 Base64 字符串
System.out.println("提取的 Base64 字符串: " + base64Image);
// 将 Base64 字符串解码为字节数组并保存为图片文件
byte[] imageBytes = Base64.getDecoder().decode(base64Image);
String outputPath = "E:\\newpage\\enhanced_output_image.jpg"; // 保存的图片路径
// 打印保存路径,检查路径是否正确
System.out.println("保存的路径: " + outputPath);
try (FileOutputStream fos = new FileOutputStream(outputPath)) {
fos.write(imageBytes);
}
System.out.println("图片已保存至: " + outputPath);
return outputPath;
} catch (Exception e) {
e.printStackTrace();
}
return null;
}
// 提取返回结果中的 Base64 图片数据
private static String extractBase64Image(String json) {
// 使用正则或者 JSON 解析提取 base64 字符串(需要根据 API 返回的 JSON 格式来提取)
String base64Image = json.substring(json.indexOf("\"image\":\"") + 8, json.lastIndexOf("\""));
return base64Image;
}
public static void main(String[] args) {
Image_definition_enhance.image_definition_enhance("E:\\图片\\1724212883022.jpg");
}
}