#!/usr/bin/env python3 """PROCITY facade-skin gap-fill — ported from thriftgod/gen_assets.py (same engine, same bible). Two backends, one prompt pack. The STYLE prefix and the blank-signboard FACADE template are lifted verbatim from thriftgod so new skins are indistinguishable from the 25 originals (Lane B overlays the shop name onto the blank sign). --local (DEFAULT CHOICE) MODELBEAST flux_local → FLUX.2-klein-4B on the M3 Ultra's MPS GPU. FREE, on-device, Apache/ungated (weights already cached). ~7s/img, one job at a time. (cloud) OpenRouter `google/gemini-3.1-flash-image`, ~$0.004/img. Needs OPENROUTER_KEY. python3 pipeline/gen_skins.py --local --dry-run # list the pack, no spend, no GPU python3 pipeline/gen_skins.py --local # generate all on-device → pipeline/.genraw/ python3 pipeline/gen_skins.py --local --only res # just the residential family python3 pipeline/gen_skins.py --harvest # copy approved .genraw/* → web/assets/gen/facade-*.jpg Everything here is a Step-4 GAP — the game already ships 25 facades and runs facade-free, so none of this blocks anyone. `--local` costs only electricity; cloud is the fallback if MODELBEAST is down. Review `.genraw/` by eye and delete rejects before `--harvest`. """ import base64, concurrent.futures, json, os, re, sys, time, urllib.request, shutil ROOT = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) RAW = os.path.join(ROOT, "pipeline", ".genraw") GEN = os.path.join(ROOT, "web", "assets", "gen") os.makedirs(RAW, exist_ok=True) MODEL = os.environ.get("TG_ART_MODEL", "google/gemini-3.1-flash-image") KEY = os.environ.get("OPENROUTER_KEY") if not KEY: # house convention — key never committed kf = os.path.expanduser("~/Documents/dealgod/lore/keys.env") if os.path.exists(kf): for line in open(kf): if line.startswith("OPENROUTER_KEY="): KEY = line.strip().split("=", 1)[1] # ── the style bible (verbatim from thriftgod gen_assets.py:20) ────────────────────────── STYLE = ("1990s Australian op-shop aesthetic, slightly faded and worn, warm fluorescent " "lighting, kitsch but loving, no text unless specified, no watermarks. ") # ── the blank-signboard shop-facade template (verbatim, gen_assets.py:69-73) ──────────── FACADE = ("Straight-on photograph of a small single-storey Australian suburban shop facade, " "{v}, blank empty signboard above the entrance, glass shop window and a door, " "no text or lettering anywhere, overcast daylight, flat front elevation filling " "the entire frame edge to edge, nothing else visible") # ── wide anchor + windowless side-wall templates (same discipline, different geometry) ─── WIDE = ("Straight-on photograph of a WIDE two-storey Australian department-store anchor " "building facade, {v}, a long blank signboard band across the top, rows of ground-floor " "display windows and a central entrance, no text or lettering anywhere, overcast daylight, " "flat front elevation filling the entire frame edge to edge, nothing else visible") SIDE = ("Straight-on photograph of the plain windowless SIDE wall of an Australian shop building, " "{v}, no shopfront, no door, a faded blank rectangular ghost-sign area, no text or lettering " "anywhere, overcast daylight, flat elevation filling the entire frame edge to edge") HOUSE = ("Straight-on photograph of a modest Australian suburban house front, {v}, front path and " "a bit of fence, no text anywhere, overcast daylight, flat front elevation filling the " "entire frame edge to edge, nothing else visible") PROMPTS = { # ── shop-type facades that the registry is thin on ────────────────────────────────── **{f"facade-{k}": FACADE.format(v=v) for k, v in { "toy-brights": "cheerful toy shop, primary-colour painted render, big low display window", "toy-lolly": "old-fashioned lolly and toy shop, pastel candy-stripe painted front", "book-barn": "a crammed secondhand book barn, tall dusty windows stacked with books, dark timber", "book-scholarly": "a genteel antiquarian bookshop, dark green paint, gold-line window, brass fittings", "video-neon": "a 1990s video rental store, glossy red and blue plastic signage band, poster light-boxes (all blank)", "video-suburban": "a suburban strip video shop, cream brick, glass double doors, returns chute", "milkbar-corner": "a classic corner milk bar, cream and green wall tiles, wide wrap-around window, ice-cream awning", "milkbar-fibro": "a little fibro milk bar with a fly-screen door, a bench out front, faded soft-drink colours", "milkbar-deco": "a streamline-deco milk bar, curved rendered corner, chrome trim, pastel mint paint", "arcade-entry": "a covered pedestrian shopping arcade entrance, tiled archway, terrazzo floor, patterned ceiling", "warehouse-roller": "an industrial discount warehouse shop, red brick with a big corrugated steel roller door, blank banner", "warehouse-tin": "a corrugated-tin clearance shed shopfront, roller door half up, concrete apron", "market-fruit": "an open fruit-and-veg shopfront, empty produce crates and astro-turf steps, striped awning", "market-edge": "a row of humble market-square edge shopfronts under one shared posted verandah, mixed paint colours", }.items()}, # ── the WIDE dept-store anchor (needs a wide UV plane in Lane B) ───────────────────── **{f"facade-{k}": WIDE.format(v=v) for k, v in { "dept-anchor": "mid-century mustard-render and glass, cantilevered awning, grand central doors", "dept-emporium": "a federation-era brick emporium, arched upper windows, ornate parapet", }.items()}, # ── corner-lot second faces: the `-side` convention (see manifest conventions) ────── **{f"facade-{k}-side": SIDE.format(v=v) for k, v in { "brick": "painted red brick, a downpipe, one small high window", "weatherboard": "cream weatherboard boards, a rain downpipe", "render": "weathered pastel painted render with hairline cracks", }.items()}, # ── residential terrace/cottage fronts (lots with use:'house') ────────────────────── **{f"facade-{k}": HOUSE.format(v=v) for k, v in { "res-terrace": "a single-fronted Victorian terrace, iron-lace verandah, dark brick, tiny front garden", "res-weatherboard": "a modest weatherboard cottage, bull-nose verandah, picket fence", "res-fibro": "a 1950s cream fibro house, hip roof, small steel-framed windows", "res-brickveneer": "a 1970s brown brick-veneer house, aluminium sliding windows, carport edge", }.items()}, } def gen(slug, prompt): body = json.dumps({"model": MODEL, "modalities": ["image", "text"], "messages": [{"role": "user", "content": STYLE + prompt}]}).encode() req = urllib.request.Request("https://openrouter.ai/api/v1/chat/completions", data=body, headers={"Authorization": f"Bearer {KEY}", "Content-Type": "application/json"}) with urllib.request.urlopen(req, timeout=120) as r: d = json.load(r) msg = d["choices"][0]["message"] imgs = msg.get("images") or [] url = imgs[0]["image_url"]["url"] if imgs else "" m = re.match(r"data:image/(\w+);base64,(.+)", url, re.S) if not m: raise RuntimeError(f"no image in response: {str(msg)[:120]}") fn = os.path.join(RAW, f"{slug}.{m.group(1)}") with open(fn, "wb") as f: f.write(base64.b64decode(m.group(2))) return fn # ── LOCAL backend: MODELBEAST flux_local (mflux / FLUX.2-klein-4B, MPS) — FREE, ~7s/img, ungated ── import subprocess, shutil as _sh MB = os.environ.get("MB_HOME", os.path.expanduser("~/Documents/MODELBEAST")) FLUX_PY = os.path.join(MB, "venvs/mflux/bin/python") FLUX_RUN = os.path.join(MB, "server/operators/flux_local/run.py") def local_available(): return os.path.exists(FLUX_PY) and os.path.exists(FLUX_RUN) def gen_local(slug, prompt, w=1024, h=576, steps=4, seed=7): outdir = os.path.join(RAW, "_job_" + slug) os.makedirs(outdir, exist_ok=True) params = json.dumps({"prompt": STYLE + prompt, "model": "flux2-klein-4b", "steps": steps, "width": w, "height": h, "seed": seed, "quantize": "8", "guidance": 3.5}) r = subprocess.run([FLUX_PY, FLUX_RUN, "--outdir", outdir, "--params", params], capture_output=True, text=True, timeout=300) pngs = [p for p in os.listdir(outdir) if p.endswith(".png")] if not pngs: raise RuntimeError((r.stderr or r.stdout or "no png")[-160:]) fn = os.path.join(RAW, f"{slug}.png") _sh.move(os.path.join(outdir, pngs[0]), fn) _sh.rmtree(outdir, ignore_errors=True) return fn def have(slug): return any(os.path.exists(os.path.join(RAW, f"{slug}.{e}")) for e in ("png", "jpeg", "jpg", "webp")) LOCAL = False # set by --local: use MODELBEAST flux_local instead of paid OpenRouter def gen_one(item): slug, prompt = item for attempt in (1, 2): try: fn = gen_local(slug, prompt) if LOCAL else gen(slug, prompt) return slug, os.path.getsize(fn) // 1024, None except Exception as e: if attempt == 2: return slug, 0, str(e)[:90] time.sleep(3) def harvest(): """Copy reviewed .genraw/*.{png,jpeg} into web/assets/gen/ as *.jpg. Manual review first — only files you keep in .genraw/ get harvested. Uses sips if available for real JPEG.""" n = 0 for f in sorted(os.listdir(RAW)): slug, ext = os.path.splitext(f) if ext.lower() not in (".png", ".jpeg", ".jpg", ".webp"): continue dst = os.path.join(GEN, f"{slug}.jpg") src = os.path.join(RAW, f) if ext.lower() in (".jpg", ".jpeg"): shutil.copyfile(src, dst) else: if os.system(f"sips -s format jpeg {src!r} --out {dst!r} >/dev/null 2>&1") != 0: shutil.copyfile(src, dst) # fallback: copy as-is (rename) n += 1 print(f" harvested {slug}.jpg") print(f"harvested {n} → {GEN} (re-run build_manifest.py to catalogue them)") if __name__ == "__main__": if "--harvest" in sys.argv: harvest(); sys.exit(0) LOCAL = "--local" in sys.argv # MODELBEAST flux_local: on-device, free, ungated only = sys.argv[sys.argv.index("--only") + 1] if "--only" in sys.argv else "" # local = one GPU job at a time (MPS won't share cleanly); cloud fans out default_workers = 1 if LOCAL else 4 workers = int(sys.argv[sys.argv.index("--workers") + 1]) if "--workers" in sys.argv else default_workers todo = {s: p for s, p in PROMPTS.items() if (not only or only in s) and not have(s)} backend = "MODELBEAST flux2-klein-4b, local·free" if LOCAL else MODEL cost = 0.0 if LOCAL else len(todo) * 0.004 print(f"{len(todo)} skins to generate ({backend}) ≈ ${cost:.2f}" + (f" filter:*{only}*" if only else "")) if "--dry-run" in sys.argv: for s in sorted(todo): print(f" {s}") sys.exit(0) if LOCAL: if not local_available(): print(f"ERROR: MODELBEAST flux_local missing ({FLUX_RUN}).\n" "Set MB_HOME or check ~/Documents/MODELBEAST/server/operators/flux_local/.") sys.exit(2) elif not KEY: print("ERROR: OPENROUTER_KEY not set and no ~/Documents/dealgod/lore/keys.env.\n" "Use --local (MODELBEAST, free) or route 2: pipeline/SKIN_PROMPTS.md.") sys.exit(2) fails = 0 with concurrent.futures.ThreadPoolExecutor(max_workers=workers) as ex: for i, (slug, kb, err) in enumerate(ex.map(gen_one, todo.items()), 1): if err: fails += 1; print(f"[{i}/{len(todo)}] {slug} FAILED: {err}") else: print(f"[{i}/{len(todo)}] {slug} ({kb}KB)") print(f"done, {fails} failures. Review .genraw/, delete rejects, then --harvest.")