[新增]ai写作页面完成

This commit is contained in:
hhhero 2024-07-07 18:14:03 +08:00
parent efcf64ed7d
commit 18e87cf967
5 changed files with 305 additions and 0 deletions

View File

@ -0,0 +1,53 @@
import request from '@/config/axios'
import { fetchEventSource } from '@microsoft/fetch-event-source'
import { getAccessToken } from '@/utils/auth'
import { config } from '@/config/axios/config'
export interface WriteParams {
/**
* 1:撰写 2:回复
*/
type: 1 | 2
/**
* 1 2
*/
prompt: string
/**
*
*/
originalContent: string
/**
*
*/
length: number
/**
*
*/
format: number
/**
*
*/
tone: number
/**
*
*/
language: number
}
export const writeStream = (data: WriteParams, onMessage, onError, onClose, ctrl) => {
// return request.post({ url: '/ai/write/generate-stream', data })
const token = getAccessToken()
return fetchEventSource(`${config.base_url}/ai/write/generate-stream`, {
method: 'post',
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${token}`
},
openWhenHidden: true,
body: JSON.stringify(data),
onmessage: onMessage,
onerror: onError,
onclose: onClose,
signal: ctrl.signal
})
}

View File

@ -0,0 +1,175 @@
<template>
<!-- 定义tab组件 -->
<DefineTab v-slot="{ active, text, itemClick }">
<span
class="inline-block w-1/2 rounded-full cursor-pointer text-center leading-[30px] relative z-1 text-[5C6370] hover:text-black"
:class="active ? 'text-black shadow-md' : 'hover:bg-[#DDDFE3]'"
@click="itemClick"
>
{{ text }}
</span>
</DefineTab>
<!-- 定义label组件 -->
<DefineLabel v-slot="{ label, hint, hintClick }">
<h3 class="mt-5 mb-3 flex items-center justify-between text-[14px]">
<span>{{ label }}</span>
<span @click="hintClick" v-if="hint" class="flex items-center text-[12px] text-[#846af7]">
<Icon icon="ep:question-filled" />
{{ hint }}
</span>
</h3>
</DefineLabel>
<!-- TODO 小屏幕的时候是定位在左边的大屏是分开的 -->
<div class="relative" v-bind="$attrs">
<!-- tab -->
<div
class="absolute left-1/2 top-2 -translate-x-1/2 w-[303px] rounded-full bg-[#DDDFE3] p-1 z-10"
>
<div
class="flex items-center relative after:content-[''] after:block after:bg-white after:h-[30px] after:w-1/2 after:absolute after:top-0 after:left-0 after:transition-transform after:rounded-full"
:class="selectedTab === 2 && 'after:transform after:translate-x-[100%]'"
>
<ReuseTab
v-for="tab in tabs"
:key="tab.value"
:text="tab.text"
:active="tab.value === selectedTab"
:itemClick="() => switchTab(tab.value)"
/>
</div>
</div>
<div
class="px-7 pb-2 pt-[46px] overflow-y-auto lg:block w-[380px] box-border bg-[#ECEDEF] h-full"
>
<div>
<template v-if="selectedTab === 1">
<ReuseLabel label="写作内容" hint="示例" />
<el-input
type="textarea"
:rows="5"
:maxlength="500"
v-model="writeForm.prompt"
placeholder="请输入写作内容"
showWordLimit
/>
</template>
<template v-else>
<ReuseLabel label="原文" hint="示例" />
<el-input
type="textarea"
:rows="5"
:maxlength="500"
v-model="writeForm.originalContent"
placeholder="请输入原文"
showWordLimit
/>
<ReuseLabel label="回复内容" />
<el-input
type="textarea"
:rows="5"
:maxlength="500"
v-model="writeForm.prompt"
placeholder="请输入回复内容"
showWordLimit
/>
</template>
<ReuseLabel label="长度" />
<Tag v-model="writeForm.length" :tags="writeTags.lenTags" />
<ReuseLabel label="格式" />
<Tag v-model="writeForm.format" :tags="writeTags.formatTags" />
<ReuseLabel label="语气" />
<Tag v-model="writeForm.tone" :tags="writeTags.toneTags" />
<ReuseLabel label="语言" />
<Tag v-model="writeForm.language" :tags="writeTags.langTags" />
<div class="flex items-center justify-center mt-3">
<el-button>重置</el-button>
<el-button @click="submit" color="#846af7">生成</el-button>
</div>
</div>
</div>
</div>
</template>
<script setup lang="ts">
import { createReusableTemplate } from '@vueuse/core'
import { ref } from 'vue'
import { ElMessage } from 'element-plus'
import Tag from './Tag.vue'
import { WriteParams } from '@/api/ai/writer'
import { omit } from 'lodash-es'
import { getIntDictOptions } from '@/utils/dict'
type TabType = WriteParams['type']
const emits = defineEmits<{
(e: 'submit', params: Partial<WriteParams>)
}>()
const selectedTab = ref<TabType>(1)
const tabs: {
text: string
value: TabType
}[] = [
{ text: '撰写', value: 1 },
{ text: '回复', value: 2 }
]
const [DefineTab, ReuseTab] = createReusableTemplate<{
active?: boolean
text: string
itemClick: () => void
}>()
const initData: WriteParams = {
type: 1,
prompt: '',
originalContent: '',
tone: 1,
language: 1,
length: 100,
format: 1
}
const writeForm = ref<WriteParams>({ ...initData })
const writeTags = {
lenTags: getIntDictOptions('ai_write_length'),
//
formatTags: getIntDictOptions('ai_write_format'),
//
toneTags: getIntDictOptions('ai_write_tone'),
//
langTags: getIntDictOptions('ai_write_language')
//
}
const [DefineLabel, ReuseLabel] = createReusableTemplate<{
label: string
class?: string
hint?: string
hintClick?: () => void
}>()
const switchTab = (value: TabType) => {
selectedTab.value = value
writeForm.value = { ...initData }
}
const submit = () => {
if (selectedTab.value === 2 && !writeForm.value.originalContent) {
ElMessage.warning('请输入原文')
return
} else if (!writeForm.value.prompt) {
ElMessage.warning(`请输入${selectedTab.value === 1 ? '写作' : '回复'}内容`)
return
}
emits('submit', {
...(selectedTab.value === 1 ? omit(writeForm.value, ['originalContent']) : writeForm.value),
type: selectedTab.value
})
}
</script>

View File

@ -0,0 +1,25 @@
<template>
<div class="h-full box-border py-6 px-7">
<div class="w-full h-full bg-white box-border p-3 sm:p-16 overflow-y-auto">
<el-input
type="textarea"
:value="msg"
autosize
:input-style="{ boxShadow: 'none' }"
resize="none"
placeholder="生成的内容……"
/>
</div>
</div>
</template>
<script setup lang="ts">
defineProps({
msg: {
type: String,
default: ''
}
})
</script>
<style scoped></style>

View File

@ -0,0 +1,32 @@
<template>
<div class="flex flex-wrap gap-[8px]">
<span
v-for="tag in props.tags"
:key="tag.value"
class="tag mb-2 border-[2px] border-solid border-[#DDDFE3] px-2 leading-6 text-[12px] bg-[#DDDFE3] rounded-[4px] cursor-pointer"
:class="modelValue === tag.value && '!border-[#846af7] text-[#846af7]'"
@click="emits('update:modelValue', tag.value)"
>
{{ tag.label }}
</span>
</div>
</template>
<script setup lang="ts">
const props = withDefaults(
defineProps<{
tags: { label: string; value: string }[]
modelValue: string
[k: string]: any
}>(),
{
tags: () => []
}
)
const emits = defineEmits<{
(e: 'update:modelValue', value: string): void
}>()
</script>
<style scoped></style>

View File

@ -0,0 +1,20 @@
<template>
<div class="h-[calc(100vh-var(--top-tool-height)-var(--app-footer-height)-40px)] -m-5 flex">
<Left class="h-full" @submit="submit" />
<Right class="flex-grow" :msg="msg" />
</div>
</template>
<script setup lang="ts">
import Left from './components/Left.vue'
import Right from './components/Right.vue'
import { writeStream } from '@/api/ai/writer'
const msg = ref('')
const submit = async (params) => {
const res = await writeStream(params)
}
</script>
<style scoped></style>