更新移动端 Dalle 绘图页面,支持模型选择

This commit is contained in:
RockYang
2024-12-26 18:50:45 +08:00
parent 8af0fec8ec
commit 18179613fc
9 changed files with 1215 additions and 1306 deletions

View File

@@ -12,69 +12,47 @@
</template>
<template #left>
<span class="setting">
<van-icon name="setting-o" @click="showPicker = true"/>
<van-icon name="add-o" @click="showPicker = true" />
</span>
</template>
<template #right>
<van-icon name="share-o" @click="showShare = true"/>
<van-icon name="share-o" @click="showShare = true" />
</template>
</van-nav-bar>
<van-share-sheet
v-model:show="showShare"
title="立即分享给好友"
:options="shareOptions"
@select="shareChat"
/>
<van-share-sheet v-model:show="showShare" title="立即分享给好友" :options="shareOptions" @select="shareChat" />
<div class="chat-list-wrapper">
<div id="message-list-box" :style="{height: winHeight + 'px'}" class="message-list-box">
<van-list
v-model:error="error"
:finished="finished"
error-text="请求失败点击重新加载"
@load="onLoad"
>
<div id="message-list-box" :style="{ height: winHeight + 'px' }" class="message-list-box">
<van-list v-model:error="error" :finished="finished" error-text="请求失败点击重新加载" @load="onLoad">
<van-cell v-for="item in chatData" :key="item" :border="false" class="message-line">
<chat-prompt
v-if="item.type==='prompt'"
:content="item.content"
:icon="item.icon"/>
<chat-reply v-else-if="item.type==='reply'"
:content="item.content"
:icon="item.icon"
:org-content="item.orgContent"/>
<chat-prompt v-if="item.type === 'prompt'" :content="item.content" :icon="item.icon" />
<chat-reply v-else-if="item.type === 'reply'" :content="item.content" :icon="item.icon" :org-content="item.orgContent" />
</van-cell>
</van-list>
</div>
</div>
<div class="chat-box-wrapper">
<van-sticky ref="bottomBarRef" :offset-bottom="0" position="bottom">
<van-cell-group inset>
<van-field
v-model="prompt"
center
clearable
placeholder="输入你的问题"
>
<!-- <template #left-icon>-->
<!-- <van-button round type="success" class="button-voice" @click="inputVoice">-->
<!-- <el-icon>-->
<!-- <Microphone/>-->
<!-- </el-icon>-->
<!-- </van-button>-->
<!-- </template>-->
<van-cell-group inset style="--van-cell-background: var(--van-cell-background-light)">
<!-- <div class="flex flex-row p-2">
<file-list :files="[{ url: 'https://www.baidu.com/img/PCtm_d9c8750bed0b3c7d089fa7d55720d6cf.png', name: 'test.png', ext: 'png', size: 1024 }]" />
</div> -->
<van-field v-model="prompt" center clearable placeholder="输入你的问题">
<template #left-icon>
<!-- <div class="flex flex-row">
<span class="rounded-full size-6 flex items-center justify-center"><i class="iconfont icon-attachment-cl"></i></span>
</div> -->
</template>
<template #button>
<van-button size="small" type="primary" @click="sendMessage">发送</van-button>
</template>
<template #extra>
<div class="icon-box">
<van-icon v-if="showStopGenerate" name="stop-circle-o" @click="stopGenerate"/>
<van-icon v-if="showReGenerate" name="play-circle-o" @click="reGenerate"/>
<van-icon v-if="showStopGenerate" name="stop-circle-o" @click="stopGenerate" />
<van-icon v-if="showReGenerate" name="play-circle-o" @click="reGenerate" />
</div>
</template>
</van-field>
@@ -83,7 +61,7 @@
</div>
</div>
<button id="copy-link-btn" style="display: none;" :data-clipboard-text="url">复制链接地址</button>
<button id="copy-link-btn" style="display: none" :data-clipboard-text="url">复制链接地址</button>
<!-- <van-overlay :show="showMic" z-index="100">-->
<!-- <div class="mic-wrapper">-->
@@ -100,21 +78,10 @@
</div>
<van-popup v-model:show="showPicker" position="bottom" class="popup">
<van-picker
:columns="columns"
v-model="selectedValues"
title="选择模型和角色"
@cancel="showPicker = false"
@confirm="newChat"
>
<van-picker :columns="columns" v-model="selectedValues" title="选择模型和角色" @cancel="showPicker = false" @confirm="newChat">
<template #option="item">
<div class="picker-option">
<van-image
v-if="item.icon"
:src="item.icon"
fit="cover"
round
/>
<van-image v-if="item.icon" :src="item.icon" fit="cover" round />
<span>{{ item.text }}</span>
</div>
</template>
@@ -123,148 +90,162 @@
</template>
<script setup>
import {nextTick, onMounted, onUnmounted, ref, watch} from "vue";
import {showImagePreview, showNotify, showToast} from "vant";
import {useRouter} from "vue-router";
import {processContent, randString, renderInputText, UUID} from "@/utils/libs";
import {httpGet} from "@/utils/http";
import { nextTick, onMounted, onUnmounted, ref, watch } from "vue";
import { showImagePreview, showNotify, showToast } from "vant";
import { useRouter } from "vue-router";
import { processContent, randString, renderInputText, UUID } from "@/utils/libs";
import { httpGet } from "@/utils/http";
import hl from "highlight.js";
import 'highlight.js/styles/a11y-dark.css'
import "highlight.js/styles/a11y-dark.css";
import ChatPrompt from "@/components/mobile/ChatPrompt.vue";
import ChatReply from "@/components/mobile/ChatReply.vue";
import {checkSession, getClientId} from "@/store/cache";
import { checkSession, getClientId } from "@/store/cache";
import Clipboard from "clipboard";
import { showMessageError} from "@/utils/dialog";
import {useSharedStore} from "@/store/sharedata";
import { showMessageError } from "@/utils/dialog";
import { useSharedStore } from "@/store/sharedata";
import emoji from "markdown-it-emoji";
import mathjaxPlugin from "markdown-it-mathjax3";
import MarkdownIt from "markdown-it";
import FileList from "@/components/FileList.vue";
const winHeight = ref(0);
const navBarRef = ref(null);
const bottomBarRef = ref(null);
const router = useRouter();
const winHeight = ref(0)
const navBarRef = ref(null)
const bottomBarRef = ref(null)
const router = useRouter()
const roles = ref([])
const roleId = ref(parseInt(router.currentRoute.value.query["role_id"]))
const role = ref({})
const models = ref([])
const modelId = ref(parseInt(router.currentRoute.value.query["model_id"]))
const modelValue = ref("")
const title = ref(router.currentRoute.value.query["title"])
const chatId = ref(router.currentRoute.value.query["chat_id"])
const loginUser = ref(null)
const roles = ref([]);
const roleId = ref(parseInt(router.currentRoute.value.query["role_id"]));
const role = ref({});
const models = ref([]);
const modelId = ref(parseInt(router.currentRoute.value.query["model_id"]));
const modelValue = ref("");
const title = ref(router.currentRoute.value.query["title"]);
const chatId = ref(router.currentRoute.value.query["chat_id"]);
const loginUser = ref(null);
// const showMic = ref(false)
const showPicker = ref(false)
const columns = ref([roles.value, models.value])
const selectedValues = ref([roleId.value, modelId.value])
const showPicker = ref(false);
const columns = ref([roles.value, models.value]);
const selectedValues = ref([roleId.value, modelId.value]);
checkSession().then(user => {
loginUser.value = user
}).catch(() => {
router.push('/login')
})
checkSession()
.then((user) => {
loginUser.value = user;
})
.catch(() => {
router.push("/login");
});
const loadModels = () => {
// 加载模型
httpGet('/api/model/list').then(res => {
models.value = res.data
if (!modelId.value) {
modelId.value = models.value[0].id
}
for (let i = 0; i < models.value.length; i++) {
models.value[i].text = models.value[i].name
models.value[i].mValue = models.value[i].value
models.value[i].value = models.value[i].id
}
modelValue.value = getModelName(modelId.value)
// 加载角色列表
httpGet(`/api/app/list/user`,{id: roleId.value}).then((res) => {
roles.value = res.data;
if (!roleId.value) {
roleId.value = roles.value[0]['id']
httpGet("/api/model/list")
.then((res) => {
models.value = res.data;
if (!modelId.value) {
modelId.value = models.value[0].id;
}
// build data for role picker
for (let i = 0; i < roles.value.length; i++) {
roles.value[i].text = roles.value[i].name
roles.value[i].value = roles.value[i].id
roles.value[i].helloMsg = roles.value[i].hello_msg
for (let i = 0; i < models.value.length; i++) {
models.value[i].text = models.value[i].name;
models.value[i].mValue = models.value[i].value;
models.value[i].value = models.value[i].id;
}
role.value = getRoleById(roleId.value)
columns.value = [roles.value, models.value]
selectedValues.value = [roleId.value, modelId.value]
loadChatHistory()
}).catch((e) => {
showNotify({type: "danger", message: '获取聊天角色失败: ' + e.messages})
modelValue.value = getModelName(modelId.value);
// 加载角色列表
httpGet(`/api/app/list/user`, { id: roleId.value })
.then((res) => {
roles.value = res.data;
if (!roleId.value) {
roleId.value = roles.value[0]["id"];
}
// build data for role picker
for (let i = 0; i < roles.value.length; i++) {
roles.value[i].text = roles.value[i].name;
roles.value[i].value = roles.value[i].id;
roles.value[i].helloMsg = roles.value[i].hello_msg;
}
role.value = getRoleById(roleId.value);
columns.value = [roles.value, models.value];
selectedValues.value = [roleId.value, modelId.value];
loadChatHistory();
})
.catch((e) => {
showNotify({ type: "danger", message: "获取聊天角色失败: " + e.messages });
});
})
}).catch(e => {
showNotify({type: "danger", message: "加载模型失败: " + e.message})
})
}
.catch((e) => {
showNotify({ type: "danger", message: "加载模型失败: " + e.message });
});
};
if (chatId.value) {
httpGet(`/api/chat/detail?chat_id=${chatId.value}`).then(res => {
title.value = res.data.title
modelId.value = res.data.model_id
roleId.value = res.data.role_id
loadModels()
}).catch(() => {
loadModels()
})
httpGet(`/api/chat/detail?chat_id=${chatId.value}`)
.then((res) => {
title.value = res.data.title;
modelId.value = res.data.model_id;
roleId.value = res.data.role_id;
loadModels();
})
.catch(() => {
loadModels();
});
} else {
title.value = "新建对话"
chatId.value = UUID()
loadModels()
title.value = "新建对话";
chatId.value = UUID();
loadModels();
}
const chatData = ref([])
const loading = ref(false)
const finished = ref(false)
const error = ref(false)
const store = useSharedStore()
const url = ref(location.protocol + '//' + location.host + '/mobile/chat/export?chat_id=' + chatId.value)
const mathjaxPlugin = require('markdown-it-mathjax3')
const md = require('markdown-it')({
const chatData = ref([]);
const loading = ref(false);
const finished = ref(false);
const error = ref(false);
const store = useSharedStore();
const url = ref(location.protocol + "//" + location.host + "/mobile/chat/export?chat_id=" + chatId.value);
const md = new MarkdownIt({
breaks: true,
html: true,
linkify: true,
typographer: true,
highlight: function (str, lang) {
const codeIndex = parseInt(Date.now()) + Math.floor(Math.random() * 10000000)
const codeIndex = parseInt(Date.now()) + Math.floor(Math.random() * 10000000);
// 显示复制代码按钮
const copyBtn = `<span class="copy-code-mobile" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span>
<textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace(/<\/textarea>/g, '&lt;/textarea>')}</textarea>`
<textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace(
/<\/textarea>/g,
"&lt;/textarea>"
)}</textarea>`;
if (lang && hl.getLanguage(lang)) {
const langHtml = `<span class="lang-name">${lang}</span>`
const langHtml = `<span class="lang-name">${lang}</span>`;
// 处理代码高亮
const preCode = hl.highlight(lang, str, true).value
const preCode = hl.highlight(lang, str, true).value;
// 将代码包裹在 pre 中
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn} ${langHtml}</pre>`
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn} ${langHtml}</pre>`;
}
// 处理代码高亮
const preCode = md.utils.escapeHtml(str)
const preCode = md.utils.escapeHtml(str);
// 将代码包裹在 pre 中
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`
}
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`;
},
});
md.use(mathjaxPlugin)
md.use(mathjaxPlugin);
md.use(emoji);
onMounted(() => {
winHeight.value = window.innerHeight - navBarRef.value.$el.offsetHeight - bottomBarRef.value.$el.offsetHeight - 70
winHeight.value = window.innerHeight - navBarRef.value.$el.offsetHeight - bottomBarRef.value.$el.offsetHeight - 70;
const clipboard = new Clipboard(".content-mobile,.copy-code-mobile,#copy-link-btn");
clipboard.on('success', (e) => {
e.clearSelection()
showNotify({type: 'success', message: '复制成功', duration: 1000})
})
clipboard.on('error', () => {
showNotify({type: 'danger', message: '复制失败', duration: 2000})
})
clipboard.on("success", (e) => {
e.clearSelection();
showNotify({ type: "success", message: "复制成功", duration: 1000 });
});
clipboard.on("error", () => {
showNotify({ type: "danger", message: "复制失败", duration: 2000 });
});
store.addMessageHandler("chat",(data) => {
store.addMessageHandler("chat", (data) => {
if (data.channel !== "chat" || data.clientId !== getClientId()) {
return
return;
}
if (data.type === 'error') {
showMessageError(data.body)
return
if (data.type === "error") {
showMessageError(data.body);
return;
}
if (isNewMsg.value) {
@@ -272,127 +253,132 @@ onMounted(() => {
type: "reply",
id: randString(32),
icon: role.value.icon,
content: data.body
content: data.body,
});
if (!title.value) {
title.value = previousText.value
title.value = previousText.value;
}
lineBuffer.value = data.body;
isNewMsg.value = false
} else if (data.type === 'end') { // 消息接收完毕
enableInput()
lineBuffer.value = ''; // 清空缓冲
isNewMsg.value = true
isNewMsg.value = false;
} else if (data.type === "end") {
// 消息接收完毕
enableInput();
lineBuffer.value = ""; // 清空缓冲
isNewMsg.value = true;
} else {
lineBuffer.value += data.body;
const reply = chatData.value[chatData.value.length - 1]
reply['orgContent'] = lineBuffer.value;
reply['content'] = md.render(processContent(lineBuffer.value));
const reply = chatData.value[chatData.value.length - 1];
reply["orgContent"] = lineBuffer.value;
reply["content"] = md.render(processContent(lineBuffer.value));
nextTick(() => {
hl.configure({ignoreUnescapedHTML: true})
const lines = document.querySelectorAll('.message-line');
const blocks = lines[lines.length - 1].querySelectorAll('pre code');
hl.configure({ ignoreUnescapedHTML: true });
const lines = document.querySelectorAll(".message-line");
const blocks = lines[lines.length - 1].querySelectorAll("pre code");
blocks.forEach((block) => {
hl.highlightElement(block)
})
scrollListBox()
hl.highlightElement(block);
});
scrollListBox();
const items = document.querySelectorAll('.message-line')
const imgs = items[items.length - 1].querySelectorAll('img')
const items = document.querySelectorAll(".message-line");
const imgs = items[items.length - 1].querySelectorAll("img");
for (let i = 0; i < imgs.length; i++) {
if (!imgs[i].src) {
continue
continue;
}
imgs[i].addEventListener('click', (e) => {
e.stopPropagation()
imgs[i].addEventListener("click", (e) => {
e.stopPropagation();
showImagePreview([imgs[i].src]);
})
});
}
})
});
}
})
})
});
});
onUnmounted(() => {
store.removeMessageHandler("chat")
})
store.removeMessageHandler("chat");
});
const newChat = (item) => {
showPicker.value = false
const options = item.selectedOptions
roleId.value = options[0].value
modelId.value = options[1].value
modelValue.value = getModelName(modelId.value)
chatId.value = UUID()
chatData.value = []
role.value = getRoleById(roleId.value)
title.value = "新建对话"
loadChatHistory()
}
showPicker.value = false;
const options = item.selectedOptions;
roleId.value = options[0].value;
modelId.value = options[1].value;
modelValue.value = getModelName(modelId.value);
chatId.value = UUID();
chatData.value = [];
role.value = getRoleById(roleId.value);
title.value = "新建对话";
loadChatHistory();
};
const onLoad = () => {
// checkSession().then(() => {
// connect()
// }).catch(() => {
// })
}
};
const loadChatHistory = () => {
httpGet('/api/chat/history?chat_id=' + chatId.value).then(res => {
const role = getRoleById(roleId.value)
// 加载状态结束
finished.value = true;
const data = res.data
if (data.length === 0) {
chatData.value.push({
type: "reply",
id: randString(32),
icon: role.icon,
content: role.hello_msg,
orgContent: role.hello_msg,
})
return
}
for (let i = 0; i < data.length; i++) {
if (data[i].type === "prompt") {
chatData.value.push(data[i]);
continue;
httpGet("/api/chat/history?chat_id=" + chatId.value)
.then((res) => {
const role = getRoleById(roleId.value);
// 加载状态结束
finished.value = true;
const data = res.data;
if (data.length === 0) {
chatData.value.push({
type: "reply",
id: randString(32),
icon: role.icon,
content: role.hello_msg,
orgContent: role.hello_msg,
});
return;
}
data[i].orgContent = data[i].content;
data[i].content = md.render(processContent(data[i].content))
chatData.value.push(data[i]);
}
for (let i = 0; i < data.length; i++) {
if (data[i].type === "prompt") {
chatData.value.push(data[i]);
continue;
}
nextTick(() => {
hl.configure({ignoreUnescapedHTML: true})
const blocks = document.querySelector("#message-list-box").querySelectorAll('pre code');
blocks.forEach((block) => {
hl.highlightElement(block)
})
data[i].orgContent = data[i].content;
data[i].content = md.render(processContent(data[i].content));
chatData.value.push(data[i]);
}
scrollListBox()
nextTick(() => {
hl.configure({ ignoreUnescapedHTML: true });
const blocks = document.querySelector("#message-list-box").querySelectorAll("pre code");
blocks.forEach((block) => {
hl.highlightElement(block);
});
scrollListBox();
});
})
}).catch(() => {
error.value = true
})
}
.catch(() => {
error.value = true;
});
};
// 创建 socket 连接
const prompt = ref('');
const prompt = ref("");
const showStopGenerate = ref(false); // 停止生成
const showReGenerate = ref(false); // 重新生成
const previousText = ref(''); // 上一次提问
const lineBuffer = ref(''); // 输出缓冲行
const canSend = ref(true)
const isNewMsg = ref(true)
const stream = ref(store.chatStream)
watch(() => store.chatStream, (newValue) => {
stream.value = newValue
});
const previousText = ref(""); // 上一次提问
const lineBuffer = ref(""); // 输出缓冲行
const canSend = ref(true);
const isNewMsg = ref(true);
const stream = ref(store.chatStream);
watch(
() => store.chatStream,
(newValue) => {
stream.value = newValue;
}
);
// const connect = function () {
// // 初始化 WebSocket 对象
// const _sessionId = getSessionId();
@@ -487,38 +473,36 @@ watch(() => store.chatStream, (newValue) => {
// socket.value = _socket;
// }
const disableInput = (force) => {
canSend.value = false;
showReGenerate.value = false;
showStopGenerate.value = !force;
}
};
const enableInput = () => {
canSend.value = true;
showReGenerate.value = previousText.value !== "";
showStopGenerate.value = false;
}
};
// 将聊天框的滚动条滑动到最底部
const scrollListBox = () => {
document.getElementById('message-list-box').scrollTo(0, document.getElementById('message-list-box').scrollHeight + 46)
}
document.getElementById("message-list-box").scrollTo(0, document.getElementById("message-list-box").scrollHeight + 46);
};
const sendMessage = () => {
if (canSend.value === false) {
showToast("AI 正在作答中,请稍后...");
return
return;
}
if (store.socket.conn.readyState !== WebSocket.OPEN) {
showToast("连接断开,正在重连...");
return
return;
}
if (prompt.value.trim().length === 0) {
showToast("请输入需要 AI 回答的问题")
showToast("请输入需要 AI 回答的问题");
return false;
}
@@ -532,91 +516,95 @@ const sendMessage = () => {
});
nextTick(() => {
scrollListBox()
})
scrollListBox();
});
disableInput(false)
store.socket.conn.send(JSON.stringify({
channel: 'chat',
type:'text',
body:{
role_id: roleId.value,
model_id: modelId.value,
chat_id: chatId.value,
content: prompt.value,
stream: stream.value
}
}));
disableInput(false);
store.socket.conn.send(
JSON.stringify({
channel: "chat",
type: "text",
body: {
role_id: roleId.value,
model_id: modelId.value,
chat_id: chatId.value,
content: prompt.value,
stream: stream.value,
},
})
);
previousText.value = prompt.value;
prompt.value = '';
prompt.value = "";
return true;
}
};
const stopGenerate = () => {
showStopGenerate.value = false;
httpGet("/api/chat/stop?session_id=" + getClientId()).then(() => {
enableInput()
})
}
enableInput();
});
};
const reGenerate = () => {
disableInput(false)
const text = '重新生成上述问题的答案:' + previousText.value;
disableInput(false);
const text = "重新生成上述问题的答案:" + previousText.value;
// 追加消息
chatData.value.push({
type: "prompt",
id: randString(32),
icon: loginUser.value.avatar,
content: renderInputText(text)
content: renderInputText(text),
});
store.socket.conn.send(JSON.stringify({
channel: 'chat',
type:'text',
body:{
role_id: roleId.value,
model_id: modelId.value,
chat_id: chatId.value,
content: previousText.value,
stream: stream.value
}
}));
}
store.socket.conn.send(
JSON.stringify({
channel: "chat",
type: "text",
body: {
role_id: roleId.value,
model_id: modelId.value,
chat_id: chatId.value,
content: previousText.value,
stream: stream.value,
},
})
);
};
const showShare = ref(false)
const showShare = ref(false);
const shareOptions = [
{name: '微信', icon: 'wechat'},
{name: '复制链接', icon: 'link'},
]
{ name: "微信", icon: "wechat" },
{ name: "复制链接", icon: "link" },
];
const shareChat = (option) => {
showShare.value = false
showShare.value = false;
if (option.icon === "wechat") {
showToast({message: "当前会话已经导出,请通过浏览器或者微信的自带分享功能分享给好友", duration: 5000})
showToast({ message: "当前会话已经导出,请通过浏览器或者微信的自带分享功能分享给好友", duration: 5000 });
router.push({
path: "/mobile/chat/export",
query: {title: title.value, chat_id: chatId.value, role: role.value.name, model: modelValue.value}
})
query: { title: title.value, chat_id: chatId.value, role: role.value.name, model: modelValue.value },
});
} else if (option.icon === "link") {
document.getElementById('copy-link-btn').click();
document.getElementById("copy-link-btn").click();
}
}
};
const getRoleById = function (rid) {
for (let i = 0; i < roles.value.length; i++) {
if (roles.value[i]['id'] === rid) {
if (roles.value[i]["id"] === rid) {
return roles.value[i];
}
}
return null;
}
};
const getModelName = (model_id) => {
for (let i = 0; i < models.value.length; i++) {
if (models.value[i].id === model_id) {
return models.value[i].text
return models.value[i].text;
}
}
return ""
}
return "";
};
// // eslint-disable-next-line no-undef
// const recognition = new webkitSpeechRecognition() || SpeechRecognition();
@@ -647,4 +635,4 @@ const getModelName = (model_id) => {
<style lang="stylus">
@import "@/assets/css/mobile/chat-session.styl"
</style>
</style>