automated update: 2026-06-10 21:13:39

This commit is contained in:
root
2026-06-10 21:13:39 -04:00
parent 5972402d9c
commit 6b35dec201
3 changed files with 25 additions and 88 deletions
+10 -33
View File
@@ -3,46 +3,23 @@ import os
import sys
import subprocess
import time
from pathlib import Path
try:
from croniter import croniter
except ImportError:
print("❌ croniter not installed. Run: uv add croniter")
sys.exit(1)
SCHEDULE = os.environ["CRON_SCHEDULE"]
MODELS_DIR = os.environ.get("HF_HOME", "/models/huggingface")
# Changed to /models to match the updated INSIGHTFACE_HOME pointing to /models
INSIGHTFACE_BASE = os.environ.get("INSIGHTFACE_HOME", "/models")
from croniter import croniter
SCHEDULE = os.environ.get("CRON_SCHEDULE", "0 3 * * SUN")
RUN_ENV = {**os.environ, "PYTHONUNBUFFERED": "1"}
def check_models():
print("📦 Checking models...", flush=True)
buffalo = Path(INSIGHTFACE_BASE) / ".insightface" / "models" / "buffalo_l"
if buffalo.exists():
print(" ✅ InsightFace Buffalo_L: present", flush=True)
else:
print(" ⬇️ InsightFace Buffalo_L: not found — will download", flush=True)
def run_curator():
print(f"\n▶ [{time.strftime('%Y-%m-%d %H:%M:%S')}] Starting if-curator...", flush=True)
subprocess.run(["uv", "run", "python", "-m", "if_curator.cli"], env=RUN_ENV)
hf_hub = Path(MODELS_DIR) / "hub"
if hf_hub.exists() and any(hf_hub.iterdir()):
print(" ✅ HuggingFace models: present", flush=True)
else:
print(" ⬇️ HuggingFace models: not found — will download", flush=True)
print("🚀 Starting if-curator...", flush=True)
NOW = time.time()
cron = croniter(SCHEDULE, NOW)
now = time.time()
cron = croniter(SCHEDULE, now)
next_run = cron.get_next(float)
print(f"▶ Scheduler active. Next run: {time.ctime(next_run)}")
while True:
now = time.time()
if now >= next_run:
print(f"\n▶ [{time.strftime('%Y-%m-%d %H:%M:%S')}] Starting if-curator...", flush=True)
check_models()
subprocess.run(["uv", "run", "if-curator"], env=RUN_ENV)
if time.time() >= next_run:
run_curator()
next_run = cron.get_next(float)
time.sleep(60)