Feat: Use AvatarUpload to replace the avatar settings on the dataset and search pages #3221 (#9785)

### What problem does this PR solve?

Feat: Use AvatarUpload to replace the avatar settings on the dataset and
search pages #3221
### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-08-28 14:45:20 +08:00
committed by GitHub
parent 4720849ac0
commit 5fe8cf6018
6 changed files with 83 additions and 259 deletions

View File

@ -1,7 +1,7 @@
import { AvatarUpload } from '@/components/avatar-upload';
import { FormContainer } from '@/components/form-container';
import { SelectWithSearch } from '@/components/originui/select-with-search';
import { RAGFlowFormItem } from '@/components/ragflow-form';
import { Avatar, AvatarFallback, AvatarImage } from '@/components/ui/avatar';
import { Button, ButtonLoading } from '@/components/ui/button';
import {
FormControl,
@ -12,10 +12,8 @@ import {
} from '@/components/ui/form';
import { Input } from '@/components/ui/input';
import { PermissionRole } from '@/constants/permission';
import { useUpdateKnowledge } from '@/hooks/knowledge-hooks';
import { transformFile2Base64 } from '@/utils/file-util';
import { Pencil, Upload } from 'lucide-react';
import { useEffect, useMemo, useState } from 'react';
import { useUpdateKnowledge } from '@/hooks/use-knowledge-request';
import { useMemo } from 'react';
import { useFormContext } from 'react-hook-form';
import { useTranslation } from 'react-i18next';
import { useParams } from 'umi';
@ -23,8 +21,6 @@ import { useParams } from 'umi';
export function GeneralForm() {
const form = useFormContext();
const { t } = useTranslation();
const [avatarFile, setAvatarFile] = useState<File | null>(null);
const [avatarBase64Str, setAvatarBase64Str] = useState(''); // Avatar Image base64
const { saveKnowledgeConfiguration, loading: submitLoading } =
useUpdateKnowledge();
@ -43,26 +39,6 @@ export function GeneralForm() {
}));
}, [t]);
// init avatar file if it exists in defaultValues
useEffect(() => {
if (!avatarFile) {
let avatarList = defaultValues['avatar'];
if (avatarList && avatarList.length > 0) {
setAvatarBase64Str(avatarList[0].thumbUrl);
}
}
}, [avatarFile, defaultValues]);
// input[type=file] on change event, get img base64
useEffect(() => {
if (avatarFile) {
(async () => {
// make use of img compression transformFile2Base64
setAvatarBase64Str(await transformFile2Base64(avatarFile));
})();
}
}, [avatarFile]);
return (
<>
<FormContainer className="space-y-10 p-10">
@ -90,62 +66,14 @@ export function GeneralForm() {
<FormField
control={form.control}
name="avatar"
render={() => (
render={({ field }) => (
<FormItem className="items-center space-y-0">
<div className="flex">
<FormLabel className="text-sm text-muted-foreground whitespace-nowrap w-1/4">
{t('setting.avatar')}
</FormLabel>
<FormControl className="w-3/4">
<div className="flex justify-start items-end space-x-2">
<div className="relative group">
{!avatarBase64Str ? (
<div className="w-[64px] h-[64px] grid place-content-center border border-dashed rounded-md">
<div className="flex flex-col items-center">
<Upload />
<p>{t('common.upload')}</p>
</div>
</div>
) : (
<div className="w-[64px] h-[64px] relative grid place-content-center">
<Avatar className="w-[64px] h-[64px] rounded-md">
<AvatarImage
className=" block"
src={avatarBase64Str}
alt=""
/>
<AvatarFallback></AvatarFallback>
</Avatar>
<div className="absolute inset-0 bg-[#000]/20 group-hover:bg-[#000]/60">
<Pencil
size={20}
className="absolute right-2 bottom-0 opacity-50 hidden group-hover:block"
/>
</div>
</div>
)}
<Input
placeholder=""
// {...field}
type="file"
title=""
accept="image/*"
className="absolute top-0 left-0 w-full h-full opacity-0 cursor-pointer"
onChange={(ev) => {
const file = ev.target?.files?.[0];
if (
/\.(jpg|jpeg|png|webp|bmp)$/i.test(file?.name ?? '')
) {
setAvatarFile(file!);
}
ev.target.value = '';
}}
/>
</div>
<div className="margin-1 text-muted-foreground">
{t('knowledgeConfiguration.photoTip')}
</div>
</div>
<AvatarUpload {...field}></AvatarUpload>
</FormControl>
</div>
<div className="flex pt-1">
@ -209,8 +137,8 @@ export function GeneralForm() {
onClick={() => {
(async () => {
let isValidate = await form.trigger('name');
const { name, description, permission } = form.getValues();
const avatar = avatarBase64Str;
const { name, description, permission, avatar } =
form.getValues();
if (isValidate) {
saveKnowledgeConfiguration({