From a1d7e81859f554f3a53680cc35f0f49bf1f77098 Mon Sep 17 00:00:00 2001
From: wwf <1971391498@qq.com>
Date: 星期四, 14 五月 2026 14:37:02 +0800
Subject: [PATCH] 导入项目
---
src/views/ai/image/index/components/stableDiffusion/index.vue | 257 +++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 257 insertions(+), 0 deletions(-)
diff --git a/src/views/ai/image/index/components/stableDiffusion/index.vue b/src/views/ai/image/index/components/stableDiffusion/index.vue
new file mode 100644
index 0000000..1fa68d5
--- /dev/null
+++ b/src/views/ai/image/index/components/stableDiffusion/index.vue
@@ -0,0 +1,257 @@
+<!-- dall3 -->
+<template>
+ <div class="prompt">
+ <el-text tag="b">鐢婚潰鎻忚堪</el-text>
+ <el-text tag="p">寤鸿浣跨敤鈥滃舰瀹硅瘝 + 鍔ㄨ瘝 + 椋庢牸鈥濈殑鏍煎紡锛屼娇鐢ㄢ�滐紝鈥濋殧寮�</el-text>
+ <el-input
+ v-model="prompt"
+ maxlength="1024"
+ :rows="5"
+ class="w-100% mt-15px"
+ input-style="border-radius: 7px;"
+ placeholder="渚嬪锛氱璇濋噷鐨勫皬灞嬪簲璇ユ槸浠�涔堟牱瀛愶紵"
+ show-word-limit
+ type="textarea"
+ />
+ </div>
+ <div class="flex flex-col mt-30px">
+ <div>
+ <el-text tag="b">闅忔満鐑瘝</el-text>
+ </div>
+ <el-space wrap class="flex flex-row flex-wrap justify-start mt-15px">
+ <el-button
+ round
+ class="m-0"
+ :type="selectHotWord === hotWord ? 'primary' : 'default'"
+ v-for="hotWord in ImageHotEnglishWords"
+ :key="hotWord"
+ @click="handleHotWordClick(hotWord)"
+ >
+ {{ hotWord }}
+ </el-button>
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">閲囨牱鏂规硶</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-select v-model="sampler" placeholder="Select" size="large" class="!w-350px">
+ <el-option
+ v-for="item in StableDiffusionSamplers"
+ :key="item.key"
+ :label="item.name"
+ :value="item.key"
+ />
+ </el-select>
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">CLIP</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-select v-model="clipGuidancePreset" placeholder="Select" size="large" class="!w-350px">
+ <el-option
+ v-for="item in StableDiffusionClipGuidancePresets"
+ :key="item.key"
+ :label="item.name"
+ :value="item.key"
+ />
+ </el-select>
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">椋庢牸</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-select v-model="stylePreset" placeholder="Select" size="large" class="!w-350px">
+ <el-option
+ v-for="item in StableDiffusionStylePresets"
+ :key="item.key"
+ :label="item.name"
+ :value="item.key"
+ />
+ </el-select>
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">鍥剧墖灏哄</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-input v-model="width" class="w-170px" placeholder="鍥剧墖瀹藉害" />
+ <el-input v-model="height" class="w-170px" placeholder="鍥剧墖楂樺害" />
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">杩唬姝ユ暟</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-input
+ v-model="steps"
+ type="number"
+ size="large"
+ class="!w-350px"
+ placeholder="Please input"
+ />
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">寮曞绯绘暟</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-input
+ v-model="scale"
+ type="number"
+ size="large"
+ class="!w-350px"
+ placeholder="Please input"
+ />
+ </el-space>
+ </div>
+ <div class="mt-30px">
+ <div>
+ <el-text tag="b">闅忔満鍥犲瓙</el-text>
+ </div>
+ <el-space wrap class="mt-15px w-full">
+ <el-input
+ v-model="seed"
+ type="number"
+ size="large"
+ class="!w-350px"
+ placeholder="Please input"
+ />
+ </el-space>
+ </div>
+ <div class="flex justify-center mt-50px">
+ <el-button
+ type="primary"
+ size="large"
+ round
+ :loading="drawIn"
+ :disabled="prompt.length === 0"
+ @click="handleGenerateImage"
+ >
+ {{ drawIn ? '鐢熸垚涓�' : '鐢熸垚鍐呭' }}
+ </el-button>
+ </div>
+</template>
+<script setup lang="ts">
+import { ImageApi, ImageDrawReqVO, ImageVO } from '@/api/ai/image'
+import { hasChinese } from '@/views/ai/utils/utils'
+import {
+ AiPlatformEnum,
+ ImageHotEnglishWords,
+ StableDiffusionClipGuidancePresets,
+ StableDiffusionSamplers,
+ StableDiffusionStylePresets
+} from '@/views/ai/utils/constants'
+import { ModelVO } from '@/api/ai/model/model'
+
+const message = useMessage() // 娑堟伅寮圭獥
+
+// 鎺ユ敹鐖剁粍浠朵紶鍏ョ殑妯″瀷鍒楄〃
+const props = defineProps({
+ models: {
+ type: Array<ModelVO>,
+ default: () => [] as ModelVO[]
+ }
+})
+const emits = defineEmits(['onDrawStart', 'onDrawComplete']) // 瀹氫箟 emits
+
+// 瀹氫箟灞炴��
+const drawIn = ref<boolean>(false) // 鐢熸垚涓�
+const selectHotWord = ref<string>('') // 閫変腑鐨勭儹璇�
+// 琛ㄥ崟
+const prompt = ref<string>('') // 鎻愮ず璇�
+const width = ref<number>(512) // 鍥剧墖瀹藉害
+const height = ref<number>(512) // 鍥剧墖楂樺害
+const sampler = ref<string>('DDIM') // 閲囨牱鏂规硶
+const steps = ref<number>(20) // 杩唬姝ユ暟
+const seed = ref<number>(42) // 鎺у埗鐢熸垚鍥惧儚鐨勯殢鏈烘��
+const scale = ref<number>(7.5) // 寮曞绯绘暟
+const clipGuidancePreset = ref<string>('NONE') // 鏂囨湰鎻愮ず鐩稿尮閰嶇殑鍥惧儚(clip_guidance_preset) 绠�绉� CLIP
+const stylePreset = ref<string>('3d-model') // 椋庢牸
+
+/** 閫夋嫨鐑瘝 */
+const handleHotWordClick = async (hotWord: string) => {
+ // 鎯呭喌涓�锛氬彇娑堥�変腑
+ if (selectHotWord.value == hotWord) {
+ selectHotWord.value = ''
+ return
+ }
+
+ // 鎯呭喌浜岋細閫変腑
+ selectHotWord.value = hotWord // 閫変腑
+ prompt.value = hotWord // 鏇挎崲鎻愮ず璇�
+}
+
+/** 鍥剧墖鐢熸垚 */
+const handleGenerateImage = async () => {
+ // 浠� models 涓煡鎵惧尮閰嶇殑妯″瀷
+ const selectModel = 'stable-diffusion-v1-6'
+ const matchedModel = props.models.find(
+ (item) => item.model === selectModel && item.platform === AiPlatformEnum.STABLE_DIFFUSION
+ )
+ if (!matchedModel) {
+ message.error('璇ユā鍨嬩笉鍙敤锛岃閫夋嫨鍏跺畠妯″瀷')
+ return
+ }
+
+ // 浜屾纭
+ if (hasChinese(prompt.value)) {
+ message.alert('鏆備笉鏀寔涓枃锛�')
+ return
+ }
+ await message.confirm(`纭鐢熸垚鍐呭?`)
+
+ try {
+ // 鍔犺浇涓�
+ drawIn.value = true
+ // 鍥炶皟
+ emits('onDrawStart', AiPlatformEnum.STABLE_DIFFUSION)
+ // 鍙戦�佽姹�
+ const form = {
+ modelId: matchedModel.id,
+ prompt: prompt.value, // 鎻愮ず璇�
+ width: width.value, // 鍥剧墖瀹藉害
+ height: height.value, // 鍥剧墖楂樺害
+ options: {
+ seed: seed.value, // 闅忔満绉嶅瓙
+ steps: steps.value, // 鍥剧墖鐢熸垚姝ユ暟
+ scale: scale.value, // 寮曞绯绘暟
+ sampler: sampler.value, // 閲囨牱绠楁硶
+ clipGuidancePreset: clipGuidancePreset.value, // 鏂囨湰鎻愮ず鐩稿尮閰嶇殑鍥惧儚 CLIP
+ stylePreset: stylePreset.value // 椋庢牸
+ }
+ } as ImageDrawReqVO
+ await ImageApi.drawImage(form)
+ } finally {
+ // 鍥炶皟
+ emits('onDrawComplete', AiPlatformEnum.STABLE_DIFFUSION)
+ // 鍔犺浇缁撴潫
+ drawIn.value = false
+ }
+}
+
+/** 濉厖鍊� */
+const settingValues = async (detail: ImageVO) => {
+ prompt.value = detail.prompt
+ width.value = detail.width
+ height.value = detail.height
+ seed.value = detail.options?.seed
+ steps.value = detail.options?.steps
+ scale.value = detail.options?.scale
+ sampler.value = detail.options?.sampler
+ clipGuidancePreset.value = detail.options?.clipGuidancePreset
+ stylePreset.value = detail.options?.stylePreset
+}
+
+/** 鏆撮湶缁勪欢鏂规硶 */
+defineExpose({ settingValues })
+</script>
+
--
Gitblit v1.8.0