软件构造

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");
    }
}

 

posted @ 2024-12-20 09:48  KuanDong24  阅读(2)  评论(0编辑  收藏  举报