SeeDream 4.5 图片编辑/生成

POST /v1/images/generations

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
SeeDream 4.0 图片生成 - Python版本
支持标准 OpenAI Image API 格式
"""

import requests
import json
import base64
import os
import datetime
from typing import Optional, Dict, Any

class SeeDreamImageGenerator:
    def __init__(self, api_key: str, api_url: str = "https://api.apiyi.com/v1/images/generations"):
        """
        初始化 SeeDream 图片生成器

        Args:
            api_key: API密钥
            api_url: API地址
        """
        self.api_key = api_key
        self.api_url = api_url
        self.headers = {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {api_key}"
        }

    def generate_image(self,
                      prompt: str,
                      model: str = "seedream-4-5-251128",
                      size: str = "2048x2048",
                      quality: str = "standard",
                      n: int = 1,
                      response_format: str = "url",
                      output_dir: str = ".") -> Dict[str, Any]:
        """
        生成图片

        Args:
            prompt: 图片描述提示词
            model: 使用的模型
            size: 图片尺寸
                  方法1: 分辨率规格 ("1K", "2K", "4K") - 让模型根据提示词决定最终尺寸
                  方法2: 精确像素 ("2048x2048", "1920x1080" 等) - 范围[1280x720, 4096x4096]
            quality: 图片质量 (standard 或 hd)
            n: 生成数量
            response_format: 响应格式 (url 或 b64_json)
            output_dir: 输出目录（仅当response_format为b64_json时使用）

        Returns:
            API响应结果
        """
        print("🚀 开始生成图片...")
        print(f"提示词: {prompt}")
        print(f"模型: {model}")
        print(f"尺寸: {size}")
        print(f"质量: {quality}")

        try:
            # 准备请求数据
            payload = {
                "model": model,
                "prompt": prompt,
                "size": size,
                "quality": quality,
                "n": n,
                "response_format": response_format
            }

            print("📡 发送API请求...")

            # 发送请求
            response = requests.post(
                self.api_url,
                headers=self.headers,
                json=payload,
                timeout=60
            )

            if response.status_code != 200:
                error_msg = f"API请求失败，状态码: {response.status_code}"
                try:
                    error_detail = response.json()
                    error_msg += f", 错误详情: {error_detail}"
                except:
                    error_msg += f", 响应内容: {response.text[:500]}"
                print(f"❌ {error_msg}")
                return {"error": error_msg}

            print("✅ API请求成功")

            # 解析响应
            result = response.json()

            # 处理响应数据
            if response_format == "b64_json" and "data" in result:
                print("🔍 正在保存base64图片...")
                for i, item in enumerate(result["data"]):
                    if "b64_json" in item:
                        self._save_base64_image(
                            item["b64_json"],
                            output_dir,
                            f"seedream_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{i}.png"
                        )
            elif response_format == "url" and "data" in result:
                print("🎨 生成的图片URL:")
                for item in result["data"]:
                    if "url" in item:
                        print(f"  🔗 {item['url']}")

            return result

        except requests.exceptions.Timeout:
            error_msg = "请求超时（60秒）"
            print(f"❌ {error_msg}")
            return {"error": error_msg}
        except requests.exceptions.ConnectionError as e:
            error_msg = f"连接错误: {str(e)}"
            print(f"❌ {error_msg}")
            return {"error": error_msg}
        except Exception as e:
            error_msg = f"未知错误: {str(e)}"
            print(f"❌ {error_msg}")
            return {"error": error_msg}

    def _save_base64_image(self, b64_data: str, output_dir: str, filename: str):
        """
        保存base64图片到本地

        Args:
            b64_data: base64编码的图片数据
            output_dir: 输出目录
            filename: 文件名
        """
        try:
            # 解码base64数据
            image_data = base64.b64decode(b64_data)

            # 创建输出目录
            os.makedirs(output_dir, exist_ok=True)

            # 保存文件
            output_path = os.path.join(output_dir, filename)
            with open(output_path, 'wb') as f:
                f.write(image_data)

            print(f"🖼️  图片保存成功: {output_path}")
            print(f"📊 文件大小: {len(image_data)} 字节")

        except Exception as e:
            print(f"❌ 保存图片失败: {str(e)}")

def main():
    """
    主函数 - 演示 SeeDream 4.0 图片生成
    """
    # 配置参数
    API_KEY = "sk-"  # 请替换为你的实际API密钥

    # 示例提示词
    prompts = [
        "A serene Japanese garden with cherry blossoms in full bloom, koi pond, traditional bridge, 4K, ultra detailed",
        "Futuristic cityscape at night with neon lights and flying vehicles, cyberpunk style, high quality",
        "Cute cartoon cat astronaut floating in space with planets in background, pixar style"
    ]

    print("="*60)
    print("SeeDream 4.0 图片生成器 - Python版本")
    print("="*60)
    print(f"开始时间: {datetime.datetime.now()}")

    # 创建生成器实例
    generator = SeeDreamImageGenerator(API_KEY)

    # 生成示例图片
    for i, prompt in enumerate(prompts, 1):
        print(f"\n🎯 示例 {i}/{len(prompts)}")
        print("-"*40)

        result = generator.generate_image(
            prompt=prompt,
            size="2K",          # 可选: "1K", "2K", "4K" 或精确尺寸如 "2048x2048"
            quality="hd",       # 可选: standard, hd
            n=1,                # 生成数量
            response_format="url"  # 可选: url, b64_json
        )

        if "error" not in result:
            print(f"✅ 图片 {i} 生成成功!")
        else:
            print(f"❌ 图片 {i} 生成失败")

        print("-"*40)

    print(f"\n结束时间: {datetime.datetime.now()}")
    print("="*60)

if __name__ == "__main__":
    main()

curl -X POST "https://api.apiyi.com/v1/images/generations" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer YOUR_API_KEY" \
  -d '{
    "model": "seedream-4-5-251128",
    "prompt": "A beautiful sunset over mountains with dramatic clouds",
    "size": "4K",
    "quality": "hd",
    "n": 1,
    "response_format": "url"
  }'

好的提示词：
"A majestic mountain landscape at golden hour, snow-capped peaks reflecting
warm sunlight, crystal clear lake in foreground, pine forests, ultra detailed,
4K resolution, photorealistic"

简单提示词：
"mountain landscape"  # 结果可能不够理想