The Creative Volume Engine
100 Ads/Day Per Offer

{
  "mechanism": "Neurotoxic acid accumulates after 60, dissolves myelin sheath like battery fluid on bare wire. Pink Himalayan salt has 84 trace minerals that bond with this acid and neutralize it.",
  "target_demo": "US/CA adults 55-75, diabetic or pre-diabetic, on Gabapentin or Lyrica, waking up at 3AM with burning feet or electric shocks",
  "nightmare": "Losing the ability to walk, needing a cane permanently, falling and breaking a hip, ending up in a nursing home, becoming a burden on family",
  "ideal_state": "Sleeping through the night, walking to the mailbox pain-free, throwing prescription bottles in the trash, playing with grandkids again",
  "prior_failures": "Gabapentin (made it worse), Lyrica (side effects), neurologists ($14K spent), physical therapy (temporary), creams and ointments",
  "emotional_triggers": "Rage at doctors who dismissed them, shame at being on pills, hope from finally finding something simple, fear of losing independence",
  "authority_anchors": "Dr. Oz, 60 Minutes, CNN, Harvard, MIT researchers, Dr. Lee (world-renowned neurologist), private neuropathy forums",
  "positioning": "$2.89 at Walmart, weighs less than a sugar packet, 7-second morning ritual, biblical pink salt trick, pink water before bed"
}

"""
creative_volume_engine.py — 100 nerve pain image ads/day
Runs unattended. Fires all 3 campaigns in parallel.
"""
import json, asyncio, httpx, anthropic, google.generativeai as genai
from pathlib import Path
from datetime import date
from PIL import Image, ImageDraw, ImageFont

FRAGMENTS = json.loads(Path("fragments.json").read_text())
KIE_KEY   = "2a553737acbc34b4403d59f5a250bbfc"
TODAY     = date.today().isoformat()
OUT       = Path(f"images/{TODAY}"); OUT.mkdir(parents=True, exist_ok=True)

# ---------- Step 1: generate prompts (dual-model diversity) ----------
async def gen_prompts_claude(campaign, n):
    client = anthropic.Anthropic()
    msg = client.messages.create(
        model="claude-opus-4-6",
        max_tokens=4000,
        system=f"You are a direct-response ad creative director. Generate {n} image-ad prompts for kie.ai Nano Banana 2, vertical 9:16, photorealistic. Source material = fragments below. Obey compliance (no skin closeups, no pills, no disease names in image).",
        messages=[{"role":"user","content":f"FRAGMENTS: {FRAGMENTS}\n\nCAMPAIGN: {campaign}\n\nReturn JSON array of {n} prompts."}]
    )
    return json.loads(msg.content[0].text)

async def gen_prompts_gemini(campaign, n):
    genai.configure(api_key="...")
    model = genai.GenerativeModel("gemini-2.0-flash-exp")
    r = model.generate_content(f"Generate {n} nerve-pain ad image prompts. FRAGMENTS: {FRAGMENTS}. CAMPAIGN: {campaign}. Return JSON array.")
    return json.loads(r.text)

# ---------- Step 2: render images ----------
async def kie_render(prompt, client):
    r = await client.post("https://api.kie.ai/api/v1/jobs/createTask",
        headers={"Authorization": f"Bearer {KIE_KEY}"},
        json={"model":"nano-banana-2","prompt":prompt,"aspect_ratio":"9:16"})
    return r.json()["data"]["taskId"]

async def kie_poll(task_id, client):
    while True:
        r = await client.get(f"https://api.kie.ai/api/v1/jobs/recordInfo?taskId={task_id}",
            headers={"Authorization": f"Bearer {KIE_KEY}"})
        d = r.json()["data"]
        if d["state"] == "success": return d["resultJson"]["imageUrl"]
        if d["state"] == "fail":    return None
        await asyncio.sleep(15)

# ---------- Step 3: red-square overlay (no API cost) ----------
def red_square(hook_text, filename):
    img = Image.new("RGB", (1080, 1920), "#CC1F1F")
    draw = ImageDraw.Draw(img)
    font = ImageFont.truetype("Arial Bold.ttf", 82)
    # simple center-wrap text in yellow
    draw.text((80, 600), hook_text, fill="#FFD84D", font=font)
    img.save(OUT / filename)

# ---------- Step 4: orchestrate ----------
async def main():
    # Campaign A: 20 remix prompts (from spy tool manual upload)
    remix_prompts = json.loads(Path("spy_winners_today.json").read_text())
    # Campaign B: 40 hook lines (no image gen)
    hook_prompts  = await gen_prompts_claude("red_square_hooks", 40)
    # Campaign C: 40 fragment-pipeline prompts
    claude_C      = await gen_prompts_claude("product_hero_split_lifestyle", 20)
    gemini_C      = await gen_prompts_gemini("product_hero_split_lifestyle", 20)

    async with httpx.AsyncClient(timeout=120) as c:
        # Fire all Nano Banana tasks in parallel
        render_jobs = remix_prompts + claude_C + gemini_C  # 60 images
        task_ids = await asyncio.gather(*(kie_render(p, c) for p in render_jobs))
        urls     = await asyncio.gather(*(kie_poll(t, c) for t in task_ids))

        # Download all
        for i, url in enumerate(urls):
            if url:
                img = (await c.get(url)).content
                (OUT / f"ai_{i:03d}.png").write_bytes(img)

    # Generate red-square overlays
    for i, hook in enumerate(hook_prompts):
        red_square(hook, f"hook_{i:03d}.png")

    print(f"Done: {len(list(OUT.glob('*.png')))} images ready at {OUT}")
    # Next step: meta_launcher.py uploads all 100 to 3 bid-cap campaigns.

if __name__ == "__main__":
    asyncio.run(main())