One API for Marketing & AdTech Platforms

from openai import OpenAI
import concurrent.futures
import json

client = OpenAI(
    api_key="YOUR_CREATIVEAI_KEY",
    base_url="https://api.creativeai.run/v1",
)

# Full A/B testing pipeline for ad creatives
# Step 1: Generate variants from brand brief
# Step 2: Submit to ad platform for split testing
# Step 3: (Optional) Generate video versions of winners

brand_brief = {
    "brand": "TechFlow",
    "colors": "deep blue, electric cyan, white",
    "style": "clean, modern, premium tech",
    "product": "SaaS analytics dashboard",
}

# Define A/B test matrix — vary one element at a time
test_matrix = [
    {
        "test_id": "headline-tone",
        "variants": [
            {
                "name": "professional",
                "prompt": f"Ad banner for {brand_brief['product']}, {brand_brief['colors']} palette, "
                          f"professional corporate tone, clean data visualization, {brand_brief['style']}",
            },
            {
                "name": "playful",
                "prompt": f"Ad banner for {brand_brief['product']}, {brand_brief['colors']} palette, "
                          f"playful energetic tone, animated chart elements, {brand_brief['style']}",
            },
        ],
    },
    {
        "test_id": "layout-style",
        "variants": [
            {
                "name": "left-aligned",
                "prompt": f"Ad banner for {brand_brief['product']}, {brand_brief['colors']} palette, "
                          f"product screenshot on right, text space on left, {brand_brief['style']}",
            },
            {
                "name": "centered",
                "prompt": f"Ad banner for {brand_brief['product']}, {brand_brief['colors']} palette, "
                          f"centered hero layout, product screenshot below headline, {brand_brief['style']}",
            },
        ],
    },
]

def generate_test_variants(test):
    """Generate all variants for one A/B test."""
    results = []
    for variant in test["variants"]:
        resp = client.images.generate(
            model="seedream-3",
            prompt=variant["prompt"],
            n=2,
            size="1792x1024",
            quality="high",
        )
        results.append({
            "name": variant["name"],
            "images": [img.url for img in resp.data],
        })
    return {"test_id": test["test_id"], "variants": results}

# Run all tests in parallel
with concurrent.futures.ThreadPoolExecutor(max_workers=4) as executor:
    test_results = list(executor.map(generate_test_variants, test_matrix))

for test in test_results:
    print(f"\nTest: {test['test_id']}")
    for v in test["variants"]:
        print(f"  {v['name']}: {len(v['images'])} images ready for split test")

# Output: structured data ready for your ad platform's A/B test API
print(json.dumps(test_results, indent=2, default=str))