mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-04-03 06:40:22 +00:00
Compare commits
2 Commits
6475e042b5
...
19fc63e3af
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
19fc63e3af | ||
|
|
b8b58e980b |
46
web/package-lock.json
generated
46
web/package-lock.json
generated
@ -56,6 +56,7 @@
|
||||
"react-day-picker": "^8.10.1",
|
||||
"react-device-detect": "^2.2.3",
|
||||
"react-dom": "^18.3.1",
|
||||
"react-dropzone": "^14.3.8",
|
||||
"react-grid-layout": "^1.5.0",
|
||||
"react-hook-form": "^7.52.1",
|
||||
"react-i18next": "^15.2.0",
|
||||
@ -3526,6 +3527,15 @@
|
||||
"resolved": "https://registry.npmjs.org/asynckit/-/asynckit-0.4.0.tgz",
|
||||
"integrity": "sha512-Oei9OH4tRh0YqU3GxhX79dM/mwVgvbZJaSNaRk+bshkj0S5cfHcgYakreBjrHwatXKbz+IoIdYLxrKim2MjW0Q=="
|
||||
},
|
||||
"node_modules/attr-accept": {
|
||||
"version": "2.2.5",
|
||||
"resolved": "https://registry.npmjs.org/attr-accept/-/attr-accept-2.2.5.tgz",
|
||||
"integrity": "sha512-0bDNnY/u6pPwHDMoF0FieU354oBi0a8rD9FcsLwzcGWbc8KS8KPIi7y+s13OlVY+gMWc/9xEMUgNE6Qm8ZllYQ==",
|
||||
"license": "MIT",
|
||||
"engines": {
|
||||
"node": ">=4"
|
||||
}
|
||||
},
|
||||
"node_modules/autoprefixer": {
|
||||
"version": "10.4.20",
|
||||
"resolved": "https://registry.npmjs.org/autoprefixer/-/autoprefixer-10.4.20.tgz",
|
||||
@ -5112,6 +5122,18 @@
|
||||
"node": "^10.12.0 || >=12.0.0"
|
||||
}
|
||||
},
|
||||
"node_modules/file-selector": {
|
||||
"version": "2.1.2",
|
||||
"resolved": "https://registry.npmjs.org/file-selector/-/file-selector-2.1.2.tgz",
|
||||
"integrity": "sha512-QgXo+mXTe8ljeqUFaX3QVHc5osSItJ/Km+xpocx0aSqWGMSCf6qYs/VnzZgS864Pjn5iceMRFigeAV7AfTlaig==",
|
||||
"license": "MIT",
|
||||
"dependencies": {
|
||||
"tslib": "^2.7.0"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">= 12"
|
||||
}
|
||||
},
|
||||
"node_modules/fill-range": {
|
||||
"version": "7.0.1",
|
||||
"resolved": "https://registry.npmjs.org/fill-range/-/fill-range-7.0.1.tgz",
|
||||
@ -7221,6 +7243,23 @@
|
||||
"node": ">=6"
|
||||
}
|
||||
},
|
||||
"node_modules/react-dropzone": {
|
||||
"version": "14.3.8",
|
||||
"resolved": "https://registry.npmjs.org/react-dropzone/-/react-dropzone-14.3.8.tgz",
|
||||
"integrity": "sha512-sBgODnq+lcA4P296DY4wacOZz3JFpD99fp+hb//iBO2HHnyeZU3FwWyXJ6salNpqQdsZrgMrotuko/BdJMV8Ug==",
|
||||
"license": "MIT",
|
||||
"dependencies": {
|
||||
"attr-accept": "^2.2.4",
|
||||
"file-selector": "^2.1.0",
|
||||
"prop-types": "^15.8.1"
|
||||
},
|
||||
"engines": {
|
||||
"node": ">= 10.13"
|
||||
},
|
||||
"peerDependencies": {
|
||||
"react": ">= 16.8 || 18.0.0"
|
||||
}
|
||||
},
|
||||
"node_modules/react-grid-layout": {
|
||||
"version": "1.5.0",
|
||||
"resolved": "https://registry.npmjs.org/react-grid-layout/-/react-grid-layout-1.5.0.tgz",
|
||||
@ -8548,9 +8587,10 @@
|
||||
"integrity": "sha512-Y/arvbn+rrz3JCKl9C4kVNfTfSm2/mEp5FSz5EsZSANGPSlQrpRI5M4PKF+mJnE52jOO90PnPSc3Ur3bTQw0gA=="
|
||||
},
|
||||
"node_modules/tslib": {
|
||||
"version": "2.6.2",
|
||||
"resolved": "https://registry.npmjs.org/tslib/-/tslib-2.6.2.tgz",
|
||||
"integrity": "sha512-AEYxH93jGFPn/a2iVAwW87VuUIkR1FVUKB77NwMF7nBTDkDrrT/Hpt/IrCJ0QXhW27jTBDcf5ZY7w6RiqTMw2Q=="
|
||||
"version": "2.8.1",
|
||||
"resolved": "https://registry.npmjs.org/tslib/-/tslib-2.8.1.tgz",
|
||||
"integrity": "sha512-oJFu94HQb+KVduSUQL7wnpmqnfmLsOA/nAh6b6EH0wCEoK0/mPeXU6c3wKDV83MkOuHPRHtSXKKU99IBazS/2w==",
|
||||
"license": "0BSD"
|
||||
},
|
||||
"node_modules/type-check": {
|
||||
"version": "0.4.0",
|
||||
|
||||
@ -62,6 +62,7 @@
|
||||
"react-day-picker": "^8.10.1",
|
||||
"react-device-detect": "^2.2.3",
|
||||
"react-dom": "^18.3.1",
|
||||
"react-dropzone": "^14.3.8",
|
||||
"react-grid-layout": "^1.5.0",
|
||||
"react-hook-form": "^7.52.1",
|
||||
"react-i18next": "^15.2.0",
|
||||
|
||||
@ -1,6 +1,7 @@
|
||||
{
|
||||
"description": {
|
||||
"addFace": "Walk through adding a new face to the Face Library."
|
||||
"addFace": "Walk through adding a new collection to the Face Library.",
|
||||
"placeholder": "Enter a name for this collection"
|
||||
},
|
||||
"details": {
|
||||
"person": "Person",
|
||||
@ -15,30 +16,47 @@
|
||||
"desc": "Upload an image to scan for faces and include for {{pageToggle}}"
|
||||
},
|
||||
"createFaceLibrary": {
|
||||
"title": "Create Face Library",
|
||||
"desc": "Create a new face library",
|
||||
"title": "Create Collection",
|
||||
"desc": "Create a new collection",
|
||||
"new": "Create New Face",
|
||||
"nextSteps": "It is recommended to use the Train tab to select and train images for each person as they are detected. When building a strong foundation it is strongly recommended to only train on images that are straight-on. Ignore images from cameras that recognize faces from an angle."
|
||||
"nextSteps": "To build a strong foundation:<li>Use the Train tab to select and train on images for each detected person.</li><li>Focus on straight-on images for best results; avoid training images that capture faces at an angle.</li></ul>"
|
||||
},
|
||||
"train": {
|
||||
"title": "Train",
|
||||
"aria": "Select train"
|
||||
},
|
||||
"selectItem": "Select {{item}}",
|
||||
"selectFace": "Select Face",
|
||||
"deleteFaceLibrary": {
|
||||
"title": "Delete Name",
|
||||
"desc": "Are you sure you want to delete the collection {{name}}? This will permanently delete all associated faces."
|
||||
},
|
||||
"button": {
|
||||
"deleteFaceAttempts": "Delete Face Attempts",
|
||||
"addFace": "Add Face",
|
||||
"uploadImage": "Upload Image",
|
||||
"reprocessFace": "Reprocess Face"
|
||||
},
|
||||
"readTheDocs": "Read the documentation to view more details on refining images for the Face Library",
|
||||
"imageEntry": {
|
||||
"validation": {
|
||||
"selectImage": "Please select an image file."
|
||||
},
|
||||
"dropActive": "Drop the image here...",
|
||||
"dropInstructions": "Drag and drop an image here, or click to select",
|
||||
"maxSize": "Max size: {{size}}MB"
|
||||
},
|
||||
"readTheDocs": "Read the documentation",
|
||||
"trainFaceAs": "Train Face as:",
|
||||
"trainFace": "Train Face",
|
||||
"toast": {
|
||||
"success": {
|
||||
"uploadedImage": "Successfully uploaded image.",
|
||||
"addFaceLibrary": "{{name}} has successfully been added to the Face Library!",
|
||||
"deletedFace": "Successfully deleted face.",
|
||||
"deletedFace_one": "Successfully deleted {{count}} face.",
|
||||
"deletedFace_other": "Successfully deleted {{count}} faces.",
|
||||
"deletedName_zero": "Empty collection deleted successfully.",
|
||||
"deletedName_one": "{{count}} face has been successfully deleted.",
|
||||
"deletedName_other": "{{count}} faces have been successfully deleted.",
|
||||
"trainedFace": "Successfully trained face.",
|
||||
"updatedFaceScore": "Successfully updated face score."
|
||||
},
|
||||
@ -46,6 +64,7 @@
|
||||
"uploadingImageFailed": "Failed to upload image: {{errorMessage}}",
|
||||
"addFaceLibraryFailed": "Failed to set face name: {{errorMessage}}",
|
||||
"deleteFaceFailed": "Failed to delete: {{errorMessage}}",
|
||||
"deleteNameFailed": "Failed to delete name: {{errorMessage}}",
|
||||
"trainFailed": "Failed to train: {{errorMessage}}",
|
||||
"updateFaceScoreFailed": "Failed to update face score: {{errorMessage}}"
|
||||
}
|
||||
|
||||
@ -1,38 +1,82 @@
|
||||
import { Form, FormControl, FormField, FormItem } from "@/components/ui/form";
|
||||
import { Input } from "@/components/ui/input";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import {
|
||||
Form,
|
||||
FormControl,
|
||||
FormField,
|
||||
FormItem,
|
||||
FormMessage,
|
||||
} from "@/components/ui/form";
|
||||
import { cn } from "@/lib/utils";
|
||||
import { zodResolver } from "@hookform/resolvers/zod";
|
||||
import React, { useCallback } from "react";
|
||||
import { useCallback, useState } from "react";
|
||||
import { useDropzone } from "react-dropzone";
|
||||
import { useForm } from "react-hook-form";
|
||||
|
||||
import { useTranslation } from "react-i18next";
|
||||
import { LuUpload, LuX } from "react-icons/lu";
|
||||
import { z } from "zod";
|
||||
|
||||
type ImageEntryProps = {
|
||||
onSave: (file: File) => void;
|
||||
children?: React.ReactNode;
|
||||
maxSize?: number;
|
||||
accept?: Record<string, string[]>;
|
||||
};
|
||||
export default function ImageEntry({ onSave, children }: ImageEntryProps) {
|
||||
|
||||
export default function ImageEntry({
|
||||
onSave,
|
||||
children,
|
||||
maxSize = 10 * 1024 * 1024, // 10MB default
|
||||
accept = { "image/*": [".jpeg", ".jpg", ".png", ".gif", ".webp"] },
|
||||
}: ImageEntryProps) {
|
||||
const { t } = useTranslation(["views/faceLibrary"]);
|
||||
const [preview, setPreview] = useState<string | null>(null);
|
||||
|
||||
const formSchema = z.object({
|
||||
file: z.instanceof(FileList, { message: "Please select an image file." }),
|
||||
file: z.instanceof(File, { message: "Please select an image file." }),
|
||||
});
|
||||
|
||||
const form = useForm<z.infer<typeof formSchema>>({
|
||||
resolver: zodResolver(formSchema),
|
||||
});
|
||||
const fileRef = form.register("file");
|
||||
|
||||
// upload handler
|
||||
const onDrop = useCallback(
|
||||
(acceptedFiles: File[]) => {
|
||||
if (acceptedFiles.length > 0) {
|
||||
const file = acceptedFiles[0];
|
||||
form.setValue("file", file, { shouldValidate: true });
|
||||
|
||||
// Create preview
|
||||
const objectUrl = URL.createObjectURL(file);
|
||||
setPreview(objectUrl);
|
||||
|
||||
// Clean up preview URL when component unmounts
|
||||
return () => URL.revokeObjectURL(objectUrl);
|
||||
}
|
||||
},
|
||||
[form],
|
||||
);
|
||||
|
||||
const { getRootProps, getInputProps, isDragActive, isDragReject } =
|
||||
useDropzone({
|
||||
onDrop,
|
||||
maxSize,
|
||||
accept,
|
||||
multiple: false,
|
||||
});
|
||||
|
||||
const onSubmit = useCallback(
|
||||
(data: z.infer<typeof formSchema>) => {
|
||||
if (!data["file"] || Object.keys(data.file).length == 0) {
|
||||
return;
|
||||
}
|
||||
|
||||
onSave(data["file"]["0"]);
|
||||
if (!data.file) return;
|
||||
onSave(data.file);
|
||||
},
|
||||
[onSave],
|
||||
);
|
||||
|
||||
const clearSelection = () => {
|
||||
form.reset();
|
||||
setPreview(null);
|
||||
};
|
||||
|
||||
return (
|
||||
<Form {...form}>
|
||||
<form onSubmit={form.handleSubmit(onSubmit)}>
|
||||
@ -42,16 +86,55 @@ export default function ImageEntry({ onSave, children }: ImageEntryProps) {
|
||||
render={() => (
|
||||
<FormItem>
|
||||
<FormControl>
|
||||
<Input
|
||||
className="aspect-video h-40 w-full"
|
||||
type="file"
|
||||
{...fileRef}
|
||||
/>
|
||||
<div className="w-full">
|
||||
{!preview ? (
|
||||
<div
|
||||
{...getRootProps()}
|
||||
className={cn(
|
||||
"flex h-40 flex-col items-center justify-center rounded-lg border-2 border-dashed p-6 transition-colors",
|
||||
isDragActive && "border-primary bg-primary/5",
|
||||
isDragReject && "border-destructive bg-destructive/5",
|
||||
"cursor-pointer hover:border-primary hover:bg-primary/5",
|
||||
)}
|
||||
>
|
||||
<input {...getInputProps()} />
|
||||
<LuUpload className="mb-2 h-10 w-10 text-muted-foreground" />
|
||||
<p className="text-center text-sm text-muted-foreground">
|
||||
{isDragActive
|
||||
? t("imageEntry.dropActive")
|
||||
: t("imageEntry.dropInstructions")}
|
||||
</p>
|
||||
<p className="mt-1 text-xs text-muted-foreground">
|
||||
{t("imageEntry.maxSize", {
|
||||
size: Math.round(maxSize / (1024 * 1024)),
|
||||
})}
|
||||
</p>
|
||||
</div>
|
||||
) : (
|
||||
<div className="relative h-40 w-full">
|
||||
<img
|
||||
src={preview}
|
||||
alt="Preview"
|
||||
className="h-full w-full rounded-lg border object-contain"
|
||||
/>
|
||||
<Button
|
||||
type="button"
|
||||
variant="destructive"
|
||||
size="icon"
|
||||
className="absolute right-2 top-2 size-5 rounded-full"
|
||||
onClick={clearSelection}
|
||||
>
|
||||
<LuX className="size-3" />
|
||||
</Button>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
</FormControl>
|
||||
<FormMessage />
|
||||
</FormItem>
|
||||
)}
|
||||
/>
|
||||
{children}
|
||||
<div className="mt-4 flex justify-end">{children}</div>
|
||||
</form>
|
||||
</Form>
|
||||
);
|
||||
|
||||
@ -1,4 +1,10 @@
|
||||
import { Form, FormControl, FormField, FormItem } from "@/components/ui/form";
|
||||
import {
|
||||
Form,
|
||||
FormControl,
|
||||
FormField,
|
||||
FormItem,
|
||||
FormMessage,
|
||||
} from "@/components/ui/form";
|
||||
import { Input } from "@/components/ui/input";
|
||||
import { zodResolver } from "@hookform/resolvers/zod";
|
||||
import React, { useCallback } from "react";
|
||||
@ -14,50 +20,47 @@ type TextEntryProps = {
|
||||
children?: React.ReactNode;
|
||||
};
|
||||
export default function TextEntry({
|
||||
defaultValue,
|
||||
defaultValue = "",
|
||||
placeholder,
|
||||
allowEmpty,
|
||||
allowEmpty = false,
|
||||
onSave,
|
||||
children,
|
||||
}: TextEntryProps) {
|
||||
const formSchema = z.object({
|
||||
text: z.string(),
|
||||
text: allowEmpty
|
||||
? z.string().optional()
|
||||
: z.string().min(1, "Field is required"),
|
||||
});
|
||||
|
||||
const form = useForm<z.infer<typeof formSchema>>({
|
||||
resolver: zodResolver(formSchema),
|
||||
defaultValues: { text: defaultValue },
|
||||
});
|
||||
const fileRef = form.register("text");
|
||||
|
||||
// upload handler
|
||||
|
||||
const onSubmit = useCallback(
|
||||
(data: z.infer<typeof formSchema>) => {
|
||||
if (!allowEmpty && !data["text"]) {
|
||||
return;
|
||||
}
|
||||
onSave(data["text"]);
|
||||
onSave(data.text || "");
|
||||
},
|
||||
[onSave, allowEmpty],
|
||||
[onSave],
|
||||
);
|
||||
|
||||
return (
|
||||
<Form {...form}>
|
||||
<form onSubmit={form.handleSubmit(onSubmit)}>
|
||||
<form onSubmit={form.handleSubmit(onSubmit)} className="space-y-4">
|
||||
<FormField
|
||||
control={form.control}
|
||||
name="text"
|
||||
render={() => (
|
||||
render={({ field }) => (
|
||||
<FormItem>
|
||||
<FormControl>
|
||||
<Input
|
||||
className="aspect-video h-8 w-full"
|
||||
{...field}
|
||||
className="w-full"
|
||||
placeholder={placeholder}
|
||||
type="text"
|
||||
{...fileRef}
|
||||
/>
|
||||
</FormControl>
|
||||
<FormMessage className="text-xs text-destructive" />
|
||||
</FormItem>
|
||||
)}
|
||||
/>
|
||||
|
||||
@ -20,7 +20,7 @@ import { cn } from "@/lib/utils";
|
||||
import axios from "axios";
|
||||
import { useCallback, useState } from "react";
|
||||
import { isDesktop } from "react-device-detect";
|
||||
import { useTranslation } from "react-i18next";
|
||||
import { Trans, useTranslation } from "react-i18next";
|
||||
import { LuExternalLink } from "react-icons/lu";
|
||||
import { Link } from "react-router-dom";
|
||||
import { toast } from "sonner";
|
||||
@ -101,7 +101,7 @@ export default function CreateFaceWizardDialog({
|
||||
}}
|
||||
>
|
||||
<Content
|
||||
className={cn("flex flex-col gap-4", isDesktop ? "max-w-[50%]" : "p-4")}
|
||||
className={cn("flex flex-col gap-4", isDesktop ? "max-w-3xl" : "p-4")}
|
||||
>
|
||||
<Header>
|
||||
<Title>{t("button.addFace")}</Title>
|
||||
@ -110,7 +110,7 @@ export default function CreateFaceWizardDialog({
|
||||
<StepIndicator steps={STEPS} currentStep={step} />
|
||||
{step == 0 && (
|
||||
<TextEntry
|
||||
placeholder="Enter Face Name"
|
||||
placeholder={t("description.placeholder")}
|
||||
onSave={(name) => {
|
||||
setName(name);
|
||||
setStep(1);
|
||||
@ -133,12 +133,16 @@ export default function CreateFaceWizardDialog({
|
||||
</ImageEntry>
|
||||
)}
|
||||
{step == 2 && (
|
||||
<div>
|
||||
<div className="mt-2">
|
||||
{t("toast.success.addFaceLibrary", { name })}
|
||||
<p className="py-4 text-sm text-secondary-foreground">
|
||||
{t("createFaceLibrary.nextSteps")}
|
||||
<p className="py-4 text-sm text-primary-variant">
|
||||
<ul className="list-inside list-disc">
|
||||
<Trans ns="views/faceLibrary">
|
||||
createFaceLibrary.nextSteps
|
||||
</Trans>
|
||||
</ul>
|
||||
</p>
|
||||
<div className="text-s my-4 flex items-center text-primary">
|
||||
<div className="my-2 flex items-center text-sm text-primary">
|
||||
<Link
|
||||
to="https://docs.frigate.video/configuration/face_recognition"
|
||||
target="_blank"
|
||||
|
||||
@ -19,10 +19,9 @@ import {
|
||||
DropdownMenuItem,
|
||||
DropdownMenuLabel,
|
||||
DropdownMenuTrigger,
|
||||
DropdownMenuSeparator,
|
||||
} from "@/components/ui/dropdown-menu";
|
||||
import { ScrollArea, ScrollBar } from "@/components/ui/scroll-area";
|
||||
import { Toaster } from "@/components/ui/sonner";
|
||||
import { ToggleGroup, ToggleGroupItem } from "@/components/ui/toggle-group";
|
||||
import {
|
||||
Tooltip,
|
||||
TooltipContent,
|
||||
@ -174,16 +173,28 @@ export default function FaceLibrary() {
|
||||
);
|
||||
|
||||
const onDelete = useCallback(
|
||||
(name: string, ids: string[]) => {
|
||||
(name: string, ids: string[], isName: boolean = false) => {
|
||||
axios
|
||||
.post(`/faces/${name}/delete`, { ids })
|
||||
.then((resp) => {
|
||||
setSelectedFaces([]);
|
||||
|
||||
if (resp.status == 200) {
|
||||
toast.success(t("toast.success.deletedFace"), {
|
||||
position: "top-center",
|
||||
});
|
||||
if (isName) {
|
||||
toast.success(
|
||||
t("toast.success.deletedName", { count: ids.length }),
|
||||
{
|
||||
position: "top-center",
|
||||
},
|
||||
);
|
||||
} else {
|
||||
toast.success(
|
||||
t("toast.success.deletedFace", { count: ids.length }),
|
||||
{
|
||||
position: "top-center",
|
||||
},
|
||||
);
|
||||
}
|
||||
|
||||
if (faceImages.length == 1) {
|
||||
// face has been deleted
|
||||
@ -198,9 +209,15 @@ export default function FaceLibrary() {
|
||||
error.response?.data?.message ||
|
||||
error.response?.data?.detail ||
|
||||
"Unknown error";
|
||||
toast.error(t("toast.error.deleteFaceFailed", { errorMessage }), {
|
||||
position: "top-center",
|
||||
});
|
||||
if (isName) {
|
||||
toast.error(t("toast.error.deleteNameFailed", { errorMessage }), {
|
||||
position: "top-center",
|
||||
});
|
||||
} else {
|
||||
toast.error(t("toast.error.deleteFaceFailed", { errorMessage }), {
|
||||
position: "top-center",
|
||||
});
|
||||
}
|
||||
});
|
||||
},
|
||||
[faceImages, refreshFaces, setPageToggle, t],
|
||||
@ -258,6 +275,7 @@ export default function FaceLibrary() {
|
||||
faces={faces}
|
||||
trainImages={trainImages}
|
||||
setPageToggle={setPageToggle}
|
||||
onDelete={onDelete}
|
||||
/>
|
||||
{selectedFaces?.length > 0 ? (
|
||||
<div className="flex items-center justify-center gap-2">
|
||||
@ -321,6 +339,7 @@ type LibrarySelectorProps = {
|
||||
faces: string[];
|
||||
trainImages: string[];
|
||||
setPageToggle: (toggle: string | undefined) => void;
|
||||
onDelete: (name: string, ids: string[], isName: boolean) => void;
|
||||
};
|
||||
function LibrarySelector({
|
||||
pageToggle,
|
||||
@ -328,79 +347,117 @@ function LibrarySelector({
|
||||
faces,
|
||||
trainImages,
|
||||
setPageToggle,
|
||||
onDelete,
|
||||
}: LibrarySelectorProps) {
|
||||
const { t } = useTranslation(["views/faceLibrary"]);
|
||||
const [confirmDelete, setConfirmDelete] = useState<string | null>(null);
|
||||
|
||||
return isDesktop ? (
|
||||
<ScrollArea className="w-full whitespace-nowrap">
|
||||
<div className="flex flex-row">
|
||||
<ToggleGroup
|
||||
className="*:rounded-md *:px-3 *:py-4"
|
||||
type="single"
|
||||
size="sm"
|
||||
value={pageToggle}
|
||||
onValueChange={(value: string) => {
|
||||
if (value) {
|
||||
setPageToggle(value);
|
||||
}
|
||||
}}
|
||||
const handleDeleteFace = useCallback(
|
||||
(faceName: string) => {
|
||||
// Get all image IDs for this face
|
||||
const imageIds = faceData?.[faceName] || [];
|
||||
|
||||
onDelete(faceName, imageIds, true);
|
||||
setPageToggle("train");
|
||||
},
|
||||
[faceData, onDelete, setPageToggle],
|
||||
);
|
||||
|
||||
return (
|
||||
<>
|
||||
<Dialog
|
||||
open={!!confirmDelete}
|
||||
onOpenChange={(open) => !open && setConfirmDelete(null)}
|
||||
>
|
||||
<DialogContent>
|
||||
<DialogHeader>
|
||||
<DialogTitle>{t("deleteFaceLibrary.title")}</DialogTitle>
|
||||
<DialogDescription>
|
||||
{t("deleteFaceLibrary.desc", { name: confirmDelete })}
|
||||
</DialogDescription>
|
||||
</DialogHeader>
|
||||
<div className="flex justify-end gap-2">
|
||||
<Button variant="outline" onClick={() => setConfirmDelete(null)}>
|
||||
{t("button.cancel", { ns: "common" })}
|
||||
</Button>
|
||||
<Button
|
||||
variant="destructive"
|
||||
onClick={() => {
|
||||
if (confirmDelete) {
|
||||
handleDeleteFace(confirmDelete);
|
||||
setConfirmDelete(null);
|
||||
}
|
||||
}}
|
||||
>
|
||||
{t("button.delete", { ns: "common" })}
|
||||
</Button>
|
||||
</div>
|
||||
</DialogContent>
|
||||
</Dialog>
|
||||
|
||||
<DropdownMenu>
|
||||
<DropdownMenuTrigger asChild>
|
||||
<Button className="flex justify-between capitalize">
|
||||
{pageToggle || t("selectFace")}
|
||||
<span className="ml-2 text-primary-variant">
|
||||
({(pageToggle && faceData?.[pageToggle]?.length) || 0})
|
||||
</span>
|
||||
</Button>
|
||||
</DropdownMenuTrigger>
|
||||
<DropdownMenuContent
|
||||
className="scrollbar-container max-h-[40dvh] min-w-[220px] overflow-y-auto"
|
||||
align="start"
|
||||
>
|
||||
{trainImages.length > 0 && (
|
||||
<DropdownMenuItem
|
||||
className="flex cursor-pointer items-center justify-start gap-2"
|
||||
aria-label={t("train.aria")}
|
||||
onClick={() => setPageToggle("train")}
|
||||
>
|
||||
<div>{t("train.title")}</div>
|
||||
<div className="text-secondary-foreground">
|
||||
({trainImages.length})
|
||||
</div>
|
||||
</DropdownMenuItem>
|
||||
)}
|
||||
{trainImages.length > 0 && faces.length > 0 && (
|
||||
<>
|
||||
<ToggleGroupItem
|
||||
value="train"
|
||||
className={`flex scroll-mx-10 items-center justify-between gap-2 ${pageToggle == "train" ? "" : "*:text-muted-foreground"}`}
|
||||
data-nav-item="train"
|
||||
aria-label={t("train.aria")}
|
||||
>
|
||||
<div>{t("train.title")}</div>
|
||||
</ToggleGroupItem>
|
||||
<div>|</div>
|
||||
<DropdownMenuSeparator />
|
||||
<div className="mb-1 ml-1.5 text-xs text-secondary-foreground">
|
||||
Collections
|
||||
</div>
|
||||
</>
|
||||
)}
|
||||
|
||||
{Object.values(faces).map((face) => (
|
||||
<ToggleGroupItem
|
||||
<DropdownMenuItem
|
||||
key={face}
|
||||
className={`flex scroll-mx-10 items-center justify-between gap-2 ${pageToggle == face ? "" : "*:text-muted-foreground"}`}
|
||||
value={face}
|
||||
data-nav-item={face}
|
||||
aria-label={t("selectItem", { item: face })}
|
||||
className="group flex items-center justify-between"
|
||||
>
|
||||
<div className="capitalize">
|
||||
{face} ({faceData?.[face].length})
|
||||
<div
|
||||
className="flex-grow cursor-pointer capitalize"
|
||||
onClick={() => setPageToggle(face)}
|
||||
>
|
||||
{face}
|
||||
<span className="ml-2 text-muted-foreground">
|
||||
({faceData?.[face].length})
|
||||
</span>
|
||||
</div>
|
||||
</ToggleGroupItem>
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="icon"
|
||||
className="size-7 opacity-0 transition-opacity group-hover:opacity-100"
|
||||
onClick={(e) => {
|
||||
e.stopPropagation();
|
||||
setConfirmDelete(face);
|
||||
}}
|
||||
>
|
||||
<LuTrash2 className="size-4 text-destructive" />
|
||||
</Button>
|
||||
</DropdownMenuItem>
|
||||
))}
|
||||
</ToggleGroup>
|
||||
<ScrollBar orientation="horizontal" className="h-0" />
|
||||
</div>
|
||||
</ScrollArea>
|
||||
) : (
|
||||
<DropdownMenu>
|
||||
<DropdownMenuTrigger asChild>
|
||||
<Button className="capitalize">{pageToggle}</Button>
|
||||
</DropdownMenuTrigger>
|
||||
<DropdownMenuContent>
|
||||
{trainImages.length > 0 && (
|
||||
<DropdownMenuItem
|
||||
className="capitalize"
|
||||
aria-label={t("train.aria")}
|
||||
onClick={() => setPageToggle("train")}
|
||||
>
|
||||
<div>{t("train.title")}</div>
|
||||
</DropdownMenuItem>
|
||||
)}
|
||||
{Object.values(faces).map((face) => (
|
||||
<DropdownMenuItem
|
||||
className="capitalize"
|
||||
onClick={() => setPageToggle(face)}
|
||||
>
|
||||
{face} ({faceData?.[face].length})
|
||||
</DropdownMenuItem>
|
||||
))}
|
||||
</DropdownMenuContent>
|
||||
</DropdownMenu>
|
||||
</DropdownMenuContent>
|
||||
</DropdownMenu>
|
||||
</>
|
||||
);
|
||||
}
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user