from google import genai
from google.genai import types
from PIL import Image
from google.genai.client import HttpOptions
client = genai.Client(http_options=HttpOptions(base_url="https://work.poloapi.com"),api_key="sk-V4tPnDgzFPa7nxWrvKnNJsW8ZcBXXPuGmjfgvPVRnwpHoeob")

prompt = ("Create a picture of a nano banana dish in a fancy restaurant with a Gemini theme")
response = client.models.generate_content(
    model="gemini-3-pro-image-preview",
    contents=[prompt],
)

# 检查是否有候选答案
if not response.candidates:
    print("API未返回任何候选答案")
else:
    candidate = response.candidates[0]
    if not candidate.content:
        print("API返回的候选答案中没有内容")
    elif not hasattr(candidate.content, 'parts') or not candidate.content.parts:
        print("API返回的候选答案内容中没有parts")
    else:
        for part in candidate.content.parts:
            if hasattr(part, 'text') and part.text is not None:
                print(part.text)
            elif hasattr(part, 'inline_data') and part.inline_data is not None:
                image_data = part.inline_data
                if image_data.data is not None:
                    # 保存图片数据到文件
                    with open('generated_image.png', 'wb') as f:
                        f.write(image_data.data)
                    print("图片生成成功: generated_image.png")