文生图 #79
|
|
@ -0,0 +1,81 @@
|
||||||
|
import axios from 'axios'
|
||||||
|
import request from '@/utils/request'
|
||||||
|
import { getToken } from "@/utils/auth";
|
||||||
|
|
||||||
|
// 文生图片
|
||||||
|
export function convertTextToPicture(data) {
|
||||||
|
return axios({
|
||||||
|
url: 'https://ai.ysaix.com:7853/prompt',
|
||||||
|
method: 'post',
|
||||||
|
headers: {
|
||||||
|
'Content-Type': 'application/json',
|
||||||
|
'Accept': '*/*'
|
||||||
|
},
|
||||||
|
data: data
|
||||||
|
})
|
||||||
|
}
|
||||||
|
// 获取任务列表
|
||||||
|
export function getQueue() {
|
||||||
|
return axios({
|
||||||
|
url: `https://ai.ysaix.com:7853/queue`,
|
||||||
|
method: 'get',
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取生图任务id
|
||||||
|
export function getPromptId(id) {
|
||||||
|
return axios({
|
||||||
|
url: `https://ai.ysaix.com:7853/history/${id}`,
|
||||||
|
method: 'get',
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 获取生成图片路径
|
||||||
|
export function getPicture(data) {
|
||||||
|
return axios({
|
||||||
|
url: 'https://ai.ysaix.com:7853/view',
|
||||||
|
method: 'get',
|
||||||
|
params: data
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 大模型对话生成prompt模板
|
||||||
|
export function chattoprompt(dataset_id,prompt) {
|
||||||
|
return axios({
|
||||||
|
url: '/api/v1/parse/docs',
|
||||||
|
method: 'post',
|
||||||
|
headers: {
|
||||||
|
'Authorization': 'Bearer ragflow-IwMDI1MGU2YTU3NjExZWZiNWEzMDI0Mm',
|
||||||
|
},
|
||||||
|
data: {
|
||||||
|
'dataset_id': dataset_id,
|
||||||
|
'prompt': prompt
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// prompt敏感词校验
|
||||||
|
export function textSensitiveWord(data) {
|
||||||
|
return request({
|
||||||
|
url: '/verify/text',
|
||||||
|
method: 'post',
|
||||||
|
data: {
|
||||||
|
'text' : data
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 图片上传资源库
|
||||||
|
export function uploadPicture(data) {
|
||||||
|
return axios({
|
||||||
|
url: '/dev-api/smarttalk/file/upload',
|
||||||
|
method: 'post',
|
||||||
|
headers: {
|
||||||
|
'Accept': '*/*',
|
||||||
|
'Content-Type': 'multipart/form-data',
|
||||||
|
'Authorization': "Bearer " + getToken()
|
||||||
|
},
|
||||||
|
data: data
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
|
@ -0,0 +1,627 @@
|
||||||
|
<template>
|
||||||
|
<div>
|
||||||
|
<el-row :gutter="40" style="height:auto">
|
||||||
|
<el-col :span="8">
|
||||||
|
<div>
|
||||||
|
<div style="text-align: left">反面提示词</div>
|
||||||
|
<br>
|
||||||
|
<el-input v-model="formData.prompt[12].inputs.negative_prompt" placeholder="试试在这里输入你不想其进入图画的词"
|
||||||
|
type="textarea"></el-input>
|
||||||
|
</div>
|
||||||
|
<br>
|
||||||
|
<div>
|
||||||
|
<div style="text-align: left">可选提示词</div>
|
||||||
|
<br>
|
||||||
|
<div style="font-size: 14px;text-align: left">风格特点</div>
|
||||||
|
<el-checkbox-group v-model="checkList">
|
||||||
|
<el-checkbox v-for="(prompt, index) in promptOption1" :key="index" :label="prompt" :value="prompt" />
|
||||||
|
</el-checkbox-group>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<div style="font-size: 14px;text-align: left">主体对象</div>
|
||||||
|
<el-checkbox-group v-model="checkList">
|
||||||
|
<el-checkbox v-for="(prompt, index) in promptOption2" :key="index" :label="prompt" :value="prompt" />
|
||||||
|
</el-checkbox-group>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<div style="font-size: 14px;text-align: left">场景</div>
|
||||||
|
<el-checkbox-group v-model="checkList">
|
||||||
|
<el-checkbox v-for="(prompt, index) in promptOption3" :key="index" :label="prompt" :value="prompt" />
|
||||||
|
</el-checkbox-group>
|
||||||
|
</div>
|
||||||
|
<div style="text-align: right">
|
||||||
|
<el-button type="primary" @click="addCheckListToPrompt">添加</el-button>
|
||||||
|
<el-button type="primary" @click="deletePrompt">清空</el-button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!-- 具体参数相关内容 -->
|
||||||
|
<el-form label-position="top" size="small">
|
||||||
|
<el-form-item label="图片尺寸" style=" margin-bottom: 0px">
|
||||||
|
<el-row class="ratio-options" :gutter="10" style="margin-bottom: -15px;margin-top: -25px">
|
||||||
|
<el-col v-for="ratio in ratioOptions" :key="ratio.value" :span="4">
|
||||||
|
<el-button :type="form.ratio === ratio.value ? 'primary' : 'default'" @click="setRatio(ratio.value)"
|
||||||
|
class="ratio-option" block>
|
||||||
|
{{ ratio.label }}
|
||||||
|
</el-button>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item label="图片比例" style=" margin-bottom: 0px">
|
||||||
|
<el-row class="ratio-options" :gutter="10" style="margin-bottom: -15px;margin-top: -25px">
|
||||||
|
<el-col v-for="ratio in ratioOptions2" :key="ratio.value" :span="4">
|
||||||
|
<el-button :type="form.ratios2 === ratio.value ? 'primary' : 'default'" @click="setRatio2(ratio.value)"
|
||||||
|
class="ratio-option" block>
|
||||||
|
{{ ratio.label }}
|
||||||
|
</el-button>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item label="图片数量" style=" margin-bottom: 0px">
|
||||||
|
<el-row style="margin-bottom: -15px;margin-top: -25px">
|
||||||
|
<el-button style="width: 10%" :type="num_picture === 1 ? 'primary' : 'default'" @click="changePicture(1)">
|
||||||
|
1
|
||||||
|
</el-button>
|
||||||
|
<el-button style="width: 10%" :type="num_picture === 2 ? 'primary' : 'default'" @click="changePicture(2)">
|
||||||
|
2
|
||||||
|
</el-button>
|
||||||
|
<el-button style="width: 10%" :type="num_picture === 3 ? 'primary' : 'default'" @click="changePicture(3)">
|
||||||
|
3
|
||||||
|
</el-button>
|
||||||
|
<el-button style="width: 10%" :type="num_picture === 4 ? 'primary' : 'default'" @click="changePicture(4)">
|
||||||
|
4
|
||||||
|
</el-button>
|
||||||
|
</el-row>
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item label="随机种子">
|
||||||
|
<el-slider v-model="formData.prompt[14].inputs.seed" :min="1" :max="10000000" show-input />
|
||||||
|
</el-form-item>
|
||||||
|
<el-form-item label="cfg (数值越高，生图过程与提示词越相关)">
|
||||||
|
<el-slider v-model="formData.prompt[14].inputs.cfg" :max="20" show-input />
|
||||||
|
</el-form-item>
|
||||||
|
</el-form>
|
||||||
|
</el-col>
|
||||||
|
<el-col :span="16">
|
||||||
|
<div
|
||||||
|
:style="{ height: divHeight + 'px', 'overflow-y': 'auto', 'margin-bottom': '10px', 'background-color': '#f5f5f5', 'padding': '10px' }">
|
||||||
|
<div v-for="(resultItem, resultIndex) in resultList" :key="resultIndex">
|
||||||
|
<div style="display: flex; flex-wrap: wrap; justify-content: flex-end;">
|
||||||
|
<el-card style="max-width: 50%; margin-right: 10px; display: inline-block;background-color: lightblue;">
|
||||||
|
<div>
|
||||||
|
<p style="word-wrap: break-word; overflow-wrap: break-word;text-align: left">{{ resultItem }}</p>
|
||||||
|
</div>
|
||||||
|
</el-card>
|
||||||
|
</div>
|
||||||
|
<el-row :gutter="10" justify="center">
|
||||||
|
<el-col :span="6" v-for="(url, index) in pictureurl[resultIndex]" :key="index">
|
||||||
|
<el-card style="margin-bottom: 5%;height:95%">
|
||||||
|
<el-image style="width: 100%;" fit="cover" :src="url" :preview-src-list="pictureurl"
|
||||||
|
:initial-index="index" class="equal-size-image"></el-image>
|
||||||
|
<div style="text-align: center; margin-top: 15px">
|
||||||
|
<el-button :disabled="buttonStates[resultIndex][index].disabled" size="small" type="primary"
|
||||||
|
@click="saveImage(resultIndex, index, url, resultItem)">{{ buttonStates[resultIndex][index].text
|
||||||
|
}}</el-button>
|
||||||
|
</div>
|
||||||
|
</el-card>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
</div>
|
||||||
|
<el-row :gutter="10" justify="center">
|
||||||
|
<el-col v-if="pictureLoading" :span="6" v-for="n in generateArray(this.skeletonNumber)" :key="n">
|
||||||
|
<el-card>
|
||||||
|
<el-skeleton class="custom-skeleton-item" animated>
|
||||||
|
<template #template>
|
||||||
|
<el-skeleton-item variant="image" class="custom-skeleton-item"></el-skeleton-item>
|
||||||
|
</template>
|
||||||
|
</el-skeleton>
|
||||||
|
</el-card>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
<el-row :gutter="10" justify="center">
|
||||||
|
<el-col :span="12">
|
||||||
|
<el-card v-for="(item, index) in resultData" :key="index"
|
||||||
|
style="display: flex;flex-direction: column;align-items: center;margin-bottom: 10px" justify="center"
|
||||||
|
@click="handleCardClick(item)" :class="{ 'card-hover': !pictureLoading }">
|
||||||
|
<p style="word-wrap: break-word; overflow-wrap: break-word;">{{ item }}</p>
|
||||||
|
</el-card>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
</div>
|
||||||
|
<div style="text-align: center">
|
||||||
|
<el-input style="width: 70%;" v-model="promptData" placeholder="试试输入你心中的画面，尽量描述具体点哦，可以按照这个格式来写： 提示词=主体+风格+场景"
|
||||||
|
type="textarea">
|
||||||
|
</el-input>
|
||||||
|
<el-button type="primary" :disabled="pictureLoading" @click="fetchData">
|
||||||
|
{{ !pictureLoading ? "生成图片" : "请等待" }}
|
||||||
|
</el-button>
|
||||||
|
</div>
|
||||||
|
</el-col>
|
||||||
|
</el-row>
|
||||||
|
</div>
|
||||||
|
</template>
|
||||||
|
|
||||||
|
<script>
|
||||||
|
import { convertTextToPicture, getQueue, getPromptId, getPicture, chattoprompt, textSensitiveWord, uploadPicture } from "@/api/aiGeneratedImage/index.js";
|
||||||
|
import CryptoJS from 'crypto-js'
|
||||||
|
import { useRoute } from 'vue-router'
|
||||||
|
export default {
|
||||||
|
|
||||||
|
data() {
|
||||||
|
return {
|
||||||
|
form: {
|
||||||
|
ratio: "512",
|
||||||
|
ratios2: "1/1",
|
||||||
|
},
|
||||||
|
ratioOptions: [
|
||||||
|
{ value: "512", label: "512" },
|
||||||
|
{ value: "768", label: "768" },
|
||||||
|
{ value: "1024", label: "1024" },
|
||||||
|
{ value: "1280", label: "1280" },
|
||||||
|
{ value: "2048", label: "2048" },
|
||||||
|
],
|
||||||
|
ratioOptions2: [
|
||||||
|
{ value: "1/1", label: "1:1" },
|
||||||
|
{ value: "4/3", label: "4:3" },
|
||||||
|
{ value: "3/4", label: "3:4" },
|
||||||
|
{ value: "9/16", label: "9:16" },
|
||||||
|
{ value: "16/9", label: "16:9" },
|
||||||
|
],
|
||||||
|
//图片数据
|
||||||
|
imageData: {
|
||||||
|
imageUrls: [],
|
||||||
|
prompt: [],
|
||||||
|
time: [],
|
||||||
|
id: [],
|
||||||
|
},
|
||||||
|
promptOption1: ["赛博朋克", "水墨风", "莫奈风格", "二次元", "中国风", "写实风格", "水彩风格", "工笔画", "素描风", "未来主义", "超现实主义", "映像派"],
|
||||||
|
promptOption2: ["男生", "女生", "老年人", "船舶", "蝴蝶", "狮子", "兔子", "飞机", "中年人", "大树", "长江", "坦克"],
|
||||||
|
promptOption3: ["雨林", "沙漠", "湖泊", "天空", "城市", "乡村", "太空", "教室"],
|
||||||
|
promptData: "",
|
||||||
|
|
||||||
|
//模型参数
|
||||||
|
formData: {
|
||||||
|
client_id: "533ef3a3-39c0-4e39-9ced-37d290f371f8",
|
||||||
|
prompt: {
|
||||||
|
3: {
|
||||||
|
inputs: {
|
||||||
|
images: ["10", 0],
|
||||||
|
},
|
||||||
|
class_type: "PreviewImage",
|
||||||
|
_meta: {
|
||||||
|
title: "Preview Image",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
6: {
|
||||||
|
inputs: {
|
||||||
|
model: "Kwai-Kolors/Kolors",
|
||||||
|
precision: "fp16",
|
||||||
|
},
|
||||||
|
class_type: "DownloadAndLoadKolorsModel",
|
||||||
|
_meta: {
|
||||||
|
title: "(Down)load Kolors Model",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
10: {
|
||||||
|
inputs: {
|
||||||
|
samples: ["14", 0],
|
||||||
|
vae: ["11", 0],
|
||||||
|
},
|
||||||
|
class_type: "VAEDecode",
|
||||||
|
_meta: {
|
||||||
|
title: "VAE Decode",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
11: {
|
||||||
|
inputs: {
|
||||||
|
vae_name: "sdxl.vae.safetensors",
|
||||||
|
},
|
||||||
|
class_type: "VAELoader",
|
||||||
|
_meta: {
|
||||||
|
title: "Load VAE",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
12: {
|
||||||
|
inputs: {
|
||||||
|
prompt: "",
|
||||||
|
negative_prompt: "",
|
||||||
|
num_images_per_prompt: 1,
|
||||||
|
chatglm3_model: ["13", 0],
|
||||||
|
},
|
||||||
|
class_type: "KolorsTextEncode",
|
||||||
|
_meta: {
|
||||||
|
title: "Kolors Text Encode",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
13: {
|
||||||
|
inputs: {
|
||||||
|
precision: "quant8",
|
||||||
|
},
|
||||||
|
class_type: "DownloadAndLoadChatGLM3",
|
||||||
|
_meta: {
|
||||||
|
title: "(Down)load ChatGLM3 Model",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
14: {
|
||||||
|
inputs: {
|
||||||
|
width: 1024,
|
||||||
|
height: 1024,
|
||||||
|
seed: 1000102404233412,
|
||||||
|
steps: 25,
|
||||||
|
cfg: 5,
|
||||||
|
scheduler: "EulerDiscreteScheduler",
|
||||||
|
denoise_strength: 1,
|
||||||
|
kolors_model: ["6", 0],
|
||||||
|
kolors_embeds: ["12", 0],
|
||||||
|
},
|
||||||
|
class_type: "KolorsSampler",
|
||||||
|
_meta: {
|
||||||
|
title: "Kolors Sampler",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
picture: {
|
||||||
|
filename: "",
|
||||||
|
type: "temp",
|
||||||
|
subfolder: "",
|
||||||
|
preview: "",
|
||||||
|
channel: "",
|
||||||
|
},
|
||||||
|
buttonStates: [], // 用于存储按钮状态的二维数组，初始为空
|
||||||
|
resultList: [],
|
||||||
|
pictureurl: [],
|
||||||
|
percentage: 50,
|
||||||
|
num_picture: 1,
|
||||||
|
skeletonNumber: 0,
|
||||||
|
cfg_value: 5,
|
||||||
|
pictureLoading: false,
|
||||||
|
QueueNumber: 0, //前方排队任务数量
|
||||||
|
userIdInComponent: '',
|
||||||
|
activeName: '1',
|
||||||
|
checkList: [], //prompt生成
|
||||||
|
totalData: 0,
|
||||||
|
// deleteImageSrc: deletebutton,
|
||||||
|
timer: null,
|
||||||
|
isTimerPaused: false,
|
||||||
|
resultData: [],
|
||||||
|
divHeight: 0,
|
||||||
|
dataset_id: '',
|
||||||
|
courseName: '',
|
||||||
|
levelFirstId: '',
|
||||||
|
levelSecondId: '',
|
||||||
|
textbookId: '',
|
||||||
|
};
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
//生图占位图数量
|
||||||
|
generateArray(length) {
|
||||||
|
return Array.from({ length }, (_, index) => index);
|
||||||
|
},
|
||||||
|
|
||||||
|
//随机种子
|
||||||
|
Randomseed() {
|
||||||
|
this.formData.prompt[14].inputs.seed = Math.floor(
|
||||||
|
Math.random() * 10000001
|
||||||
|
);
|
||||||
|
},
|
||||||
|
sleep(ms) {
|
||||||
|
return new Promise((resolve) => setTimeout(resolve, ms));
|
||||||
|
},
|
||||||
|
|
||||||
|
//生成图片
|
||||||
|
async fetchData() {
|
||||||
|
|
||||||
|
// 判断prompt是否为空
|
||||||
|
if (this.promptData == "") {
|
||||||
|
this.$message.error("prompt不能为空");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const prompt = this.promptData
|
||||||
|
//敏感词校验
|
||||||
|
await textSensitiveWord(prompt)
|
||||||
|
|
||||||
|
this.setRatio(this.form.ratio)
|
||||||
|
this.changeSize()
|
||||||
|
this.resultList.push(this.promptData)
|
||||||
|
this.skeletonNumber = this.num_picture
|
||||||
|
const skeletonItemNumber = this.num_picture
|
||||||
|
// const userId = this.userIdInComponent
|
||||||
|
// const model = 'kolors'
|
||||||
|
|
||||||
|
// 处理prompt生成多张图片
|
||||||
|
let temp = "";
|
||||||
|
if (skeletonItemNumber > 1) {
|
||||||
|
for (let i = 1; i < skeletonItemNumber; i++) {
|
||||||
|
temp += "|" + this.promptData;
|
||||||
|
}
|
||||||
|
this.formData.prompt[12].inputs.prompt = this.promptData + temp;
|
||||||
|
} else {
|
||||||
|
this.formData.prompt[12].inputs.prompt = this.promptData;
|
||||||
|
}
|
||||||
|
// 将数据转成json
|
||||||
|
const transformedData = {
|
||||||
|
client_id: this.formData.client_id,
|
||||||
|
prompt: Object.fromEntries(
|
||||||
|
Object.entries(this.formData.prompt).map(([key, value]) => [
|
||||||
|
key,
|
||||||
|
{
|
||||||
|
inputs: value.inputs,
|
||||||
|
class_type: value.class_type,
|
||||||
|
_meta: value._meta,
|
||||||
|
},
|
||||||
|
])
|
||||||
|
),
|
||||||
|
};
|
||||||
|
|
||||||
|
const jsonData = JSON.stringify(transformedData, null, 2);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
//生图过程
|
||||||
|
try {
|
||||||
|
this.pictureLoading = true;
|
||||||
|
const response = await convertTextToPicture(jsonData); //发起生图任务
|
||||||
|
const pictureId = response.data.prompt_id;
|
||||||
|
const queue_Number = response.data.number
|
||||||
|
let queue;
|
||||||
|
do {
|
||||||
|
queue = await getQueue(); //查询生图队列
|
||||||
|
if (this.hasId(queue, pictureId)) {
|
||||||
|
this.QueueNumber = queue_Number - queue.data.queue_running[0][0] // 计算队列剩余数量
|
||||||
|
await this.sleep(500);
|
||||||
|
}
|
||||||
|
} while (this.hasId(queue, pictureId));
|
||||||
|
|
||||||
|
const pictureData = await getPromptId(pictureId); //生图任务id查询结果
|
||||||
|
|
||||||
|
const jsonString = JSON.parse(pictureData.request.responseText);
|
||||||
|
const urls = []
|
||||||
|
const buttonState = [];
|
||||||
|
for (let i = 0; i < skeletonItemNumber; i++) {
|
||||||
|
this.picture.filename = jsonString[pictureId].outputs[3].images[i].filename;
|
||||||
|
const pictureURL = await getPicture(this.picture); //获得生成图片的url
|
||||||
|
const url0 = pictureURL.request.responseURL;
|
||||||
|
urls.push(url0)
|
||||||
|
buttonState.push({
|
||||||
|
disabled: false,
|
||||||
|
text: "插入本课素材资源库",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
this.skeletonNumber = 0
|
||||||
|
this.pictureurl.push(urls)
|
||||||
|
this.buttonStates.push(buttonState)
|
||||||
|
this.pictureLoading = false;
|
||||||
|
|
||||||
|
} catch (error) {
|
||||||
|
this.pictureLoading = false;
|
||||||
|
console.error("Error fetching data:", error);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
//生图队列的查询
|
||||||
|
hasId(obj, id) {
|
||||||
|
if (typeof obj !== "object" || obj === null) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (Object.values(obj).includes(id)) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return Object.values(obj).some((value) => this.hasId(value, id));
|
||||||
|
},
|
||||||
|
|
||||||
|
setRatio(ratio) {
|
||||||
|
this.form.ratio = ratio;
|
||||||
|
switch (ratio) {
|
||||||
|
case "512":
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height = 512;
|
||||||
|
break;
|
||||||
|
case "768":
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height = 768;
|
||||||
|
break;
|
||||||
|
case "1024":
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height = 1024;
|
||||||
|
break;
|
||||||
|
case "1280":
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height = 1280;
|
||||||
|
break;
|
||||||
|
case "2048":
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height = 2048;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height = 512;
|
||||||
|
}
|
||||||
|
},
|
||||||
|
setRatio2(ratio) {
|
||||||
|
this.form.ratios2 = ratio;
|
||||||
|
},
|
||||||
|
changeSize() {
|
||||||
|
const a = this.form.ratios2
|
||||||
|
if (a == "1/1") {
|
||||||
|
this.formData.prompt[14].inputs.width = this.formData.prompt[14].inputs.height
|
||||||
|
} else if (a == "4/3") {
|
||||||
|
this.formData.prompt[14].inputs.height = (this.formData.prompt[14].inputs.width) * 3 / 4
|
||||||
|
} else if (a == "3/4") {
|
||||||
|
this.formData.prompt[14].inputs.width = (this.formData.prompt[14].inputs.height) * 3 / 4
|
||||||
|
} else if (a == "9/16") {
|
||||||
|
this.formData.prompt[14].inputs.width = (this.formData.prompt[14].inputs.height) * 9 / 16
|
||||||
|
} else {
|
||||||
|
this.formData.prompt[14].inputs.height = (this.formData.prompt[14].inputs.width) * 9 / 16
|
||||||
|
}
|
||||||
|
},
|
||||||
|
changePicture(num) {
|
||||||
|
this.num_picture = num;
|
||||||
|
},
|
||||||
|
|
||||||
|
// 生成提示词模板
|
||||||
|
async createPrompt() {
|
||||||
|
const dataset_id = this.dataset_id
|
||||||
|
const courseName = this.courseName
|
||||||
|
const prompt = `结合${courseName}，给我三句详细的提示词用于生成图片，以1.2.3的形式`
|
||||||
|
const response = await chattoprompt(dataset_id, prompt)
|
||||||
|
const resultData = response.data.answer
|
||||||
|
const promptData = []
|
||||||
|
|
||||||
|
for (let i = 1; i <= 3; i++) {
|
||||||
|
const startIndex = resultData.indexOf(`${i}.`) + `${i}.`.length;
|
||||||
|
const endIndex = resultData.indexOf("。", startIndex);
|
||||||
|
promptData.push(resultData.substring(startIndex, endIndex).trim());
|
||||||
|
}
|
||||||
|
this.resultData = promptData
|
||||||
|
},
|
||||||
|
|
||||||
|
handleCardClick(item) {
|
||||||
|
this.promptData = item
|
||||||
|
this.fetchData()
|
||||||
|
},
|
||||||
|
|
||||||
|
|
||||||
|
//保存图片到素材库
|
||||||
|
async saveImage(resultIndex, index, url, resultItem) {
|
||||||
|
this.buttonStates[resultIndex][index].disabled = true;
|
||||||
|
this.buttonStates[resultIndex][index].text = "正在保存...";
|
||||||
|
const numberIndex = url.indexOf('filename=');
|
||||||
|
const path = url.substring(numberIndex + 9);
|
||||||
|
const pngIndex = path.indexOf('.png');
|
||||||
|
const finalPath = path.substring(0, pngIndex + 4);
|
||||||
|
try {
|
||||||
|
const blob = await this.getImageBlob(`https://ai.ysaix.com:7853/view?filename=${finalPath}&type=temp`);
|
||||||
|
|
||||||
|
const hash = CryptoJS.MD5(blob).toString();
|
||||||
|
|
||||||
|
const formData = new FormData();
|
||||||
|
|
||||||
|
let file = new File([blob], `${resultItem}.png`, {
|
||||||
|
type: 'image/png'
|
||||||
|
})
|
||||||
|
|
||||||
|
// 添加参数
|
||||||
|
formData.append('md5', hash);
|
||||||
|
formData.append('file', file);
|
||||||
|
formData.append('textbookId', this.textbookId);
|
||||||
|
formData.append('levelFirstId', this.levelFirstId);
|
||||||
|
formData.append('levelSecondId', this.levelSecondId);
|
||||||
|
formData.append('fileSource', '个人');
|
||||||
|
formData.append('fileRoot', '备课');
|
||||||
|
formData.append('fileShowName', `${resultItem}.png`);
|
||||||
|
formData.append('fileFlag', '素材');
|
||||||
|
const responsedata = uploadPicture(formData);
|
||||||
|
responsedata.then((response) => {
|
||||||
|
if (response.data && response.data.code === 200) {
|
||||||
|
// 若上传成功，更新对应按钮的状态
|
||||||
|
this.buttonStates[resultIndex][index].text = "已保存";
|
||||||
|
this.buttonStates[resultIndex][index].disabled = true;
|
||||||
|
}
|
||||||
|
}).catch((error) => {
|
||||||
|
console.error(error);
|
||||||
|
this.buttonStates[resultIndex][index].disabled = false;
|
||||||
|
this.buttonStates[resultIndex][index].text = "插入本课素材资源库";
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.error(error);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
getImageBlob(url) {
|
||||||
|
return new Promise((resolve, reject) => {
|
||||||
|
const xhr = new XMLHttpRequest();
|
||||||
|
xhr.open('GET', url, true);
|
||||||
|
xhr.responseType = 'blob';
|
||||||
|
xhr.onload = function () {
|
||||||
|
if (this.status === 200) {
|
||||||
|
resolve(this.response);
|
||||||
|
} else {
|
||||||
|
reject(new Error(`图片获取失败，状态码：${this.status}`));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
xhr.onerror = function () {
|
||||||
|
reject(new Error('图片获取发生网络错误'));
|
||||||
|
};
|
||||||
|
xhr.send();
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
// convertImageToBase64(url) {
|
||||||
|
// return request({
|
||||||
|
// url: '/common/convertImageToBase64',
|
||||||
|
// method: 'get',
|
||||||
|
// params: {
|
||||||
|
// url: url
|
||||||
|
// }
|
||||||
|
// })
|
||||||
|
// },
|
||||||
|
// handleSaveImage(url) {
|
||||||
|
// this.convertImageToBase64(url).then(res => {
|
||||||
|
// this.$emit("saveImage", res)
|
||||||
|
// }, err => {
|
||||||
|
// this.$message.error(err);
|
||||||
|
// })
|
||||||
|
// },
|
||||||
|
|
||||||
|
//prompt生成器的勾选内容添加到输入框
|
||||||
|
addCheckListToPrompt() {
|
||||||
|
const joinedString = this.checkList.join(',');
|
||||||
|
this.promptData = ''
|
||||||
|
this.promptData += joinedString;
|
||||||
|
},
|
||||||
|
|
||||||
|
//清空prompt
|
||||||
|
deletePrompt() {
|
||||||
|
this.promptData = '',
|
||||||
|
this.checkList = []
|
||||||
|
},
|
||||||
|
},
|
||||||
|
mounted() {
|
||||||
|
const route = useRoute();
|
||||||
|
this.dataset_id = route.query.datasetId;
|
||||||
|
this.courseName = route.query.coursetitle;
|
||||||
|
this.levelFirstId = route.query.levelFirstId;
|
||||||
|
this.levelSecondId = route.query.levelSecondId;
|
||||||
|
this.textbookId = route.query.textbookId;
|
||||||
|
this.createPrompt()
|
||||||
|
this.Randomseed();
|
||||||
|
this.divHeight = window.innerHeight * 0.80;
|
||||||
|
window.addEventListener('resize', () => {
|
||||||
|
this.divHeight = window.innerHeight * 0.85;
|
||||||
|
});
|
||||||
|
},
|
||||||
|
beforeDestroy() {
|
||||||
|
window.removeEventListener('resize', () => {
|
||||||
|
this.divHeight = window.innerHeight * 0.85;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
};
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<style scoped>
|
||||||
|
.el-row {
|
||||||
|
padding: 20px
|
||||||
|
}
|
||||||
|
|
||||||
|
.count-option,
|
||||||
|
.ratio-option {
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
.custom-skeleton-item {
|
||||||
|
width: 100%;
|
||||||
|
height: 200px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.time-display p {
|
||||||
|
font-size: 9px;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
.equal-size-image img {
|
||||||
|
position: absolute;
|
||||||
|
top: 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.card-hover:hover {
|
||||||
|
background-color: #f0f0f0;
|
||||||
|
cursor: pointer;
|
||||||
|
}
|
||||||
|
|
||||||
|
.disabled-cursor {
|
||||||
|
cursor: not-allowed !important;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
|
@ -85,6 +85,12 @@ export const constantRoutes = [
|
||||||
name: 'questionUpload',
|
name: 'questionUpload',
|
||||||
meta: { title: '习题上传' }
|
meta: { title: '习题上传' }
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
path: 'aiKolors',
|
||||||
|
component: () => import('@/components/ai-kolors/index.vue'),
|
||||||
|
name: 'aiKolors',
|
||||||
|
meta: { title: '文生图片' }
|
||||||
|
},
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -395,5 +395,6 @@ export const dataSetJson = {
|
||||||
"课标-高中-英语": "e889fcac9fd011efb22a0242ac140006",
|
"课标-高中-英语": "e889fcac9fd011efb22a0242ac140006",
|
||||||
"课标-高中-数学": "e03aa4fe9fd011ef91270242ac140006",
|
"课标-高中-数学": "e03aa4fe9fd011ef91270242ac140006",
|
||||||
"课标-高中-地理": "270516829fd111efb13c0242ac140006",
|
"课标-高中-地理": "270516829fd111efb13c0242ac140006",
|
||||||
|
"课标-高中-政治": "a7df2b01aafd11ef8bb40242ac140002",
|
||||||
"鉴权": "ragflow-IwMDI1MGU2YTU3NjExZWZiNWEzMDI0Mm"
|
"鉴权": "ragflow-IwMDI1MGU2YTU3NjExZWZiNWEzMDI0Mm"
|
||||||
}
|
}
|
||||||
|
|
@ -13,6 +13,7 @@
|
||||||
<el-button type="success" @click="openPPTist">打开PPTist</el-button>
|
<el-button type="success" @click="openPPTist">打开PPTist</el-button>
|
||||||
<el-button type="info" @click="onchange('/model/examination')">考试分析</el-button>
|
<el-button type="info" @click="onchange('/model/examination')">考试分析</el-button>
|
||||||
<el-button type="primary" v-menus="dt.menus">测试</el-button>
|
<el-button type="primary" v-menus="dt.menus">测试</el-button>
|
||||||
|
<el-button type="success" @click="onchange('/model/aiKolors')">文生图片</el-button>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
|
|
@ -54,6 +55,8 @@ import msgUtils from '@/plugins/modal' // 消息工具
|
||||||
import { createWindow, sessionStore } from '@/utils/tool' // 相关工具
|
import { createWindow, sessionStore } from '@/utils/tool' // 相关工具
|
||||||
import * as API_entpcourse from '@/api/education/entpcourse' // 相关api
|
import * as API_entpcourse from '@/api/education/entpcourse' // 相关api
|
||||||
import * as API_entpcoursefile from '@/api/education/entpcoursefile' // 相关api
|
import * as API_entpcoursefile from '@/api/education/entpcoursefile' // 相关api
|
||||||
|
import { dataSetJson } from '@/utils/comm' // 数据集id文生图
|
||||||
|
import { sessionStore } from '@/utils/store' // 学科名字文生图
|
||||||
// 组件引入
|
// 组件引入
|
||||||
import ChooseTextbook from '@/components/choose-textbook/index.vue'
|
import ChooseTextbook from '@/components/choose-textbook/index.vue'
|
||||||
import { menusEvent } from '@/plugins/vue3-menus' // 右键菜单
|
import { menusEvent } from '@/plugins/vue3-menus' // 右键菜单
|
||||||
|
|
@ -156,6 +159,21 @@ const onchange = (path) => {
|
||||||
if (path == '/model/newClassTaskAssign') {
|
if (path == '/model/newClassTaskAssign') {
|
||||||
// 作业管理
|
// 作业管理
|
||||||
router.push({ path, query: { courseObj: JSON.stringify(courseObj) } })
|
router.push({ path, query: { courseObj: JSON.stringify(courseObj) } })
|
||||||
|
} else if (path == '/model/aiKolors') {
|
||||||
|
// ai生图
|
||||||
|
let subjectdata = sessionStore.get('subject.curNode')
|
||||||
|
let datasubject = `课标-${subjectdata.edustage}-${subjectdata.edusubject}`
|
||||||
|
console.log(subjectdata)
|
||||||
|
router.push({
|
||||||
|
path,
|
||||||
|
query: {
|
||||||
|
datasetId: dataSetJson[datasubject],
|
||||||
|
coursetitle: courseObj.coursetitle,
|
||||||
|
levelFirstId: subjectdata.parentid,
|
||||||
|
levelSecondId: subjectdata.id,
|
||||||
|
textbookId: subjectdata.rootid,
|
||||||
|
}
|
||||||
|
});
|
||||||
} else {
|
} else {
|
||||||
router.push(path)
|
router.push(path)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue