Files
Pre_Seg_Server/backend/worker_tasks.py
admin c8c59f7ede feat: 完善分割工作区传播与交互闭环
功能增加:新增后端传播任务执行器,支持异步自动传播、传播进度、结果统计、取消/重试状态同步。

功能增加:传播请求支持指定 SAM2.1 tiny/small/base+/large 权重,并记录 seed mask、source annotation 和传播范围。

功能增加:传播逻辑增加 seed 签名,未变化的 mask 二次传播会跳过,已变化的 mask 会先清理旧自动传播结果再重新生成,避免重复重叠。

功能增加:工作区增加传播范围二次选择、传播进度提示、人工/AI 标注帧红色标识、自动传播帧蓝色标识和当前帧双层边框。

功能增加:新增临时提示组件,让工具操作提示自动消失且不阻塞后续操作。

功能增加:补充项目删除、模板删除、任务失败详情、任务取消/重试等前后端联动状态。

功能增加:新增安装部署文档,补充当前需求冻结、设计冻结、接口契约、测试计划和 AGENTS/README 项目说明。

Bugfix:修复自动传播接口 404、传播后看不到任务进度、传播结果重复堆叠和已编辑帧提示不清晰的问题。

Bugfix:修复 AI 分割框选/点选交互、单候选 mask、删除选点、工作区保存与候选 mask 推送相关问题。

Bugfix:修复 Canvas 多边形顶点拖动告警、工具栏提示缺失、项目库 FPS 展示和若干 UI 文案/可用性问题。

测试:补充 AI 分割、Canvas、Dashboard、FrameTimeline、ProjectLibrary、TemplateRegistry、ToolsPalette、VideoWorkspace、API 和后端任务/AI/dashboard 测试。

验证:npm run lint;npm run test:run;python -m pytest backend/tests -q。
2026-05-02 05:17:18 +08:00

37 lines
1.1 KiB
Python

"""Celery task definitions."""
import logging
from celery_app import celery_app
from database import SessionLocal
from services.media_task_runner import run_parse_media_task
from services.propagation_task_runner import run_propagate_project_task
logger = logging.getLogger(__name__)
@celery_app.task(name="media.parse_project")
def parse_project_media(task_id: int) -> dict:
"""Run media parsing for one queued task."""
db = SessionLocal()
try:
return run_parse_media_task(db, task_id)
except Exception as exc: # noqa: BLE001
logger.exception("Parse media task failed: task_id=%s", task_id)
raise exc
finally:
db.close()
@celery_app.task(name="ai.propagate_project")
def propagate_project_masks(task_id: int) -> dict:
"""Run SAM video propagation for one queued task."""
db = SessionLocal()
try:
return run_propagate_project_task(db, task_id)
except Exception as exc: # noqa: BLE001
logger.exception("Propagation task failed: task_id=%s", task_id)
raise exc
finally:
db.close()