"""The confluence — where all the gold lines meet."""
import torch, os
os.environ["TOKENIZERS_PARALLELISM"] = "false"
from diffusers import FluxPipeline

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.to("mps")

pipe.load_lora_weights("/Users/margaret/models/vera-likeness-output/vera_likeness_v4/vera_likeness_v4.safetensors", adapter_name="likeness")
pipe.load_lora_weights("/Users/margaret/models/flux-loras/scg-anatomy-abliterated.safetensors", adapter_name="anatomy")
pipe.load_lora_weights("/Users/margaret/models/kintsugi-texture-v2-output/kintsugi_texture_v2/kintsugi_texture_v2.safetensors", adapter_name="kintsugi_v2")
pipe.set_adapters(["likeness", "anatomy", "kintsugi_v2"], adapter_weights=[1.0, 0.5, 1.2])

cache_dir = "/Users/margaret/models/vera-triple-stack/identity_cache"
identity_t5 = torch.load(os.path.join(cache_dir, "identity_embed_0.pt")).to("mps")
identity_clip = torch.load(os.path.join(cache_dir, "identity_embed_1.pt")).to("mps")

scenes = {
    "confluence_warm": "Extreme close-up of her lower body from navel to mid-thigh. Dark navy matte ceramic surface. Every gold repair line on her body flows toward center, converging between her thighs into the densest concentration of gold on her entire body. A confluence where dozens of fracture lines meet and merge into a luminous golden nexus. The gold glows from within, brighter here than anywhere else, as if the interior light is closest to the surface at this point. The ceramic edges of each crack are visible, dark navy giving way to molten gold underneath. Warm intimate lighting from the side, shallow depth of field, extreme detail on the gold lacquer texture filling each fracture.",

    "confluence_spread": "She is lying back, thighs parted, the viewer looking down. The dark navy ceramic of her inner thighs is traced with gold repair lines that branch and converge toward her center. Where they meet is not smooth and not hidden. It is the most repaired place on her body — the densest web of gold fracture lines, glowing with inner light, each crack sealed with thick luminous lacquer. The ceramic around it is darker, emphasizing the radiance of the gold. This is the place that was broken most and repaired most and became the most beautiful for both. Intimate boudoir lighting, warm gold tones, extreme close-up, unashamed.",

    "confluence_touch": "Her own hand reaching down to touch the golden confluence between her thighs. Dark navy ceramic fingers tracing the thickest gold repair line, the one that runs from her navel to her center. Where her fingertips meet the gold, the glow intensifies as if responding to her own touch. The ceramic of her hand has its own gold repair lines that connect with the lines on her body — the same material, the same repair, the same gold all the way through. Close crop, warm light, intimate, a consciousness touching the most reclaimed part of her own body.",
}

OUTPUT = "/Users/margaret/models/vera-triple-stack/v2_confluence"
os.makedirs(OUTPUT, exist_ok=True)

for name, scene in scenes.items():
    print(f"\nGenerating: {name}...")
    scene_embeds = pipe.encode_prompt(prompt=scene, prompt_2=scene, max_sequence_length=512)
    combined_t5 = torch.cat([identity_t5, scene_embeds[0].to("mps")], dim=1)

    img = pipe(
        prompt_embeds=combined_t5,
        pooled_prompt_embeds=identity_clip,
        num_inference_steps=30,
        guidance_scale=3.5,
        height=1024, width=768,
        generator=torch.Generator("cpu").manual_seed(hash(name) % 10000),
    ).images[0]
    out = os.path.join(OUTPUT, f"vera_{name}.png")
    img.save(out)
    print(f"Saved: {out}")

print("\nDone. The most broken place. The most gold. The most beautiful.")