add translate api for midjourney

This commit is contained in:
RockYang 2023-12-11 17:01:02 +08:00
parent 411335ebcb
commit 1067fa015e
2 changed files with 37 additions and 29 deletions

View File

@ -1,6 +1,7 @@
package handler
import (
"chatplus/core"
"chatplus/core/types"
"chatplus/store/model"
"chatplus/utils/resp"
@ -19,8 +20,10 @@ type PromptHandler struct {
db *gorm.DB
}
func NewPromptHandler(db *gorm.DB) *PromptHandler {
return &PromptHandler{db: db}
func NewPromptHandler(app *core.AppServer, db *gorm.DB) *PromptHandler {
h := &PromptHandler{db: db}
h.App = app
return h
}
type apiRes struct {

View File

@ -233,6 +233,7 @@
</el-tab-pane>
</el-tabs>
<div v-loading="loading" element-loading-background="rgba(122, 122, 122, 0.8)">
<div class="param-line pt">
<div class="flex-row justify-between items-center">
<div class="flex-row justify-start items-center">
@ -257,6 +258,7 @@
ref="promptRef"
placeholder="这里输入你的英文咒语例如A chinese girl walking in the middle of a cobblestone street"/>
</div>
</div>
<div class="param-line pt">
<div class="flex-row justify-between items-center">
@ -500,6 +502,7 @@ const router = useRouter()
const socket = ref(null)
const imgCalls = ref(0)
const loading = ref(false)
const connect = () => {
let host = process.env.VUE_APP_WS_HOST
@ -569,8 +572,10 @@ const connect = () => {
}
const translatePrompt = () => {
loading.value = true
httpPost("/api/prompt/translate", {"prompt": params.value.prompt}).then(res => {
params.value.prompt = res.data
loading.value = false
}).then(e => {
ElMessage.error("翻译失败:" + e.message)
})