feat: update image example

Signed-off-by: yuluo-yx <yuluo08290126@gmail.com>
alibaba · Oct 7, 2024 · 6cb68d7 · 6cb68d7
1 parent e355581
commit 6cb68d7
Showing 1 changed file with 15 additions and 6 deletions.
diff --git a/...tmodel-example/src/main/java/com/alibaba/cloud/ai/example/model/ImageModelController.java b/...tmodel-example/src/main/java/com/alibaba/cloud/ai/example/model/ImageModelController.java
@@ -34,18 +34,27 @@ public class ImageModelController {
 		this.imageModel = imageModel;
 	}
 
-	@RequestMapping("/image")
-	public String image(String input) {
+	@GetMapping("/image/{input}")
+	public void image(@PathVariable("input") String input, HttpServletResponse response) {
 
 		ImageOptions options = ImageOptionsBuilder.builder()
 				.withModel("wanx-v1")
 				.build();
 
 		ImagePrompt imagePrompt = new ImagePrompt(input, options);
-		ImageResponse response = imageModel.call(imagePrompt);
-		String imageUrl = response.getResult().getOutput().getUrl();
-
-		return "redirect:" + imageUrl;
+		ImageResponse imageResponse = imageModel.call(imagePrompt);
+		String imageUrl = imageResponse.getResult().getOutput().getUrl();
+
+		try {
+			URL url = new URL(imageUrl);
+			InputStream in = url.openStream();
+
+			response.setHeader("Content-Type", MediaType.IMAGE_PNG_VALUE);
+			response.getOutputStream().write(in.readAllBytes());
+			response.getOutputStream().flush();
+		} catch (IOException e) {
+			response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+		}
 	}
 
 }