Skip to content

feat(text-annotation): improve manual text annotation and enable visibility in Label Studio #234

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
o0Shark0o wants to merge 1 commit into from
Closed

feat(text-annotation): improve manual text annotation and enable visibility in Label Studio #234

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion frontend/src/components/business/DatasetFileTransfer.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ const DatasetFileTransfer: React.FC<DatasetFileTransferProps> = ({
selectedFilesMap,
onSelectedFilesChange,
onDatasetSelect,
datasetTypeFilter = DatasetType.TEXT,
datasetTypeFilter,
...props
}) => {
const [datasets, setDatasets] = React.useState<Dataset[]>([]);
Expand Down Expand Up @@ -85,6 +85,7 @@ const DatasetFileTransfer: React.FC<DatasetFileTransferProps> = ({
page: datasetPagination.current,
size: datasetPagination.pageSize,
keyword: datasetSearch,
// 仅在显式传入过滤类型时才按类型过滤；否则后端返回所有类型
type: datasetTypeFilter,
});
setDatasets(data.content.map(mapDataset) || []);
Expand Down
21 changes: 6 additions & 15 deletions frontend/src/pages/DataAnnotation/Create/components/CreateAnnotationTaskDialog.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -183,16 +183,8 @@ export default function CreateAnnotationTask({
// 手动标注也支持跨数据集、精确到文件的选择
const selectedFiles = Object.values(selectedFilesMap) as any[];

const imageExtensions = [".jpg", ".jpeg", ".png", ".bmp", ".gif", ".tiff", ".webp"];
const imageFileIds = selectedFiles
.filter((file) => {
const ext = file.fileName?.toLowerCase().match(/\.[^.]+$/)?.[0] || "";
return imageExtensions.includes(ext);
})
.map((file) => file.id);

if (imageFileIds.length === 0) {
message?.error?.("请至少选择一个图像文件");
if (selectedFiles.length === 0) {
message?.error?.("请至少选择一个文件");
setSubmitting(false);
return;
}
Expand All @@ -213,7 +205,7 @@ export default function CreateAnnotationTask({
description: values.description,
datasetId: effectiveDatasetId,
templateId: values.templateId,
fileIds: imageFileIds,
fileIds: selectedFiles.map((file) => file.id),
};

await createAnnotationTaskUsingPost(requestData);
Expand Down Expand Up @@ -326,8 +318,8 @@ export default function CreateAnnotationTask({
label: "手动标注",
children: (
<Form form={manualForm} layout="vertical">
{/* 选择数据集和图像文件（支持多数据集、多文件） */}
<Form.Item label="选择数据集和图像文件" required>
{/* 选择数据集和文件（支持多数据集、多文件） */}
<Form.Item label="选择数据集和文件" required>
<DatasetFileTransfer
open
selectedFilesMap={selectedFilesMap}
Expand All @@ -346,12 +338,11 @@ export default function CreateAnnotationTask({
manualForm.setFieldsValue({ name: defaultName });
}
}}
datasetTypeFilter={DatasetType.IMAGE}
/>
{selectedDataset && (
<div className="mt-2 p-2 bg-blue-50 rounded border border-blue-200 text-xs">
当前数据集：<span className="font-medium">{selectedDataset.name}</span> - 已选择
<span className="font-medium text-blue-600"> {imageFileCount} </span>个图像文件
<span className="font-medium text-blue-600"> {Object.keys(selectedFilesMap).length} </span>个文件
</div>
)}
</Form.Item>
Expand Down
Loading
Loading