完善项目导入、模板与分割工作区交互

- 增强 DICOM/视频项目导入与演示数据：DICOM 按文件名自然顺序处理，导入后展示上传与解析任务进度，恢复演示出厂设置保留演示视频和演示 DICOM 项目，并补充 demo media seed 逻辑。 - 完善项目管理：项目支持重命名、删除、复制，删除使用站内确认弹窗，复制支持新项目重置和全内容复制，DICOM 项目不显示生成帧入口。 - 完善 GT Mask 与导出链路：只支持 8-bit maskid 图导入，非法/全背景图明确拒绝，尺寸自动适配，高精度 polygon 回显；统一导出默认当前帧，GT_label 使用 uint8 和真实 maskid，待分类 maskid 0 与背景一致。 - 完善分割工作区交互：新增画笔和橡皮擦并支持尺寸控制，移除创建点/线段入口，工具栏按类别分隔，AI 智能分割使用明确 AI 图标，取消黄色 seed point，清空/删除传播 mask 后同步清理空帧时间轴状态。 - 完善传播与时间轴：自动传播使用 SAM 2.1 权重任务，参考帧无遮罩时提示，传播历史按同一蓝色系递进变暗，删除/清空传播链时保留人工或独立 AI 标注来源。 - 完善模板库：新增头颈部 CT 分割默认模板，所有模板保留 maskid 0 待分类，支持鼠标复制模板、拖拽层级、JSON 批量导入预览、删除 label 和站内删除确认。 - 完善用户与高风险确认：用户改密码、删除用户、恢复演示出厂设置和清空人工/AI 标注帧均改为站内确认交互，避免浏览器原生 prompt/confirm。 - 补充前后端测试与文档：更新项目、模板、GT 导入、导出、传播、DICOM、用户管理等测试，并同步 README、AGENTS 和 doc 下实现/契约/测试计划文档。
2026-05-03 17:11:59 +08:00
parent afcddfaeb9
commit 481ffa5b67
47 changed files with 3650 additions and 676 deletions
--- a/backend/routers/projects.py
+++ b/backend/routers/projects.py
@@ -7,15 +7,38 @@ from fastapi import APIRouter, Depends, HTTPException, status
 from sqlalchemy.orm import Session

 from database import get_db
-from models import Project, Frame, User
+from models import Annotation, Mask, Project, Frame, User
 from routers.auth import get_current_user, require_editor
-from schemas import ProjectCreate, ProjectOut, ProjectUpdate, FrameCreate, FrameOut
+from schemas import ProjectCopyRequest, ProjectCreate, ProjectOut, ProjectUpdate, FrameCreate, FrameOut
 from minio_client import get_presigned_url

 logger = logging.getLogger(__name__)
 router = APIRouter(prefix="/api/projects", tags=["Projects"])


+def _next_project_copy_name(db: Session, owner_user_id: int, source_name: str) -> str:
+    base_name = f"{source_name} 副本"
+    existing_names = {
+        row[0]
+        for row in db.query(Project.name)
+        .filter(Project.owner_user_id == owner_user_id, Project.name.like(f"{base_name}%"))
+        .all()
+    }
+    if base_name not in existing_names:
+        return base_name
+    suffix = 2
+    while f"{base_name} {suffix}" in existing_names:
+        suffix += 1
+    return f"{base_name} {suffix}"
+
+
+def _prepare_project_response(project: Project) -> Project:
+    project.frame_count = len(project.frames)
+    if project.thumbnail_url:
+        project.thumbnail_url = get_presigned_url(project.thumbnail_url, expires=3600)
+    return project
+
+
 # ---------------------------------------------------------------------------
 # Projects
 # ---------------------------------------------------------------------------
@@ -59,9 +82,7 @@ def list_projects(
        .all()
    )
    for p in projects:
-        p.frame_count = len(p.frames)
-        if p.thumbnail_url:
-            p.thumbnail_url = get_presigned_url(p.thumbnail_url, expires=3600)
+        _prepare_project_response(p)
    return projects


@@ -82,10 +103,85 @@ def get_project(
    ).first()
    if not project:
        raise HTTPException(status_code=404, detail="Project not found")
-    project.frame_count = len(project.frames)
-    if project.thumbnail_url:
-        project.thumbnail_url = get_presigned_url(project.thumbnail_url, expires=3600)
-    return project
+    return _prepare_project_response(project)
+
+
+@router.post(
+    "/{project_id}/copy",
+    response_model=ProjectOut,
+    status_code=status.HTTP_201_CREATED,
+    summary="Copy a project",
+)
+def copy_project(
+    project_id: int,
+    payload: ProjectCopyRequest,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(require_editor),
+) -> Project:
+    """Copy a project. Reset copies media/frame sequence; full also copies annotations and mask metadata."""
+    source = db.query(Project).filter(
+        Project.id == project_id,
+        Project.owner_user_id == current_user.id,
+    ).first()
+    if not source:
+        raise HTTPException(status_code=404, detail="Project not found")
+
+    next_name = (payload.name or "").strip() if payload.name is not None else ""
+    if not next_name:
+        next_name = _next_project_copy_name(db, current_user.id, source.name)
+
+    copied = Project(
+        name=next_name,
+        description=source.description,
+        video_path=source.video_path,
+        thumbnail_url=source.thumbnail_url,
+        status=source.status,
+        source_type=source.source_type,
+        original_fps=source.original_fps,
+        parse_fps=source.parse_fps,
+        owner_user_id=current_user.id,
+    )
+    db.add(copied)
+    db.flush()
+
+    frame_id_map: dict[int, int] = {}
+    for frame in sorted(source.frames, key=lambda item: item.frame_index):
+        copied_frame = Frame(
+            project_id=copied.id,
+            frame_index=frame.frame_index,
+            image_url=frame.image_url,
+            width=frame.width,
+            height=frame.height,
+            timestamp_ms=frame.timestamp_ms,
+            source_frame_number=frame.source_frame_number,
+        )
+        db.add(copied_frame)
+        db.flush()
+        frame_id_map[frame.id] = copied_frame.id
+
+    if payload.mode == "full":
+        for annotation in sorted(source.annotations, key=lambda item: item.id):
+            copied_annotation = Annotation(
+                project_id=copied.id,
+                frame_id=frame_id_map.get(annotation.frame_id) if annotation.frame_id is not None else None,
+                template_id=annotation.template_id,
+                mask_data=annotation.mask_data,
+                points=annotation.points,
+                bbox=annotation.bbox,
+            )
+            db.add(copied_annotation)
+            db.flush()
+            for mask in annotation.masks:
+                db.add(Mask(
+                    annotation_id=copied_annotation.id,
+                    mask_url=mask.mask_url,
+                    format=mask.format,
+                ))
+
+    db.commit()
+    db.refresh(copied)
+    logger.info("Copied project id=%s to id=%s mode=%s", project_id, copied.id, payload.mode)
+    return _prepare_project_response(copied)


@router.patch(
@@ -108,6 +204,10 @@ def update_project(
        raise HTTPException(status_code=404, detail="Project not found")

    for key, value in payload.model_dump(exclude_unset=True).items():
+        if key == "name":
+            value = (value or "").strip()
+            if not value:
+                raise HTTPException(status_code=400, detail="Project name is required")
        setattr(project, key, value)

    db.commit()