bvandeusen
b735432d02
Answers "how are videos/all media handled by the GPU worker": a job is per ITEM, but the agent fans a VIDEO into per-frame instances (ffmpeg in the agent, the existing cadence), each stored with a timestamp — so a video becomes a BAG of frame embeddings (fixes the mean-embedding muddle) instead of one washed-out vector. Stills → frame_time NULL; animated GIF/WebP treated like short video. - image_region.frame_time (migration 0061, not yet deployed so folded in): the source frame's seconds for video/animated media; NULL for stills. RegionService passes it through. A whole frame is just kind='frame'. - gpu_job + GpuJobService (migration 0062): the durable work list that keeps the desktop agent HTTP-only — enqueue (dedupes (image,task)) / lease (FOR UPDATE SKIP LOCKED, re-claims expired leases so the queue self-heals) / heartbeat / complete / fail (re-queues until MAX_ATTEMPTS then 'error'). The server enqueues; the agent leases+submits over the web API; Redis/Postgres stay private. Tests: enqueue dedupe, lease-then-skip-when-held, expired-lease reclaim, scoped heartbeat, complete, fail-requeue-then-error. region test now covers frame_time. NEXT: the thin HTTP API (lease/submit/heartbeat) + bearer-token auth, then the agent container + control UI. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Ttrj5P7upUTueSfoJcxEqa
60 lines
2.2 KiB
Python
60 lines
2.2 KiB
Python
"""image_region: detected/proposed regions + their crop embeddings (#114)
|
|
|
|
Storage backbone of the crop pipeline. A region = normalized bbox + the crop's
|
|
embedding (CCIP for face/figure → character id; SigLIP for concept regions →
|
|
head bag-of-embeddings). Also serves as grounded-tag bbox provenance.
|
|
|
|
Revision ID: 0061
|
|
Revises: 0060
|
|
Create Date: 2026-06-29
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from pgvector.sqlalchemy import Vector
|
|
|
|
revision: str = "0061"
|
|
down_revision: Union[str, None] = "0060"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
_CCIP_DIM = 768
|
|
_SIGLIP_DIM = 1152
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"image_region",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column(
|
|
"image_record_id", sa.Integer(),
|
|
sa.ForeignKey("image_record.id", ondelete="CASCADE"), nullable=False,
|
|
),
|
|
sa.Column("kind", sa.String(length=16), nullable=False),
|
|
# Video/animated: source frame timestamp (seconds); NULL for stills.
|
|
sa.Column("frame_time", sa.Float(), nullable=True),
|
|
sa.Column("rx", sa.Float(), nullable=False),
|
|
sa.Column("ry", sa.Float(), nullable=False),
|
|
sa.Column("rw", sa.Float(), nullable=False),
|
|
sa.Column("rh", sa.Float(), nullable=False),
|
|
sa.Column("score", sa.Float(), nullable=True),
|
|
sa.Column("detector_version", sa.String(length=64), nullable=True),
|
|
sa.Column("crop_version", sa.String(length=64), nullable=True),
|
|
sa.Column("embedding_version", sa.String(length=128), nullable=True),
|
|
sa.Column("ccip_embedding", Vector(_CCIP_DIM), nullable=True),
|
|
sa.Column("siglip_embedding", Vector(_SIGLIP_DIM), nullable=True),
|
|
sa.Column(
|
|
"created_at", sa.DateTime(timezone=True), nullable=False,
|
|
server_default=sa.func.now(),
|
|
),
|
|
)
|
|
op.create_index(
|
|
"ix_image_region_image_record_id", "image_region", ["image_record_id"],
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_image_region_image_record_id", table_name="image_region")
|
|
op.drop_table("image_region")
|