Remove obsolete implementation and simplify project structure

src/anyimage/server/media/__init__.py

@@ -0,0 +1,28 @@
+from .input import (
+    IMAGE_SUFFIXES,
+    VIDEO_SUFFIXES,
+    TaskCancelled,
+    check_cancelled,
+    collect_images,
+    copy_color_images,
+    copy_frames,
+    extract_video_frames,
+    first_frames,
+    prepare_background_input,
+    prepare_input,
+)
+
+
+__all__ = (
+    "IMAGE_SUFFIXES",
+    "VIDEO_SUFFIXES",
+    "TaskCancelled",
+    "check_cancelled",
+    "collect_images",
+    "copy_color_images",
+    "copy_frames",
+    "extract_video_frames",
+    "first_frames",
+    "prepare_background_input",
+    "prepare_input",
+)

src/anyimage/server/media/input.py

@@ -0,0 +1,168 @@
+import shutil
+import tempfile
+from pathlib import Path
+
+
+IMAGE_SUFFIXES = {".jpg", ".jpeg", ".png", ".webp"}
+VIDEO_SUFFIXES = {
+    ".mp4",
+    ".mov",
+    ".avi",
+    ".mkv",
+    ".webm",
+    ".m4v",
+    ".mts",
+    ".mpg",
+    ".mpeg",
+}
+class TaskCancelled(RuntimeError):
+    pass
+
+
+def check_cancelled(cancel_check):
+    if cancel_check is not None and cancel_check():
+        raise TaskCancelled("Task cancelled")
+
+
+def collect_images(directory):
+    directory = Path(directory)
+    return sorted(
+        path
+        for path in directory.iterdir()
+        if path.is_file() and path.suffix.lower() in IMAGE_SUFFIXES
+    )
+
+
+def first_frames(images, maximum_frames, start_frame=1):
+    """Return up to maximum_frames image paths in order from start_frame."""
+    if maximum_frames <= 0:
+        return images
+    start_index = max(start_frame - 1, 0)
+    return images[start_index:start_index + maximum_frames]
+
+
+def extract_video_frames(
+    video_path,
+    output_directory,
+    maximum_frames,
+    av,
+    start_frame=1,
+    cancel_check=None,
+):
+    output_directory = Path(output_directory)
+    output_directory.mkdir(parents=True, exist_ok=True)
+    try:
+        container = av.open(str(video_path))
+    except Exception as error:
+        raise RuntimeError(f"Unable to open video: {video_path}") from error
+    image_paths = []
+    skipped = 0
+    try:
+        stream = container.streams.video[0]
+        for frame in container.decode(stream):
+            check_cancelled(cancel_check)
+            if skipped < start_frame - 1:
+                skipped += 1
+                continue
+            image_path = output_directory / f"frame.{len(image_paths) + 1:04d}.jpg"
+            frame.to_image().save(image_path, format="JPEG")
+            image_paths.append(image_path)
+            if len(image_paths) >= maximum_frames:
+                break
+    finally:
+        container.close()
+
+    if not image_paths:
+        raise RuntimeError(f"Unable to read any video frame from: {video_path}")
+    return image_paths
+
+
+def prepare_input(
+    input_path,
+    video_frames,
+    temporary_root,
+    start_frame=1,
+    cancel_check=None,
+):
+    check_cancelled(cancel_check)
+    input_path = Path(input_path)
+    if input_path.is_dir():
+        images = collect_images(input_path)
+        if not images:
+            raise RuntimeError(f"No image frames found in: {input_path}")
+        return first_frames(images, video_frames, start_frame), None
+    suffix = input_path.suffix.lower()
+    if suffix in IMAGE_SUFFIXES:
+        return [input_path], None
+    if suffix not in VIDEO_SUFFIXES:
+        supported = ", ".join(sorted(IMAGE_SUFFIXES | VIDEO_SUFFIXES))
+        raise RuntimeError(f"Unsupported input type; expected one of: {supported}")
+
+    import av
+
+    temporary_directory = Path(
+        tempfile.mkdtemp(prefix="anyimage-video-", dir=temporary_root)
+    )
+    try:
+        images = extract_video_frames(
+            input_path,
+            temporary_directory,
+            video_frames,
+            av,
+            start_frame=start_frame,
+            cancel_check=cancel_check,
+        )
+    except Exception:
+        shutil.rmtree(temporary_directory, ignore_errors=True)
+        raise
+    return images, temporary_directory
+
+
+def prepare_background_input(
+    input_path,
+    video_frames,
+    temporary_root,
+    start_frame=1,
+    cancel_check=None,
+):
+    """Resolve a BEN2 input to frame images; returns (images, temporary_directory)."""
+    input_path = Path(input_path)
+    if input_path.is_dir():
+        images = collect_images(input_path)
+        if not images:
+            raise RuntimeError(f"No image frames found in: {input_path}")
+        return first_frames(images, video_frames, start_frame), None
+    if input_path.suffix.lower() in IMAGE_SUFFIXES:
+        return [input_path], None
+    return prepare_input(
+        input_path,
+        video_frames,
+        temporary_root,
+        start_frame=start_frame,
+        cancel_check=cancel_check,
+    )
+
+
+def copy_color_images(images, output_directory):
+    """Copy input images as the canonical numbered color layer."""
+    for index, image_path in enumerate(images):
+        suffix = Path(image_path).suffix.lower() or ".jpg"
+        destination = (
+            Path(output_directory) / f"color.{index + 1:04d}{suffix}"
+        )
+        if Path(image_path).resolve() == destination.resolve():
+            continue
+        shutil.copy2(image_path, destination)
+
+
+def copy_frames(images, output_directory):
+    """Copy multiple input images as numbered raw Debug frame files."""
+    if len(images) <= 1:
+        return
+    for index, image_path in enumerate(images):
+        suffix = Path(image_path).suffix.lower() or ".jpg"
+        destination = Path(output_directory) / f"frame.{index + 1:04d}{suffix}"
+        if Path(image_path).resolve() == destination.resolve():
+            continue
+        shutil.copy2(image_path, destination)
+