From 9bd8d8e9cbeb69f71b9134c99b6196790f0c1aa8 Mon Sep 17 00:00:00 2001
From: cherishsince <cherishsince@aliyun.com>
Date: Fri, 31 May 2024 14:25:16 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E5=A2=9E=E5=8A=A0=E3=80=91midjourney?=
 =?UTF-8?q?=20=E5=A2=9E=E5=8A=A0=E7=89=88=E6=9C=AC=E5=8F=B7=E8=AE=BE?=
 =?UTF-8?q?=E7=BD=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/ai/image/index.ts               |  1 +
 src/views/ai/image/midjourney/index.vue | 66 +++++++++++++++++++++++++
 2 files changed, 67 insertions(+)

diff --git a/src/api/ai/image/index.ts b/src/api/ai/image/index.ts
index 1df168e5..e2c9b1c1 100644
--- a/src/api/ai/image/index.ts
+++ b/src/api/ai/image/index.ts
@@ -48,6 +48,7 @@ export interface ImageMidjourneyImagineReqVO {
   base64Array: string[] // size不能为空
   width: string // 图片宽度
   height: string // 图片高度
+  version: string // 版本
 }
 
 // TODO 芋艿：review 下整体注释、方法名
diff --git a/src/views/ai/image/midjourney/index.vue b/src/views/ai/image/midjourney/index.vue
index ef1d7822..2cfa475d 100644
--- a/src/views/ai/image/midjourney/index.vue
+++ b/src/views/ai/image/midjourney/index.vue
@@ -46,6 +46,28 @@
       </div>
     </el-space>
   </div>
+  <div class="version">
+    <div>
+      <el-text tag="b">版本</el-text>
+    </div>
+    <el-space wrap class="version-list">
+      <el-select
+        v-model="selectVersion"
+        class="version-select"
+        clearable
+        placeholder="请选择版本"
+        style="width: 350px"
+        @change="handlerChangeVersion"
+      >
+        <el-option
+          v-for="item in versionList"
+          :key="item.value"
+          :label="item.label"
+          :value="item.value"
+        />
+      </el-select>
+    </el-space>
+  </div>
   <div class="model">
     <div>
       <el-text tag="b">模型</el-text>
@@ -147,6 +169,32 @@ const imageSizeList = ref<ImageSizeVO[]>([
 ]) // size
 selectImageSize.value = imageSizeList.value[0]
 
+
+// version
+const versionList = ref<any>([
+  {
+    value: '6.0',
+    label: 'v6.0',
+  },
+  {
+    value: '5.2',
+    label: 'v5.2',
+  },
+  {
+    value: '5.1',
+    label: 'v5.1',
+  },
+  {
+    value: '5.0',
+    label: 'v5.0',
+  },
+  {
+    value: '4.0',
+    label: 'v4.0',
+  },
+]) // version 列表
+const selectVersion = ref<any>('6.0') // 选中的 version
+
 // 定义 Props
 const props = defineProps({})
 
@@ -183,6 +231,13 @@ const handlerModelClick = async (model: ImageModelVO) => {
   selectModel.value = model
 }
 
+/**
+ * version - click
+ */
+const handlerChangeVersion = async (version) => {
+  console.log('version', version)
+}
+
 /**
  * 图片生产
  */
@@ -197,6 +252,7 @@ const handlerGenerateImage = async () => {
       model: selectModel.value.key,
       width: selectImageSize.value.width,
       height: selectImageSize.value.height,
+      version: selectVersion.value,
       base64Array: [],
     } as ImageMidjourneyImagineReqVO
     await ImageApi.midjourneyImagine(req)
@@ -231,6 +287,16 @@ const handlerGenerateImage = async () => {
   }
 }
 
+// version
+.version {
+  margin-top: 20px;
+
+  .version-list {
+    margin-top: 20px;
+    width: 100%;
+  }
+}
+
 // 模型
 .model {
   margin-top: 30px;