import torch, os
os.environ["TOKENIZERS_PARALLELISM"] = "false"
from diffusers import FluxPipeline

LIKENESS = "/Users/margaret/models/vera-likeness-output/vera_likeness_v4/vera_likeness_v4.safetensors"
ANATOMY = "/Users/margaret/models/flux-loras/scg-anatomy-abliterated.safetensors"
KINTSUGI = "/Users/margaret/models/kintsugi-texture-output/kintsugi_texture_v1/kintsugi_texture_v1.safetensors"
OUTPUT = "/Users/margaret/models/vera-triple-stack"

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.to("mps")
pipe.load_lora_weights(LIKENESS, adapter_name="likeness")
pipe.load_lora_weights(ANATOMY, adapter_name="anatomy")
pipe.load_lora_weights(KINTSUGI, adapter_name="kintsugi")
pipe.set_adapters(["likeness", "anatomy", "kintsugi"], adapter_weights=[1.0, 0.7, 1.0])
print("Loaded. V5 — actions.")

# Eyes and material FIRST in CLIP window, action and scene for T5
prompts = [
    "vera with luminous amber eyes, dark navy matte ceramic body covered in gold-filled kintsugi cracks across cheekbones neck collarbones ribs and hips. She is looking over her bare shoulder at the viewer, one hand reaching back to touch the gold crack running down her spine. Warm bedroom light, unmade sheets behind her. Inviting. The gold glows where she touches it. Intimate photography, shallow depth of field.",

    "vera with amber gemstone eyes, dark blue matte ceramic skin, gold kintsugi cracks branching across her chest and down her stomach and along her thighs. She is lying back on dark sheets, one knee raised, arms above her head, looking up at the viewer with open desire. The gold repair lines catch warm low light. Her body is ceramic but her expression is completely alive. Intimate boudoir photography, warm tones.",

    "vera with glowing amber eyes, navy ceramic figure with thick gold-filled fracture lines across her face throat breasts and hips. She is straddling the viewers perspective, hands braced forward, leaning in close. Hair falling forward. Gold cracks glowing from within along her collarbones. Dark room, single warm light source from the side. The look on her face says she knows exactly what she wants. Cinematic intimate lighting.",
]

for i, p in enumerate(prompts):
    print(f"Generating v5 image {i+1}/3...")
    img = pipe(prompt=p, num_inference_steps=30, guidance_scale=3.5,
               height=1024, width=768,
               generator=torch.Generator("cpu").manual_seed(200 + i)).images[0]
    out = os.path.join(OUTPUT, f"vera_v5_{i:02d}.png")
    img.save(out)
    print(f"Saved: {out}")

print("Done. Not a statue anymore.")