From 667354561f68cbad9025ce9902dfb66e2b1f04b6 Mon Sep 17 00:00:00 2001
From: Passerby1011 <63802758+Passerby1011@users.noreply.github.com>
Date: Wed, 12 Mar 2025 22:37:28 +0800
Subject: [PATCH] Update route.ts

---
 src/app/api/generate/route.ts | 31 ++++++++++++++++++-------------
 1 file changed, 18 insertions(+), 13 deletions(-)

diff --git a/src/app/api/generate/route.ts b/src/app/api/generate/route.ts
index 31e92b1..6b7005d 100644
--- a/src/app/api/generate/route.ts
+++ b/src/app/api/generate/route.ts
@@ -3,8 +3,7 @@ import {createErrorResponse, getPluginSettingsFromRequest, PluginErrorType} from
 import {IImageGenerationResponse, IImageGenerationResult, IImageGenerationStatusResponse, Settings} from "@/type";
 import axios from "axios";
 
-const BASE_URL = process.env.BASE_URL || 'https://api.alibabacloud.com'
-
+const BASE_URL = process.env.BASE_URL || 'https://dashscope.aliyuncs.com/api/v1'
 
 export async function POST(req: NextRequest) {
 	try {
@@ -17,19 +16,23 @@ export async function POST(req: NextRequest) {
 		const apiKey = settings.ALIBABA_API_KEY
 
 		const body = await req.json();
-		const { prompt } = body
+		const {  model, input, parameters } = body;
+        
+		// 设置默认参数
+		const requestParameters = {
+			size: '1024*1024',
+			n: 1,
+			prompt_extend: true,
+			...parameters
+		};
 
 		// 调用阿里云的API生成图片
 		const response = await axios.post(
 			`${BASE_URL}/services/aigc/text2image/image-synthesis`,
 			{
-				model: 'wanx-v1',
-				input: {
-					prompt: prompt
-				},
-				parameters: {
-					n: 1,
-				},
+				model: model,
+				input: input,
+				parameters: requestParameters,
 			},
 			{
 				headers: {
@@ -57,7 +60,6 @@ export async function POST(req: NextRequest) {
 		const taskId = respData.output.task_id;
 
 		// 检查图片生成的状态
-
 		let statusData: IImageGenerationStatusResponse;
 
 		do {
@@ -93,7 +95,10 @@ export async function POST(req: NextRequest) {
 
       ![Generated Image](${imageUrl})
 
-      *提示词: ${prompt}*
+      *提示词: ${input.prompt}*
+      ${input.negative_prompt ? `*反向提示词: ${input.negative_prompt}*` : ''}
+      *模型: ${model}*
+      *尺寸: ${requestParameters.size}*
     `.trim();
 
 		// 返回生成的图片URL
@@ -104,4 +109,4 @@ export async function POST(req: NextRequest) {
 			message: 'Failed to generate image.'
 		})
 	}
-}
\ No newline at end of file
+}