给自动化代码增加视觉功能 (#1970)

* update:增加视觉能力。

* feat: 自动化代码增加视觉识别功能
This commit is contained in:
PiexlMax(奇淼 2024-12-22 00:14:03 +08:00 committed by GitHub
parent 601aa6e245
commit cb23254152
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 89 additions and 5 deletions

View File

@ -162,6 +162,17 @@ export const butler = (data) => {
}) })
} }
export const eye = (data) => {
return service({
url: '/autoCode/llmAuto',
method: 'post',
data: { ...data, mode: 'eye' },
timeout: 1000 * 60 * 10
})
}
export const addFunc = (data) => { export const addFunc = (data) => {
return service({ return service({
url: '/autoCode/addFunc', url: '/autoCode/addFunc',

View File

@ -18,12 +18,38 @@
v-model="prompt" v-model="prompt"
type="textarea" type="textarea"
:rows="5" :rows="5"
:maxlength="100" :maxlength="2000"
:placeholder="`现已完全免费\n试试描述你的表让AI帮你完成。\n此功能需要到插件市场个人中心获取自己的AI-Path把AI-Path填入config.yaml下的autocode-->ai-path重启项目即可使用。\n按下 Ctrl+Enter 或 Cmd+Enter 直接生成`" :placeholder="`现已完全免费\n试试复制一张图片然后按下ctrl+v或者commend+v\n试试描述你的表让AI帮你完成。\n此功能需要到插件市场个人中心获取自己的AI-Path把AI-Path填入config.yaml下的autocode-->ai-path重启项目即可使用。\n按下 Ctrl+Enter 或 Cmd+Enter 直接生成`"
resize="none" resize="none"
@focus="handleFocus" @focus="handleFocus"
@blur="handleBlur" @blur="handleBlur"
/> />
<div class="flex absolute right-28 bottom-2">
<el-tooltip effect="light">
<template #content>
<div>
完全免费前往<a
class="text-blue-600"
href="https://plugin.gin-vue-admin.com/#/layout/userInfo/center"
target="_blank"
>插件市场个人中心</a
>申请AIPath填入config.yaml的ai-path属性即可使用
</div>
</template>
<el-button
:disabled="form.onlyTemplate"
type="primary"
@click="eyeFunc()"
>
<el-icon size="18">
<ai-gva />
</el-icon>
识图
</el-button>
</el-tooltip>
</div>
<div class="flex absolute right-2 bottom-2"> <div class="flex absolute right-2 bottom-2">
<el-tooltip effect="light"> <el-tooltip effect="light">
<template #content> <template #content>
@ -784,7 +810,7 @@
preview, preview,
getMeta, getMeta,
getPackageApi, getPackageApi,
llmAuto llmAuto, butler, eye
} from '@/api/autoCode' } from '@/api/autoCode'
import { getDict } from '@/utils/dictionary' import { getDict } from '@/utils/dictionary'
import { ref, watch, toRaw, onMounted, nextTick } from 'vue' import { ref, watch, toRaw, onMounted, nextTick } from 'vue'
@ -794,11 +820,13 @@
import Sortable from 'sortablejs' import Sortable from 'sortablejs'
const handleFocus = () => { const handleFocus = () => {
document.addEventListener('keydown', handleKeydown) document.addEventListener('keydown', handleKeydown);
document.addEventListener('paste', handlePaste);
} }
const handleBlur = () => { const handleBlur = () => {
document.removeEventListener('keydown', handleKeydown) document.removeEventListener('keydown', handleKeydown);
document.removeEventListener('paste', handlePaste);
} }
const handleKeydown = (event) => { const handleKeydown = (event) => {
@ -807,6 +835,25 @@
} }
} }
const handlePaste = (event) => {
const items = event.clipboardData.items;
for (let i = 0; i < items.length; i++) {
if (items[i].type.indexOf('image') !== -1) {
const file = items[i].getAsFile();
const reader = new FileReader();
reader.onload =async (e) => {
const base64String = e.target.result;
const res = await eye({ picture: base64String,command: 'eye' })
if (res.code === 0) {
prompt.value = res.data
llmAutoFunc()
}
};
reader.readAsDataURL(file);
}
}
};
const getOnlyNumber = () => { const getOnlyNumber = () => {
let randomNumber = '' let randomNumber = ''
while (randomNumber.length < 16) { while (randomNumber.length < 16) {
@ -817,6 +864,32 @@
const prompt = ref('') const prompt = ref('')
const eyeFunc = async () => {
const input = document.createElement('input');
input.type = 'file';
input.accept = 'image/*';
input.onchange = (event) => {
const file = event.target.files[0];
if (file) {
const reader = new FileReader();
reader.onload = async (e) => {
const base64String = e.target.result;
const res = await eye({ picture: base64String,command: 'eye' })
if (res.code === 0) {
prompt.value = res.data
llmAutoFunc()
}
};
reader.readAsDataURL(file);
}
};
input.click();
}
const llmAutoFunc = async (flag) => { const llmAutoFunc = async (flag) => {
if (flag && !form.value.structName) { if (flag && !form.value.structName) {
ElMessage.error('请输入结构体名称') ElMessage.error('请输入结构体名称')