mirror of
https://github.com/marcogll/anchor23.git
synced 2026-03-15 11:24:19 +00:00
939e62ee77
- Move scripts to tools/ directory - Rename pyton.py to generate_swatches.py - Rename down.py to downscale_colors.py - Remove hardcoded API key from generate_swatches.py - Update README.md to reference local color assets instead of placeholders
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from openai import OpenAI
|
|
import base64
|
|
import os
|
|
|
|
# =========================
|
|
# CONFIGURACIÓN API
|
|
# =========================
|
|
# Set OPENAI_API_KEY as environment variable or replace below
|
|
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
if not OPENAI_API_KEY:
|
|
raise ValueError("OPENAI_API_KEY environment variable not set")
|
|
client = OpenAI(api_key=OPENAI_API_KEY)
|
|
|
|
# =========================
|
|
# CONFIGURACIÓN
|
|
# =========================
|
|
SIZE = "1024x1024" # Tamaño válido
|
|
OUTPUT_DIR = "assets/colors"
|
|
|
|
COLORS = {
|
|
"bone-white": "#F6F1EC",
|
|
"soft-cream": "#EFE7DE",
|
|
"mocha-taupe": "#B8A89A",
|
|
"deep-earth": "#6F5E4F",
|
|
"charcoal-brown": "#3F362E",
|
|
}
|
|
|
|
PROMPT_TEMPLATE = (
|
|
"A flat square color swatch, solid fill, hex color {hex}. "
|
|
"No gradients, no texture, no shadow, no border. "
|
|
"Minimal, neutral, technical style."
|
|
)
|
|
|
|
os.makedirs(OUTPUT_DIR, exist_ok=True)
|
|
|
|
for name, hex_color in COLORS.items():
|
|
prompt = PROMPT_TEMPLATE.format(hex=hex_color)
|
|
|
|
result = client.images.generate(model="gpt-image-1", prompt=prompt, size=SIZE)
|
|
|
|
image_base64 = result.data[0].b64_json
|
|
image_bytes = base64.b64decode(image_base64)
|
|
|
|
output_path = os.path.join(OUTPUT_DIR, f"{name}.png")
|
|
with open(output_path, "wb") as f:
|
|
f.write(image_bytes)
|
|
|
|
print(f"Generated {output_path}")
|