spring-ai 学习系列(6)-文生图

菩提树下的杨过

发布于 2025-08-22 08:33:52

9800

代码可运行

文章被收录于专栏：菩提树下的杨过菩提树下的杨过

运行总次数：0

代码可运行

继续来学习spring-ai如何实现“文生图”

一、pom依赖

1 <dependency>
2     <groupId>org.springframework.ai</groupId>
3     <artifactId>spring-ai-starter-model-zhipuai</artifactId>
4     <version>1.0.0</version>
5 </dependency>

spring-ai支持多种文生图的模型，这里我们使用国产的智谱大模型，可以先免费体验，api-key申请地址：https://bigmodel.cn/usercenter/proj-mgmt/apikeys

二、yaml配置

spring:
  ai:
    zhipuai:
      api-key: ${zhipuai_api_key}

这里api-key换成真实的key

三、文生图代码

@GetMapping("/image/url")
public String genImageUrl(@RequestParam String prompt) {
    var options = ImageOptionsBuilder.builder().height(256).width(256).build();
    ImageResponse response = zhiPuAiImageModel.call(
            new ImagePrompt(prompt,
                    options));
    Image output = response.getResult().getOutput();
    return output.getUrl();
}

@GetMapping("/image/display")
public ResponseEntity<ByteArrayResource> displayImage(@RequestParam String prompt) {
    try {
        var options = ImageOptionsBuilder.builder().height(256).width(256).build();
        ImageResponse response = zhiPuAiImageModel.call(
                new ImagePrompt(prompt, options));
        Image output = response.getResult().getOutput();
        
        // 获取图片URL
        String imageUrl = output.getUrl();
        
        // 下载图片数据
        URL url = URI.create(imageUrl).toURL();
        byte[] imageData = url.openStream().readAllBytes();
        
        // 创建ByteArrayResource
        ByteArrayResource resource = new ByteArrayResource(imageData);
        
        // 设置响应头 - 直接在浏览器中显示
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.IMAGE_PNG);
        
        return ResponseEntity.ok()
                .headers(headers)
                .body(resource);
                
    } catch (IOException e) {
        return ResponseEntity.internalServerError().build();
    }
}

运行效果：