Generate color palettes by generating AI images and extracting the colors

make_palette.py

import sys
import requests
import replicate
from Pylette import extract_colors

# YOU NEED A REPLICATE API TOKEN IN 'REPLICATE_API_TOKEN'

prompt = sys.argv[1] if len(sys.argv) > 1 else "sandy beach"

output = replicate.run(
    "black-forest-labs/flux-krea-dev",
    input={
        "prompt": prompt,
        "guidance": 3,
        "output_quality": 95,
        "megapixels": "1",
        "output_format": "jpg",
        "num_inference_steps": 30,
        "disable_safety_checker": True
    }
)

url = output[0]

response = requests.get(url)
if response.status_code == 200:
    with open("output_image.jpg", "wb") as f:
        f.write(response.content)
else:
    print(f"Failed to retrieve image: {response.status_code}")
    exit(1)

palette = extract_colors(image='output_image.jpg', palette_size=10)
palette.display(save_to_file=True)
#palette.save(filename='output_image_palette', extension='png')

OK, so you get the idea!