电商商品图与营销素材生成

import base64
import os

import requests
from openai import OpenAI

client = OpenAI(base_url="https://www.tutujin.com/v1", api_key=os.environ["TUTUJIN_API_KEY"])


def save(image, path):
    """Results come back as a temporary URL or as base64. Store the file right away."""
    if image.b64_json:
        data = base64.b64decode(image.b64_json)
    else:
        # some image hosts reject requests without a browser-like User-Agent
        r = requests.get(image.url, headers={"User-Agent": "Mozilla/5.0"}, timeout=120)
        r.raise_for_status()
        data = r.content
    with open(path, "wb") as f:
        f.write(data)


# 1. a lifestyle scene from a text prompt (prompts can be written by qwen3.5-plus)
scene = client.images.generate(
    model="doubao-seedream-5-0-260128",
    prompt="A white ceramic mug on a wooden breakfast table, soft morning light, product photography",
)
save(scene.data[0], "mug-scene.png")

# 2. keep the real product, change only the background
with open("mug-white-background.png", "rb") as f:
    edited = client.images.edit(
        model="nano-banana-pro",
        image=f,
        prompt="Keep the mug exactly as it is and place it on a marble kitchen counter",
    )
save(edited.data[0], "mug-marble.png")