Add File

agents-flex-image/agents-flex-image-volcengine/src/test/java/VolcengineImageTest.java

@@ -0,0 +1,99 @@
+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);
+
+
+
+    }
+}