import requests
import json
import time
from PIL import Image
import numpy as np
import io

url = "https://api.acedata.cloud/openai/images/generations"

headers = {
    "accept": "application/json",
    "authorization": "Bearer 3a3b982990c44fc5ba5428320d72330e",
    "content-type": "application/json"
}

prompt = """A single luxurious 3D circular Token coin/currency symbol icon on pure white background. The round golden coin has the word 'TOKEN' elegantly embossed in large bold capital letters across the center as the only main element. Surrounding the 'TOKEN' text is a glowing circular ring of binary code (0s and 1s) and neural network patterns. The coin has a metallic gold and cyan gradient, with subtle holographic shimmer effects, microchip circuit lines etched on the rim, and a soft glowing aura. Style: 3D rendered, ultra-detailed, photorealistic, premium fintech/AI icon design, single round coin object only, no other letters or symbols, isolated on pure white background only."""

payload = {
    "model": "gpt-image-2",
    "prompt": prompt,
    "size": "1024x1024"
}

print("正在调用 gpt-image-2 API 生成只含 TOKEN 字样的圆形货币符号...")
response = requests.post(url, json=payload, headers=headers)
result = response.json()
print(f"响应状态: {response.status_code}")

if response.status_code == 200 and 'data' in result and len(result['data']) > 0:
    img_url = result['data'][0].get('url')
    print(f"图片URL: {img_url}")

    img_response = requests.get(img_url)
    if img_response.status_code == 200:
        img = Image.open(io.BytesIO(img_response.content)).convert('RGBA')
        print(f"原始图片尺寸: {img.size}, 模式: {img.mode}")

        data = np.array(img)
        threshold = 240
        r, g, b, a = data[:,:,0], data[:,:,1], data[:,:,2], data[:,:,3]
        white_mask = (r > threshold) & (g > threshold) & (b > threshold)
        data[:,:,3] = np.where(white_mask, 0, 255)
        img_transparent = Image.fromarray(data, 'RGBA')

        timestamp = int(time.time())
        filename = f"token-coin-only-text-transparent-{timestamp}.png"
        filepath = f"/var/www/html/share/{filename}"
        img_transparent.save(filepath, 'PNG')

        print(f"\n✅ 透明PNG已保存: {filepath}")
        print(f"访问地址: http://43.156.179.74/share/{filename}")

        filename_white = f"token-coin-only-text-whitebg-{timestamp}.png"
        filepath_white = f"/var/www/html/share/{filename_white}"
        with open(filepath_white, 'wb') as f:
            f.write(img_response.content)
        print(f"原图（白底）: http://43.156.179.74/share/{filename_white}")
else:
    print(f"错误响应: {json.dumps(result, indent=2, ensure_ascii=False)[:1000]}")