mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-12-26 10:05:57 +08:00
641 lines
21 KiB
Vue
641 lines
21 KiB
Vue
<template>
|
||
<div>
|
||
<div class="page-dall">
|
||
<div class="inner custom-scroll">
|
||
<div class="sd-box">
|
||
<h2>DALL-E 创作中心</h2>
|
||
|
||
<div class="sd-params">
|
||
<el-form :model="params" label-width="80px" label-position="left">
|
||
<div class="param-line pt-1">
|
||
<el-form-item label="生图模型">
|
||
<template #default>
|
||
<div class="form-item-inner">
|
||
<el-select v-model="selectedModel" style="width: 150px" placeholder="请选择模型" @change="changeModel">
|
||
<el-option v-for="v in models" :label="v.name" :value="v" :key="v.value" />
|
||
</el-select>
|
||
</div>
|
||
</template>
|
||
</el-form-item>
|
||
</div>
|
||
|
||
<div class="param-line">
|
||
<el-form-item label="图片质量">
|
||
<template #default>
|
||
<div class="form-item-inner">
|
||
<el-select v-model="params.quality" style="width: 150px">
|
||
<el-option v-for="v in qualities" :label="v.name" :value="v.value" :key="v.value" />
|
||
</el-select>
|
||
</div>
|
||
</template>
|
||
</el-form-item>
|
||
</div>
|
||
|
||
<div class="param-line">
|
||
<el-form-item label="图片尺寸">
|
||
<template #default>
|
||
<div class="form-item-inner">
|
||
<el-select v-model="params.size" style="width: 150px">
|
||
<el-option v-for="v in sizes" :label="v" :value="v" :key="v" />
|
||
</el-select>
|
||
</div>
|
||
</template>
|
||
</el-form-item>
|
||
</div>
|
||
|
||
<div class="param-line">
|
||
<el-form-item label="图片样式">
|
||
<template #default>
|
||
<div class="form-item-inner">
|
||
<el-select v-model="params.style" style="width: 150px">
|
||
<el-option v-for="v in styles" :label="v.name" :value="v.value" :key="v.value" />
|
||
</el-select>
|
||
<el-tooltip content="生动使模型倾向于生成超真实和戏剧性的图像" raw-content placement="right">
|
||
<el-icon class="info-icon">
|
||
<InfoFilled />
|
||
</el-icon>
|
||
</el-tooltip>
|
||
</div>
|
||
</template>
|
||
</el-form-item>
|
||
</div>
|
||
|
||
<div class="param-line">
|
||
<el-input
|
||
v-model="params.prompt"
|
||
:autosize="{ minRows: 4, maxRows: 6 }"
|
||
type="textarea"
|
||
ref="promptRef"
|
||
maxlength="2000"
|
||
placeholder="请在此输入绘画提示词，您也可以点击下面的提示词助手生成绘画提示词"
|
||
v-loading="isGenerating"
|
||
/>
|
||
</div>
|
||
|
||
<el-row class="text-info">
|
||
<el-button class="generate-btn" size="small" @click="generatePrompt" color="#5865f2" :disabled="isGenerating">
|
||
<i class="iconfont icon-chuangzuo" style="margin-right: 5px"></i>
|
||
<span>生成专业绘画指令</span>
|
||
</el-button>
|
||
</el-row>
|
||
|
||
<div class="text-info">
|
||
<el-row :gutter="10">
|
||
<el-text type="primary"
|
||
>每次绘图消耗 <el-text type="warning">{{ dallPower }}算力，</el-text></el-text
|
||
>
|
||
<el-text type="primary"
|
||
>当前可用
|
||
<el-text type="warning"> {{ power }}算力</el-text>
|
||
</el-text>
|
||
</el-row>
|
||
</div>
|
||
</el-form>
|
||
</div>
|
||
<div class="submit-btn">
|
||
<el-button type="primary" :dark="false" round @click="generate"> 立即生成 </el-button>
|
||
</div>
|
||
</div>
|
||
<div class="task-list-box pl-6 pr-6 pb-4 pt-4 h-dvh">
|
||
<div class="task-list-inner">
|
||
<div class="job-list-box">
|
||
<h2 class="text-xl">任务列表</h2>
|
||
<task-list :list="runningJobs" />
|
||
<template v-if="finishedJobs.length > 0">
|
||
<h2 class="text-xl">创作记录</h2>
|
||
<div class="finish-job-list">
|
||
<div v-if="finishedJobs.length > 0">
|
||
<!-- <v3-waterfall
|
||
id="waterfall"
|
||
:list="finishedJobs"
|
||
srcKey="img_thumb"
|
||
:gap="20"
|
||
:bottomGap="-10"
|
||
:colWidth="colWidth"
|
||
:distanceToScroll="100"
|
||
:isLoading="loading"
|
||
:isOver="isOver"
|
||
@scrollReachBottom="fetchFinishJobs()"
|
||
>
|
||
<template #default="slotProp">
|
||
<div class="job-item">
|
||
<el-image
|
||
v-if="slotProp.item.img_url !== ''"
|
||
@click="previewImg(slotProp.item)"
|
||
:src="slotProp.item['img_thumb']"
|
||
fit="cover"
|
||
loading="lazy"
|
||
>
|
||
<template #placeholder>
|
||
<div class="image-slot">正在加载图片</div>
|
||
</template>
|
||
|
||
<template #error>
|
||
<div class="image-slot">
|
||
<el-icon>
|
||
<Picture />
|
||
</el-icon>
|
||
</div>
|
||
</template>
|
||
</el-image>
|
||
|
||
<el-image v-else-if="slotProp.item.progress === 101">
|
||
<template #error>
|
||
<div class="image-slot">
|
||
<div class="err-msg-container">
|
||
<div class="title">任务失败</div>
|
||
<div class="opt">
|
||
<el-popover title="错误详情" trigger="click" :width="250" :content="slotProp.item['err_msg']" placement="top">
|
||
<template #reference>
|
||
<el-button type="info">详情</el-button>
|
||
</template>
|
||
</el-popover>
|
||
<el-button type="danger" @click="removeImage(slotProp.item)">删除</el-button>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
</el-image>
|
||
|
||
<el-image v-else>
|
||
<template #error>
|
||
<div class="image-slot">
|
||
<i class="iconfont icon-loading"></i>
|
||
<span>正在下载图片</span>
|
||
</div>
|
||
</template>
|
||
</el-image>
|
||
|
||
<div class="remove">
|
||
<el-tooltip content="删除" placement="top">
|
||
<el-button type="danger" :icon="Delete" @click="removeImage(slotProp.item)" circle />
|
||
</el-tooltip>
|
||
<el-tooltip content="取消分享" placement="top" v-if="slotProp.item.publish">
|
||
<el-button type="warning" @click="publishImage(slotProp.item, false)" circle>
|
||
<i class="iconfont icon-cancel-share"></i>
|
||
</el-button>
|
||
</el-tooltip>
|
||
<el-tooltip content="分享" placement="top" v-else>
|
||
<el-button type="success" @click="publishImage(slotProp.item, true)" circle>
|
||
<i class="iconfont icon-share-bold"></i>
|
||
</el-button>
|
||
</el-tooltip>
|
||
|
||
<el-tooltip content="复制提示词" placement="top">
|
||
<el-button type="info" circle class="copy-prompt" :data-clipboard-text="slotProp.item.prompt">
|
||
<i class="iconfont icon-file"></i>
|
||
</el-button>
|
||
</el-tooltip>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
|
||
<template #footer>
|
||
<div class="no-more-data">
|
||
<span>没有更多数据了</span>
|
||
<i class="iconfont icon-face"></i>
|
||
</div>
|
||
</template>
|
||
</v3-waterfall> -->
|
||
<Waterfall
|
||
ref="waterfall"
|
||
:list="finishedJobs"
|
||
:row-key="options.rowKey"
|
||
:gutter="options.gutter"
|
||
:has-around-gutter="options.hasAroundGutter"
|
||
:width="options.width"
|
||
:breakpoints="options.breakpoints"
|
||
:img-selector="options.imgSelector"
|
||
:background-color="options.backgroundColor"
|
||
:animation-effect="options.animationEffect"
|
||
:animation-duration="options.animationDuration"
|
||
:animation-delay="options.animationDelay"
|
||
:animation-cancel="options.animationCancel"
|
||
:lazyload="options.lazyload"
|
||
:load-props="options.loadProps"
|
||
:cross-origin="options.crossOrigin"
|
||
:align="options.align"
|
||
@afterRender="afterRender"
|
||
>
|
||
<template #default="{ item, url, index }">
|
||
<div class="bg-gray-900 rounded-lg shadow-md overflow-hidden transition-all duration-300 ease-linear hover:shadow-lg hover:shadow-gray-600 group" @click="handleClick(item)">
|
||
<div class="overflow-hidden">
|
||
<LazyImg :url="url" title="title" :alt="item.name" class="cursor-pointer transition-all duration-300 ease-linear group-hover:scale-105" @load="imageLoad" @error="imageError" @success="imageSuccess" />
|
||
</div>
|
||
<div class="px-4 pt-2 pb-4 border-t border-t-gray-800">
|
||
<h2 class="pb-4 text-gray-50 group-hover:text-yellow-300">
|
||
{{ item.name }}
|
||
</h2>
|
||
<div class="pt-3 flex justify-between items-center border-t border-t-gray-600 border-opacity-50">
|
||
<div class="text-gray-50">
|
||
$ {{ item.price }}
|
||
</div>
|
||
<div>
|
||
<button class="px-3 h-7 rounded-full bg-red-500 text-sm text-white shadow-lg transition-all duration-300 hover:bg-red-600" @click.stop="handleDelete(item, index)">
|
||
删除
|
||
</button>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
</Waterfall>
|
||
</div>
|
||
<el-empty :image-size="100" :image="nodata" description="暂无记录" v-else />
|
||
|
||
<div v-show="!loading" class="flex justify-center py-10 bg-gray-900">
|
||
<button class="px-5 py-2 rounded-full bg-gray-700 text-md text-white cursor-pointer hover:bg-gray-800 transition-all duration-300" @click="fetchFinishJobs">
|
||
加载更多
|
||
</button>
|
||
</div>
|
||
</div>
|
||
</template>
|
||
|
||
<!-- end finish job list-->
|
||
</div>
|
||
</div>
|
||
<back-top :right="30" :bottom="30" />
|
||
</div>
|
||
<!-- end task list box -->
|
||
</div>
|
||
</div>
|
||
|
||
<el-image-viewer
|
||
@close="
|
||
() => {
|
||
previewURL = '';
|
||
}
|
||
"
|
||
v-if="previewURL !== ''"
|
||
:url-list="[previewURL]"
|
||
/>
|
||
</div>
|
||
</template>
|
||
|
||
<script setup>
|
||
import nodata from "@/assets/img/no-data.png";
|
||
|
||
import { nextTick, onMounted, onUnmounted, ref } from "vue";
|
||
import { Delete, InfoFilled, Picture } from "@element-plus/icons-vue";
|
||
import { httpGet, httpPost } from "@/utils/http";
|
||
import { ElMessage, ElMessageBox } from "element-plus";
|
||
import Clipboard from "clipboard";
|
||
import { checkSession, getSystemInfo } from "@/store/cache";
|
||
import { useSharedStore } from "@/store/sharedata";
|
||
import TaskList from "@/components/TaskList.vue";
|
||
import BackTop from "@/components/BackTop.vue";
|
||
import { showMessageError, showMessageOK } from "@/utils/dialog";
|
||
import BScrollBox from "@/components/ui/BScrollBox.vue";
|
||
import { LazyImg, Waterfall } from 'vue-waterfall-plugin-next'
|
||
import 'vue-waterfall-plugin-next/dist/style.css'
|
||
|
||
import error from '@/assets/img/failed.png'
|
||
|
||
const listBoxHeight = ref(0);
|
||
// const paramBoxHeight = ref(0)
|
||
const isLogin = ref(false);
|
||
const loading = ref(true);
|
||
const colWidth = ref(220);
|
||
const isOver = ref(false);
|
||
const previewURL = ref("");
|
||
const store = useSharedStore();
|
||
const models = ref([]);
|
||
|
||
const resizeElement = function () {
|
||
listBoxHeight.value = window.innerHeight - 58;
|
||
// paramBoxHeight.value = window.innerHeight - 110
|
||
};
|
||
|
||
const options = ref({
|
||
// 唯一key值
|
||
rowKey: 'id',
|
||
// 卡片之间的间隙
|
||
gutter: 10,
|
||
// 是否有周围的gutter
|
||
hasAroundGutter: true,
|
||
// 卡片在PC上的宽度
|
||
width: 200,
|
||
// 自定义行显示个数，主要用于对移动端的适配
|
||
breakpoints: {
|
||
3840: {
|
||
// 4K下
|
||
rowPerView: 8,
|
||
},
|
||
2560: {
|
||
// 2K下
|
||
rowPerView: 7,
|
||
},
|
||
1920: {
|
||
// 2K下
|
||
rowPerView: 6,
|
||
},
|
||
1600: {
|
||
// 2K下
|
||
rowPerView: 5,
|
||
},
|
||
1366: {
|
||
// 2K下
|
||
rowPerView: 4,
|
||
},
|
||
800: {
|
||
// 当屏幕宽度小于等于800
|
||
rowPerView: 3,
|
||
},
|
||
500: {
|
||
// 当屏幕宽度小于等于500
|
||
rowPerView: 2,
|
||
},
|
||
},
|
||
// 动画效果
|
||
animationEffect: 'animate__fadeInUp',
|
||
// 动画时间
|
||
animationDuration: 1000,
|
||
// 动画延迟
|
||
animationDelay: 300,
|
||
animationCancel: false,
|
||
// 背景色
|
||
backgroundColor: '#2C2E3A',
|
||
// imgSelector
|
||
imgSelector: 'img_thumb',
|
||
// 加载配置
|
||
loadProps: {
|
||
loading,
|
||
error,
|
||
ratioCalculator: (width, height) => {
|
||
console.log("width, height", width, height)
|
||
return height / width
|
||
},
|
||
},
|
||
// 是否懒加载
|
||
lazyload: true,
|
||
align: 'center',
|
||
})
|
||
|
||
function imageLoad(url) {
|
||
console.log(`${url}: 加载完成`)
|
||
}
|
||
|
||
function imageError(url) {
|
||
console.error(`${url}: 加载失败`)
|
||
}
|
||
|
||
function imageSuccess(url) {
|
||
console.log(`${url}: 加载成功`)
|
||
}
|
||
|
||
function afterRender() {
|
||
loading.value = false
|
||
console.log('计算完成')
|
||
}
|
||
|
||
resizeElement();
|
||
window.onresize = () => {
|
||
resizeElement();
|
||
};
|
||
const qualities = [
|
||
{ name: "标准", value: "standard" },
|
||
{ name: "高清", value: "hd" },
|
||
];
|
||
const dalleSizes = ["1024x1024", "1792x1024", "1024x1792"];
|
||
const fluxSizes = ["1024x1024", "1024x768", "768x1024", "1280x960", "960x1280", "1366x768", "768x1366"];
|
||
const sizes = ref(dalleSizes);
|
||
const styles = [
|
||
{ name: "生动", value: "vivid" },
|
||
{ name: "自然", value: "natural" },
|
||
];
|
||
const params = ref({
|
||
quality: "standard",
|
||
size: "1024x1024",
|
||
style: "vivid",
|
||
prompt: "",
|
||
});
|
||
|
||
const finishedJobs = ref([]);
|
||
const runningJobs = ref([]);
|
||
const allowPulling = ref(true); // 是否允许轮询
|
||
const tastPullHandler = ref(null);
|
||
const power = ref(0);
|
||
const dallPower = ref(0); // 画一张 SD 图片消耗算力
|
||
const clipboard = ref(null);
|
||
const userId = ref(0);
|
||
const selectedModel = ref(null);
|
||
|
||
onMounted(() => {
|
||
initData();
|
||
clipboard.value = new Clipboard(".copy-prompt");
|
||
clipboard.value.on("success", () => {
|
||
showMessageOK("复制成功！");
|
||
});
|
||
|
||
clipboard.value.on("error", () => {
|
||
showMessageError("复制失败！");
|
||
});
|
||
|
||
getSystemInfo()
|
||
.then((res) => {
|
||
dallPower.value = res.data["dall_power"];
|
||
})
|
||
.catch((e) => {
|
||
showMessageError("获取系统配置失败：" + e.message);
|
||
});
|
||
|
||
// 获取模型列表
|
||
httpGet("/api/dall/models")
|
||
.then((res) => {
|
||
models.value = res.data;
|
||
selectedModel.value = models.value[0];
|
||
params.value.model_id = selectedModel.value.id;
|
||
changeModel(selectedModel.value);
|
||
})
|
||
.catch((e) => {
|
||
showMessageError("获取模型列表失败：" + e.message);
|
||
});
|
||
});
|
||
|
||
onUnmounted(() => {
|
||
clipboard.value.destroy();
|
||
if (tastPullHandler.value) {
|
||
clearInterval(tastPullHandler.value);
|
||
}
|
||
});
|
||
|
||
const initData = () => {
|
||
checkSession()
|
||
.then((user) => {
|
||
power.value = user["power"];
|
||
userId.value = user.id;
|
||
isLogin.value = true;
|
||
page.value = 0;
|
||
fetchRunningJobs();
|
||
fetchFinishJobs();
|
||
|
||
// 轮询运行中任务
|
||
tastPullHandler.value = setInterval(() => {
|
||
if (allowPulling.value) {
|
||
fetchRunningJobs();
|
||
}
|
||
}, 5000);
|
||
})
|
||
.catch(() => {});
|
||
};
|
||
|
||
const fetchRunningJobs = () => {
|
||
if (!isLogin.value) {
|
||
return;
|
||
}
|
||
// 获取运行中的任务
|
||
httpGet(`/api/dall/jobs?finish=false`)
|
||
.then((res) => {
|
||
// 如果任务有更新，则更新已完成任务列表
|
||
if (res.data.items && res.data.items.length !== runningJobs.value.length) {
|
||
page.value = 0;
|
||
fetchFinishJobs();
|
||
}
|
||
if (res.data.items.length > 0) {
|
||
runningJobs.value = res.data.items;
|
||
} else {
|
||
allowPulling.value = false;
|
||
runningJobs.value = [];
|
||
}
|
||
})
|
||
.catch((e) => {
|
||
ElMessage.error("获取任务失败：" + e.message);
|
||
});
|
||
};
|
||
|
||
const page = ref(1);
|
||
const pageSize = ref(15);
|
||
// 获取已完成的任务
|
||
const fetchFinishJobs = () => {
|
||
if (!isLogin.value) {
|
||
return;
|
||
}
|
||
|
||
loading.value = true;
|
||
page.value = page.value + 1;
|
||
|
||
httpGet(`/api/dall/jobs?finish=true&page=${page.value}&page_size=${pageSize.value}`)
|
||
.then((res) => {
|
||
if (res.data.items.length < pageSize.value) {
|
||
isOver.value = true;
|
||
}
|
||
const imageList = res.data.items;
|
||
for (let i = 0; i < imageList.length; i++) {
|
||
imageList[i]["img_thumb"] = imageList[i]["img_url"] + "?imageView2/4/w/300/h/0/q/75";
|
||
}
|
||
if (page.value === 1) {
|
||
finishedJobs.value = imageList;
|
||
} else {
|
||
finishedJobs.value = finishedJobs.value.concat(imageList);
|
||
}
|
||
|
||
loading.value = false;
|
||
})
|
||
.catch((e) => {
|
||
ElMessage.error("获取任务失败：" + e.message);
|
||
});
|
||
};
|
||
|
||
// 创建绘图任务
|
||
const promptRef = ref(null);
|
||
const generate = () => {
|
||
if (params.value.prompt === "") {
|
||
promptRef.value.focus();
|
||
return ElMessage.error("请输入绘画提示词！");
|
||
}
|
||
|
||
if (!isLogin.value) {
|
||
store.setShowLoginDialog(true);
|
||
return;
|
||
}
|
||
httpPost("/api/dall/image", params.value)
|
||
.then(() => {
|
||
ElMessage.success("任务执行成功！");
|
||
power.value -= dallPower.value;
|
||
// 追加任务列表
|
||
runningJobs.value.push({
|
||
prompt: params.value.prompt,
|
||
progress: 0,
|
||
});
|
||
allowPulling.value = true;
|
||
})
|
||
.catch((e) => {
|
||
ElMessage.error("任务执行失败：" + e.message);
|
||
});
|
||
};
|
||
|
||
const removeImage = (item) => {
|
||
ElMessageBox.confirm("此操作将会删除任务和图片，继续操作码?", "删除提示", {
|
||
confirmButtonText: "确认",
|
||
cancelButtonText: "取消",
|
||
type: "warning",
|
||
})
|
||
.then(() => {
|
||
httpGet("/api/dall/remove", { id: item.id })
|
||
.then(() => {
|
||
ElMessage.success("任务删除成功");
|
||
page.value = 0;
|
||
isOver.value = false;
|
||
fetchFinishJobs();
|
||
})
|
||
.catch((e) => {
|
||
ElMessage.error("任务删除失败：" + e.message);
|
||
});
|
||
})
|
||
.catch(() => {});
|
||
};
|
||
|
||
const previewImg = (item) => {
|
||
previewURL.value = item.img_url;
|
||
};
|
||
|
||
// 发布图片到作品墙
|
||
const publishImage = (item, action) => {
|
||
let text = "图片发布";
|
||
if (action === false) {
|
||
text = "取消发布";
|
||
}
|
||
httpGet("/api/dall/publish", { id: item.id, action: action })
|
||
.then(() => {
|
||
ElMessage.success(text + "成功");
|
||
item.publish = action;
|
||
page.value = 0;
|
||
isOver.value = false;
|
||
})
|
||
.catch((e) => {
|
||
ElMessage.error(text + "失败：" + e.message);
|
||
});
|
||
};
|
||
|
||
const isGenerating = ref(false);
|
||
const generatePrompt = () => {
|
||
if (params.value.prompt === "") {
|
||
return showMessageError("请输入原始提示词");
|
||
}
|
||
isGenerating.value = true;
|
||
httpPost("/api/prompt/image", { prompt: params.value.prompt })
|
||
.then((res) => {
|
||
params.value.prompt = res.data;
|
||
isGenerating.value = false;
|
||
})
|
||
.catch((e) => {
|
||
showMessageError("生成提示词失败：" + e.message);
|
||
isGenerating.value = false;
|
||
});
|
||
};
|
||
|
||
const changeModel = (model) => {
|
||
if (model.value.startsWith("dall")) {
|
||
sizes.value = dalleSizes;
|
||
} else {
|
||
sizes.value = fluxSizes;
|
||
}
|
||
params.value.model_id = selectedModel.value.id;
|
||
};
|
||
</script>
|
||
|
||
<style lang="stylus">
|
||
@import "@/assets/css/image-dall.styl"
|
||
@import "@/assets/css/custom-scroll.styl"
|
||
</style>
|