#!/usr/bin/env python3
"""TSLaw artikel 12: cari foto Pexels via jalur cadangan (r.jina.ai -> ID -> CDN), buat contact sheet.
Tema wajib: dokumen atau kantor hukum (berkas, meja kerja, rapat bisnis, tanda tangan)."""
import os, re, urllib.request
from PIL import Image

OUT = "/home/mantapman/.hermes/profiles/atun/workspace/tslaw_listikel"
os.makedirs(OUT, exist_ok=True)
UA = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 Chrome/124 Safari/537.36"}

QUERIES = ["business-meeting-documents-office", "corporate-documents-review-desk",
           "law-office-desk-documents", "audit-documents-paperwork", "contract-signing-business"]

ids = []
for q in QUERIES:
    url = f"https://r.jina.ai/https://www.pexels.com/search/{q}/"
    try:
        txt = urllib.request.urlopen(urllib.request.Request(url, headers=UA), timeout=150).read().decode("utf-8", "ignore")
    except Exception as e:
        print(f"gagal {q}: {str(e)[:80]}")
        continue
    found = re.findall(r"pexels-photo-(\d{4,})", txt)
    uniq = list(dict.fromkeys(found))
    print(f"{q}: {len(uniq)} id -> {uniq[:12]}")
    ids += [(q, i) for i in uniq[:6]]

print(f"\ntotal kandidat: {len(ids)}")
thumbs = []
for q, pid in ids:
    u = f"https://images.pexels.com/photos/{pid}/pexels-photo-{pid}.jpeg?auto=compress&cs=tinysrgb&w=400"
    try:
        data = urllib.request.urlopen(urllib.request.Request(u, headers=UA), timeout=90).read()
        p = f"{OUT}/_kandidat12_{pid}.jpg"
        open(p, "wb").write(data)
        thumbs.append((pid, p, q))
    except Exception as e:
        print(f"  lewat {pid}: {str(e)[:50]}")

if thumbs:
    cols, cw, ch = 5, 320, 182
    rows = (len(thumbs) + cols - 1) // cols
    sheet = Image.new("RGB", (cols * cw, rows * ch), "white")
    for n, (pid, p, q) in enumerate(thumbs):
        im = Image.open(p).convert("RGB")
        w, h = im.size
        r = cw / ch
        im = im.crop(((w - int(h * r)) // 2, 0, (w - int(h * r)) // 2 + int(h * r), h)) if w / h > r else \
             im.crop((0, (h - int(w / r)) // 2, w, (h - int(w / r)) // 2 + int(w / r)))
        sheet.paste(im.resize((cw, ch), Image.LANCZOS), ((n % cols) * cw, (n // cols) * ch))
    sheet.save(f"{OUT}/_contact_sheet_12.jpg", quality=85)
    print(f"\ncontact sheet: {OUT}/_contact_sheet_12.jpg")
    print("urutan ID:")
    for n, (pid, p, q) in enumerate(thumbs):
        print(f"  {n+1:>2}. {pid}  ({q})")