import requests
import json
import base64
1. 连接你的SD WebUI本地API (默认端口 7860)
def generate_chaoshan_avatar(prompt, negative_prompt=""):
url = "http://127.0.0.1:7860/sdapi/v1/txt2img"
2. 核心Prompt工程：锁定潮汕文化与C4D质感
full_prompt = prompt + ", Chaoshan Yingge Dance hero, blue and white porcelain background, C4D render, blind box style, bright lighting, 8k, ultra-detailed"
payload = {
"prompt": full_prompt,
"negative_prompt": negative_prompt or "lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
"steps": 30,
"width": 512,
"height": 512,
"cfg_scale": 7,
"sampler_name": "DPM++ 2M Karras",
"seed": -1, # 随机种子，每次生成不同的头像
"batch_size": 4 # 一次批量生成4张头像
}
try:
response = requests.post(url, json=payload)
response.raise_for_status()
result = response.json()
3. 保存图片到本地
for i, image_data in enumerate(result'images'):
with open(f"chaoshan_avatar_{i}.png", "wb") as f:
f.write(base64.b64decode(image_data))
print("✅ 国潮头像生成成功！")
return True
except Exception as e:
print(f"❌ 生成失败: {e}")
return False
4. 调用函数，生成"搞钱奶茶"头像
if name == "main":
prompt = "A cute Q-version chaoshan hero holding a milk tea and gold ingot, standing on a large blue-green jelly dessert"
generate_chaoshan_avatar(prompt)
