From 76fe87ac060fab8446571f59e761befd05d52141 Mon Sep 17 00:00:00 2001
From: root <root@truenas.local>
Date: Wed, 10 Jun 2026 19:51:09 -0400
Subject: [PATCH] automated update: 2026-06-10 19:51:09

---
 Dockerfile | 56 ++++++++++++++++++++++++++++++++++++++++++++++++------
 1 file changed, 50 insertions(+), 6 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index b1ae28d..b43d8dd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@ FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install core dependencies
+# Install dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     software-properties-common \
     && add-apt-repository ppa:deadsnakes/ppa -y \
@@ -13,22 +13,66 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && ln -sf /usr/bin/python3.12 /usr/bin/python \
     && ln -sf /usr/bin/python3.12 /usr/bin/python3
 
-# Install uv and move to path
+# Install uv to system path
 RUN curl -LsSf https://astral.sh/uv/install.sh | sh \
     && cp /root/.local/bin/uv /usr/local/bin/uv \
     && chmod 755 /usr/local/bin/uv
 
 WORKDIR /app
 
-# Clone, sync, and add dependencies
+# Clone repo and install dependencies
 RUN git clone --depth 1 https://github.com/sudolulo/if_curator_headless.git . \
     && uv sync --extra gpu \
     && uv add croniter \
     && uv cache clean
 
-# Copy scripts and modified CLI
+# Apply enhanced upload_to_frigate function in-place
+RUN sed -i '/def upload_to_frigate/,/if failed == total_files/c\
+def upload_to_frigate(jobs: list[dict]) -> None:\
+    """Upload processed face crops to Frigate via API with detailed logging."""\
+    frigate_url = os.environ.get("FRIGATE_URL", "")\
+    if not frigate_url:\
+        rprint("[yellow]⚠️  FRIGATE_URL not set, skipping upload.[/yellow]")\
+        return\
+\
+    rprint("\\n[bold cyan]📤 Uploading to Frigate[/bold cyan]")\
+    rprint(f"  Target: [dim]{frigate_url}[/dim]")\
+\
+    total_files = 0\
+    for job in jobs:\
+        name = job["person"]["name"]\
+        person_dir = os.path.join(Config.OUTPUT_DIR, name)\
+        if not os.path.isdir(person_dir): continue\
+        total_files += sum(1 for f in os.listdir(person_dir) if f.lower().endswith((".jpg", ".jpeg", ".png", ".webp")))\
+\
+    if total_files == 0:\
+        rprint("  [yellow]No images found to upload.[/yellow]")\
+        return\
+\
+    rprint(f"  People: [bold]{len(jobs)}[/bold], Total images: [bold]{total_files}[/bold]")\
+    uploaded, failed = 0, 0\
+\
+    with Progress(SpinnerColumn(), TextColumn("[progress.description]{task.description}"), BarColumn(), TaskProgressColumn(), console=console) as progress:\
+        upload_task = progress.add_task("[green]Uploading to Frigate", total=total_files)\
+        for job in jobs:\
+            name = job["person"]["name"]\
+            encoded_name = quote(name, safe="")\
+            person_dir = os.path.join(Config.OUTPUT_DIR, name)\
+            person_files = sorted(f for f in os.listdir(person_dir) if f.lower().endswith((".jpg", ".jpeg", ".png", ".webp"))) if os.path.isdir(person_dir) else []\
+            for fname in person_files:\
+                fpath = os.path.join(person_dir, fname)\
+                try:\
+                    with open(fpath, "rb") as f:\
+                        resp = requests.post(f"{frigate_url}/api/faces/train/{encoded_name}/classify", files={"file": (fname, f, "image/jpeg")}, timeout=30)\
+                    if resp.status_code == 200: uploaded += 1\
+                    else: failed += 1\
+                except Exception: failed += 1\
+                progress.advance(upload_task)\
+\
+    rprint("\\n  [bold]Frigate Upload Summary:[/bold]\\n    ✅ Succeeded: [green]{uploaded}[/green]\\n    ❌ Failed:    [red]{failed}[/red]")' /app/if_curator/cli.py
+
+# Copy scripts
 COPY entrypoint.sh scheduler.py /app/
-COPY cli.py /app/if_curator/cli.py
 RUN chmod +x /app/entrypoint.sh
 
 # Setup non-root user and persistent model paths
@@ -39,7 +83,7 @@ RUN groupadd -g 568 apps \
 
 USER appuser
 
-# Baked-in configurations: User only needs to mount /models
+# Baked-in configurations
 ENV FORCE_CPU=false \
     HF_HOME=/models/huggingface \
     INSIGHTFACE_HOME=/models