SeeDream 4.0-4.5 Image API

POST /v1/images/generations

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

"""
SeeDream 4.0 Image Generation - Python Version
Supports standard OpenAI Image API format
"""

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"):
        """
        Initialize SeeDream Image Generator

        Args:
            api_key: API key
            api_url: API address
        """
        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]:
        """
        Generate image

        Args:
            prompt: Image description prompt
            model: Model to use
            size: Image size
                  Method 1: Resolution spec ("1K", "2K", "4K") - let model decide final dimensions
                  Method 2: Exact pixels ("2048x2048", "1920x1080", etc.) - range [1280x720, 4096x4096]
            quality: Image quality (standard or hd)
            n: Generation count
            response_format: Response format (url or b64_json)
            output_dir: Output directory (used only when response_format is b64_json)

        Returns:
            API response result
        """
        print("🚀 Starting image generation...")
        print(f"Prompt: {prompt}")
        print(f"Model: {model}")
        print(f"Size: {size}")
        print(f"Quality: {quality}")

        try:
            # Prepare request data
            payload = {
                "model": model,
                "prompt": prompt,
                "size": size,
                "quality": quality,
                "n": n,
                "response_format": response_format
            }

            print("📡 Sending API request...")

            # Send request
            response = requests.post(
                self.api_url,
                headers=self.headers,
                json=payload,
                timeout=60
            )

            if response.status_code != 200:
                error_msg = f"API request failed, status code: {response.status_code}"
                try:
                    error_detail = response.json()
                    error_msg += f", error details: {error_detail}"
                except:
                    error_msg += f", response content: {response.text[:500]}"
                print(f"❌ {error_msg}")
                return {"error": error_msg}

            print("✅ API request successful")

            # Parse response
            result = response.json()

            # Process response data
            if response_format == "b64_json" and "data" in result:
                print("🔍 Saving base64 images...")
                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("🎨 Generated image URLs:")
                for item in result["data"]:
                    if "url" in item:
                        print(f"  🔗 {item['url']}")

            return result

        except requests.exceptions.Timeout:
            error_msg = "Request timeout (60 seconds)"
            print(f"❌ {error_msg}")
            return {"error": error_msg}
        except requests.exceptions.ConnectionError as e:
            error_msg = f"Connection error: {str(e)}"
            print(f"❌ {error_msg}")
            return {"error": error_msg}
        except Exception as e:
            error_msg = f"Unknown error: {str(e)}"
            print(f"❌ {error_msg}")
            return {"error": error_msg}

    def _save_base64_image(self, b64_data: str, output_dir: str, filename: str):
        """
        Save base64 image locally

        Args:
            b64_data: Base64 encoded image data
            output_dir: Output directory
            filename: Filename
        """
        try:
            # Decode base64 data
            image_data = base64.b64decode(b64_data)

            # Create output directory
            os.makedirs(output_dir, exist_ok=True)

            # Save file
            output_path = os.path.join(output_dir, filename)
            with open(output_path, 'wb') as f:
                f.write(image_data)

            print(f"🖼️  Image saved successfully: {output_path}")
            print(f"📊 File size: {len(image_data)} bytes")

        except Exception as e:
            print(f"❌ Failed to save image: {str(e)}")

def main():
    """
    Main function - demonstrates SeeDream 4.0 image generation
    """
    # Configuration parameters
    API_KEY = "sk-"  # Replace with your actual API key

    # Example prompts
    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 Image Generator - Python Version")
    print("="*60)
    print(f"Start time: {datetime.datetime.now()}")

    # Create generator instance
    generator = SeeDreamImageGenerator(API_KEY)

    # Generate example images
    for i, prompt in enumerate(prompts, 1):
        print(f"\n🎯 Example {i}/{len(prompts)}")
        print("-"*40)

        result = generator.generate_image(
            prompt=prompt,
            size="2K",          # Options: "1K", "2K", "4K" or exact size like "2048x2048"
            quality="hd",       # Options: standard, hd
            n=1,                # Generation count
            response_format="url"  # Options: url, b64_json
        )

        if "error" not in result:
            print(f"✅ Image {i} generated successfully!")
        else:
            print(f"❌ Image {i} generation failed")

        print("-"*40)

    print(f"\nEnd time: {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"
  }'

Good prompt:
"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"

Simple prompt:
"mountain landscape"  # Results may not be ideal