content: import com.agentsflex.core.image.GenerateImageRequest;
  import com.agentsflex.core.image.ImageResponse;
  import com.agentsflex.image.volcengine.VolcengineImageModel;
  import com.agentsflex.image.volcengine.VolcengineImageModelConfig;
  import com.alibaba.fastjson.JSONObject;
  import org.junit.Test;
  
  import java.io.File;
  import java.io.IOException;
  import java.nio.file.Files;
  import java.nio.file.Paths;
  import java.util.ArrayList;
  import java.util.Base64;
  import java.util.List;
  
  public class VolcengineImageTest {
      @Test
      public void testGenImage(){
          VolcengineImageModelConfig config = new VolcengineImageModelConfig();
          config.setAccessKey("*********************");
          config.setSecretKey("*********************");
  
          VolcengineImageModel imageModel = new VolcengineImageModel(config);
  
          GenerateImageRequest request = new GenerateImageRequest();
  
          JSONObject req=new JSONObject();
          //请求Body(查看接口文档请求参数-请求示例，将请求参数内容复制到此)参考通用2.1-文生图
          req.put("req_key","high_aes_general_v21_L");
          req.put("prompt","千军万马");
          req.put("model_version","general_v2.1_L");
          req.put("req_schedule_conf","general_v20_9B_pe");
          req.put("llm_seed",-1);
          req.put("seed",-1);
          req.put("scale",3.5);
          req.put("ddim_steps",25);
          req.put("width",512);
          req.put("height",512);
          req.put("use_pre_llm",true);
          req.put("use_sr",true);
          req.put("sr_seed",-1);
          req.put("sr_strength",0.4);
          req.put("sr_scale",3.5);
          req.put("sr_steps",20);
          req.put("is_only_sr",false);
          req.put("return_url",true);
          // 创建子级JSONObject
          JSONObject subData = new JSONObject();
          subData.put("add_logo", true);
          subData.put("position", 2);
          subData.put("language", 0);
          subData.put("opacity", 0.3);
          subData.put("logo_text_content", "wangyangyang");
          req.put("logo_info",subData);
  
          request.setOptions(req);
  
          ImageResponse generate = imageModel.generate(request);
          System.out.println(generate);
      }
  
  
  
  
      @Test( )
      public void testImg2ImgXLSft() throws IOException {
          VolcengineImageModelConfig config = new VolcengineImageModelConfig();
          config.setAccessKey("*********************");
          config.setSecretKey("*********************");
  
          VolcengineImageModel imageModel = new VolcengineImageModel(config);
  
          GenerateImageRequest request = new GenerateImageRequest();
  
          JSONObject req=new JSONObject();
          req.put("req_key","i2i_xl_sft");
          List<String> images=new ArrayList<>();
  
          File file = new File(System.getProperty("user.dir"), "../../testresource/ark_demo_img_1.png");
          // 将图片读取为字节数组
          byte[] imageBytes = Files.readAllBytes(Paths.get(file.toURI()));
  
          // 将字节数组编码为Base64
          String base64String = Base64.getEncoder().encodeToString(imageBytes);
  
          images.add(base64String);
  //        images.add("https://ark-project.tos-cn-beijing.volces.com/doc_image/ark_demo_img_1.png");
          req.put("binary_data_base64",images);
          req.put("prompt","根据图片内容生成风格、服装及发型一样的亚洲美女图片");
          req.put("return_url",true);
          request.setOptions(req);
  
          ImageResponse generate = imageModel.img2imggenerate(request);
          System.out.println(generate);
  
  
  
      }
  }
  