admin
4d65c37c73
- backend/schemas.py: TemplateUpdate 添加 classes/rules 字段 - backend/models.py: Template 添加 description 列 - backend/routers/templates.py: create/update 打包/解包 mapping_rules.classes (已有) - backend/main.py: seed 腹腔镜胆囊切除术35分类模板 - src/lib/api.ts: updateTemplate 改 PATCH，补齐 color/z_index，_mapTemplate 对齐 TS 接口 - src/store/useStore.ts: 新增 activeTemplateId/setActiveTemplateId - src/components/TemplateRegistry.tsx: 随机颜色(HSL轮盘)、HTML5拖拽排序、批量JSON导入、一键载入腹腔镜模板、handleSave 补齐必填字段 - src/components/OntologyInspector.tsx: 完全重写，从store读取模板，支持模板切换和自定义分类 - src/components/VideoWorkspace.tsx: 进入时自动加载模板列表 - src/components/ProjectLibrary.tsx: 修复状态字符串 TS 严格类型报错 - 工程分析/: 更新实现方案与经验记录 Timestamp: 20260430_222830
294 lines
10 KiB
Python
294 lines
10 KiB
Python
"""FastAPI application entrypoint."""
|
||
|
||
import asyncio
|
||
import logging
|
||
import os
|
||
import shutil
|
||
import tempfile
|
||
from contextlib import asynccontextmanager
|
||
|
||
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
|
||
from fastapi.middleware.cors import CORSMiddleware
|
||
|
||
from config import settings
|
||
from database import Base, engine, SessionLocal
|
||
from minio_client import ensure_bucket_exists, upload_file
|
||
from redis_client import ping as redis_ping
|
||
|
||
from routers import projects, templates, media, ai, export, auth
|
||
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format="%(asctime)s | %(levelname)s | %(name)s | %(message)s",
|
||
)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
DEFAULT_VIDEO_PATH = "/home/wkmgc/Desktop/Seg_Server/Data_MyVideo_1.mp4"
|
||
|
||
|
||
def _seed_default_project_sync() -> None:
|
||
"""Synchronously seed the default video project on first startup."""
|
||
import cv2
|
||
from models import Project, Frame
|
||
from services.frame_parser import parse_video, upload_frames_to_minio, extract_thumbnail
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
existing = db.query(Project).filter(Project.name == "Data_MyVideo_1").first()
|
||
if existing is not None:
|
||
return
|
||
|
||
if not os.path.exists(DEFAULT_VIDEO_PATH):
|
||
logger.warning("Default video not found at %s", DEFAULT_VIDEO_PATH)
|
||
return
|
||
|
||
project = Project(
|
||
name="Data_MyVideo_1",
|
||
description="默认演示视频",
|
||
status="pending",
|
||
source_type="video",
|
||
parse_fps=30.0,
|
||
)
|
||
db.add(project)
|
||
db.commit()
|
||
db.refresh(project)
|
||
|
||
with open(DEFAULT_VIDEO_PATH, "rb") as f:
|
||
data = f.read()
|
||
object_name = f"uploads/{project.id}/Data_MyVideo_1.mp4"
|
||
upload_file(object_name, data, content_type="video/mp4", length=len(data))
|
||
|
||
project.video_path = object_name
|
||
db.commit()
|
||
|
||
# Parse frames
|
||
tmp_dir = tempfile.mkdtemp(prefix=f"seg_seed_{project.id}_")
|
||
try:
|
||
local_path = os.path.join(tmp_dir, "video.mp4")
|
||
with open(local_path, "wb") as f:
|
||
f.write(data)
|
||
output_dir = os.path.join(tmp_dir, "frames")
|
||
os.makedirs(output_dir, exist_ok=True)
|
||
frame_files, original_fps = parse_video(local_path, output_dir, fps=30, max_frames=100)
|
||
project.original_fps = original_fps
|
||
|
||
# Extract thumbnail
|
||
thumbnail_path = os.path.join(tmp_dir, "thumbnail.jpg")
|
||
try:
|
||
extract_thumbnail(local_path, thumbnail_path)
|
||
with open(thumbnail_path, "rb") as f:
|
||
thumb_data = f.read()
|
||
thumb_object = f"projects/{project.id}/thumbnail.jpg"
|
||
upload_file(thumb_object, thumb_data, content_type="image/jpeg", length=len(thumb_data))
|
||
project.thumbnail_url = thumb_object
|
||
except Exception as exc: # noqa: BLE001
|
||
logger.warning("Thumbnail extraction failed: %s", exc)
|
||
|
||
object_names = upload_frames_to_minio(frame_files, project.id)
|
||
|
||
for idx, obj_name in enumerate(object_names):
|
||
img = cv2.imread(frame_files[idx])
|
||
h, w = img.shape[:2] if img is not None else (None, None)
|
||
frame = Frame(
|
||
project_id=project.id,
|
||
frame_index=idx,
|
||
image_url=obj_name,
|
||
width=w,
|
||
height=h,
|
||
)
|
||
db.add(frame)
|
||
|
||
project.status = "ready"
|
||
db.commit()
|
||
logger.info("Seeded default project id=%s with %d frames", project.id, len(object_names))
|
||
finally:
|
||
shutil.rmtree(tmp_dir, ignore_errors=True)
|
||
except Exception as exc:
|
||
logger.error("Failed to seed default project: %s", exc)
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
def _seed_default_templates_sync() -> None:
|
||
"""Seed default ontology templates on first startup."""
|
||
from models import Template
|
||
|
||
db = SessionLocal()
|
||
try:
|
||
if db.query(Template).first() is not None:
|
||
return
|
||
|
||
# Laparoscopic cholecystectomy template (35 classes)
|
||
colors = [
|
||
(134, 124, 118), (0, 157, 142), (245, 161, 0), (255, 172, 159), (146, 175, 236), (155, 62, 0),
|
||
(255, 91, 0), (255, 234, 0), (85, 111, 181), (155, 132, 0), (181, 227, 14), (72, 0, 255),
|
||
(255, 0, 255), (29, 32, 136), (240, 16, 116), (160, 15, 95), (0, 155, 33), (0, 160, 233),
|
||
(52, 184, 178), (66, 115, 82), (90, 120, 41), (255, 0, 0), (117, 0, 0), (167, 24, 233),
|
||
(42, 8, 66), (112, 113, 150), (0, 255, 0), (255, 255, 255), (0, 255, 255), (181, 85, 105),
|
||
(113, 102, 140), (202, 202, 200), (197, 83, 181), (136, 162, 196), (138, 251, 213),
|
||
]
|
||
names = [
|
||
'针', '线', '肿瘤', '血管阻断夹', '棉球', '双极电凝',
|
||
'肝脏', '胆囊', '分离钳', '脂肪', '止血海绵', '肝总管',
|
||
'吸引器', '剪刀', '超声刀', '止血纱布', '胆总管', '生物夹',
|
||
'无损伤钳', '钳夹', '喷洒', '胆囊管', '动脉', '电凝',
|
||
'静脉', '标本袋', '引流管', '纱布', '金属钛夹', '韧带',
|
||
'肝蒂', '推结器', '乳胶管-血管阻断', '吻合器', '术中超声',
|
||
]
|
||
classes = []
|
||
for idx, (rgb, name) in enumerate(zip(colors, names)):
|
||
color_hex = f"#{rgb[0]:02x}{rgb[1]:02x}{rgb[2]:02x}"
|
||
classes.append({
|
||
"id": f"cls-lap-{idx}",
|
||
"name": name,
|
||
"color": color_hex,
|
||
"zIndex": (len(names) - idx) * 10,
|
||
"category": "腹腔镜胆囊切除术",
|
||
})
|
||
|
||
template = Template(
|
||
name="腹腔镜胆囊切除术",
|
||
description="腹腔镜胆囊切除术（LC）手术器械与解剖结构语义分割模板，共35个分类",
|
||
color="#06b6d4",
|
||
z_index=0,
|
||
mapping_rules={"classes": classes, "rules": []},
|
||
)
|
||
db.add(template)
|
||
db.commit()
|
||
logger.info("Seeded default template '腹腔镜胆囊切除术' with %d classes", len(classes))
|
||
except Exception as exc:
|
||
logger.error("Failed to seed default templates: %s", exc)
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""Application lifespan: startup and shutdown hooks."""
|
||
# Startup
|
||
logger.info("Starting up SegServer backend...")
|
||
|
||
# Initialize database tables
|
||
try:
|
||
Base.metadata.create_all(bind=engine)
|
||
logger.info("Database tables initialized.")
|
||
except Exception as exc: # noqa: BLE001
|
||
logger.error("Database initialization failed: %s", exc)
|
||
|
||
# Check MinIO bucket
|
||
try:
|
||
ensure_bucket_exists()
|
||
except Exception as exc: # noqa: BLE001
|
||
logger.error("MinIO bucket check failed: %s", exc)
|
||
|
||
# Check Redis
|
||
if redis_ping():
|
||
logger.info("Redis connection OK.")
|
||
else:
|
||
logger.warning("Redis connection failed.")
|
||
|
||
# Seed default templates
|
||
try:
|
||
asyncio.create_task(asyncio.to_thread(_seed_default_templates_sync))
|
||
except Exception as exc: # noqa: BLE001
|
||
logger.error("Failed to start default template seeding: %s", exc)
|
||
|
||
# Seed default project in background thread so it doesn't block startup
|
||
try:
|
||
asyncio.create_task(asyncio.to_thread(_seed_default_project_sync))
|
||
except Exception as exc: # noqa: BLE001
|
||
logger.error("Failed to start default project seeding: %s", exc)
|
||
|
||
yield
|
||
|
||
# Shutdown
|
||
logger.info("Shutting down SegServer backend...")
|
||
engine.dispose()
|
||
|
||
|
||
app = FastAPI(
|
||
title="SegServer API",
|
||
description="Semantic Segmentation System Backend",
|
||
version="1.0.0",
|
||
lifespan=lifespan,
|
||
)
|
||
|
||
# CORS
|
||
app.add_middleware(
|
||
CORSMiddleware,
|
||
allow_origins=settings.cors_origins,
|
||
allow_credentials=True,
|
||
allow_methods=["*"],
|
||
allow_headers=["*"],
|
||
)
|
||
|
||
# Routers
|
||
app.include_router(auth.router)
|
||
app.include_router(projects.router)
|
||
app.include_router(templates.router)
|
||
app.include_router(media.router)
|
||
app.include_router(ai.router)
|
||
app.include_router(export.router)
|
||
|
||
|
||
@app.get("/health", tags=["Health"])
|
||
def health_check() -> dict:
|
||
"""Health check endpoint."""
|
||
return {"status": "ok", "service": "SegServer"}
|
||
|
||
|
||
# ---------------------------------------------------------------------------
|
||
# WebSocket: 实时进度推送
|
||
# ---------------------------------------------------------------------------
|
||
class ConnectionManager:
|
||
"""Manage WebSocket connections for progress broadcasting."""
|
||
|
||
def __init__(self):
|
||
self.active_connections: list[WebSocket] = []
|
||
|
||
async def connect(self, websocket: WebSocket):
|
||
await websocket.accept()
|
||
self.active_connections.append(websocket)
|
||
logger.info("WebSocket client connected. Total: %d", len(self.active_connections))
|
||
|
||
def disconnect(self, websocket: WebSocket):
|
||
if websocket in self.active_connections:
|
||
self.active_connections.remove(websocket)
|
||
logger.info("WebSocket client disconnected. Total: %d", len(self.active_connections))
|
||
|
||
async def broadcast(self, message: dict):
|
||
"""Broadcast a message to all connected clients."""
|
||
for connection in self.active_connections.copy():
|
||
try:
|
||
await connection.send_json(message)
|
||
except Exception as exc:
|
||
logger.warning("WebSocket send failed: %s", exc)
|
||
self.disconnect(connection)
|
||
|
||
|
||
manager = ConnectionManager()
|
||
|
||
|
||
@app.websocket("/ws/progress")
|
||
async def websocket_progress(websocket: WebSocket):
|
||
"""WebSocket endpoint for real-time parsing/AI progress updates."""
|
||
await manager.connect(websocket)
|
||
try:
|
||
while True:
|
||
# Receive client messages (heartbeat / subscription requests)
|
||
data = await websocket.receive_text()
|
||
logger.debug("WebSocket received: %s", data)
|
||
|
||
# Echo heartbeat to keep connection alive
|
||
await websocket.send_json({
|
||
"type": "status",
|
||
"status": "connected",
|
||
"message": "Progress stream active",
|
||
"timestamp": str(logging.time.time() if hasattr(logging, 'time') else __import__('time').time()),
|
||
})
|
||
except WebSocketDisconnect:
|
||
manager.disconnect(websocket)
|
||
except Exception as exc:
|
||
logger.error("WebSocket error: %s", exc)
|
||
manager.disconnect(websocket)
|