src/preprocess/pipeline.py

"""预处理层入口"""
from dataclasses import dataclass, field
from typing import List
from PIL import Image

from .scene_classifier import classify_scene, SceneResult
from .image_processor import process


@dataclass
class PreprocessResult:
    scene_result: SceneResult
    images: List[Image.Image]   # 切片后的图片列表
    warnings: List[str] = field(default_factory=list)


def run(img: Image.Image, fmt: str) -> PreprocessResult:
    scene_result = classify_scene(img, fmt)
    result = process(img, scene_result)
    return PreprocessResult(
        scene_result=scene_result,
        images=result["images"],
        warnings=result["warnings"],
    )