#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Nano Banana Pro 图片生成 - 完整演示脚本
包含三个场景：文生图、单图生图、多图混合
"""
import requests
import base64
import os
from datetime import datetime
# ========== 配置区 ==========
API_KEY = "sk-YOUR_API_KEY"
API_URL = "https://api.yelinai.com/v1beta/models/gemini-3-pro-image-preview:generateContent"
# ============================
def generate_text_to_image(prompt, aspect_ratio="16:9", image_size="2K"):
"""场景1: 文生图"""
print(f"\n📸 文生图: {prompt}")
payload = {
"contents": [{"parts": [{"text": prompt}]}],
"generationConfig": {
"responseModalities": ["IMAGE"],
"imageConfig": {
"aspectRatio": aspect_ratio,
"imageSize": image_size
}
}
}
return call_api(payload, f"text_{image_size}")
def generate_image_to_image(input_image, prompt, aspect_ratio="1:1", image_size="2K"):
"""场景2: 单图生图"""
print(f"\n🎨 单图生图: {input_image}")
with open(input_image, "rb") as f:
img_b64 = base64.b64encode(f.read()).decode("utf-8")
payload = {
"contents": [{
"parts": [
{"text": prompt},
{"inline_data": {"mime_type": "image/jpeg", "data": img_b64}}
]
}],
"generationConfig": {
"responseModalities": ["IMAGE"],
"imageConfig": {
"aspectRatio": aspect_ratio,
"imageSize": image_size
}
}
}
return call_api(payload, f"styled_{image_size}")
def generate_multi_image_mix(images, prompt, aspect_ratio="16:9", image_size="2K"):
"""场景3: 多图混合"""
print(f"\n🖼️ 多图混合: {len(images)} 张图片")
parts = [{"text": prompt}]
for img in images:
with open(img, "rb") as f:
parts.append({"inline_data": {"mime_type": "image/jpeg", "data": base64.b64encode(f.read()).decode()}})
payload = {
"contents": [{"parts": parts}],
"generationConfig": {
"responseModalities": ["IMAGE"],
"imageConfig": {
"aspectRatio": aspect_ratio,
"imageSize": image_size
}
}
}
return call_api(payload, f"mixed_{image_size}")
def call_api(payload, prefix):
"""调用 API"""
headers = {"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"}
try:
response = requests.post(API_URL, headers=headers, json=payload, timeout=180)
result = response.json()
image_data = result["candidates"][0]["content"]["parts"][0]["inlineData"]["data"]
filename = f"{prefix}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png"
with open(filename, "wb") as f:
f.write(base64.b64decode(image_data))
print(f"✅ 已保存: {filename}")
return filename
except Exception as e:
print(f"❌ 错误: {e}")
return None
# ========== 主程序 ==========
def main():
print("🎨 Nano Banana Pro 图片生成 - 完整演示")
# 场景 1: 文生图
img1 = generate_text_to_image(
"A futuristic cyberpunk city at night, 4k",
aspect_ratio="16:9",
image_size="4K"
)
# 场景 2: 单图生图 (需要 test_cat.jpg)
if os.path.exists("test_cat.jpg"):
img2 = generate_image_to_image(
"test_cat.jpg",
"Van Gogh Starry Night style",
aspect_ratio="1:1",
image_size="2K"
)
# 场景 3: 多图混合 (需要 test_cat.jpg, test_apple.jpg)
if os.path.exists("test_cat.jpg") and os.path.exists("test_apple.jpg"):
generate_multi_image_mix(
["test_cat.jpg", "test_apple.jpg"],
"A cat eating an apple",
aspect_ratio="16:9",
image_size="2K"
)
print("\n✅ 演示完成！")
if __name__ == "__main__":
main()
如果未设置计费模式，API调用会失败。必须先完成此配置！
2
选择计费方式
重要：必须选择”按次计费”类型，不要选择”按量计费”
3
保存令牌
复制生成的令牌，格式为