Nano Banana Pro Image Edit

POST /v1/chat/completions

{
  "model": "gemini-2.5-flash-image",
  "stream": false,
  "messages": [
    {
      "role": "user",
      "content": [
        {
          "type": "text",
          "text": "Edit instruction text"
        },
        {
          "type": "image_url",
          "image_url": {
            "url": "Image URL 1"
          }
        },
        {
          "type": "image_url",
          "image_url": {
            "url": "Image URL 2"
          }
        }
      ]
    }
  ]
}

#!/usr/bin/env python3
import requests
import json
import base64
import re
from datetime import datetime
import sys

# Configuration
API_KEY = "sk-"  # Replace with your actual key
API_URL = "https://api.apiyi.com/v1/chat/completions"

def edit_images():
    """
    Main image editing function
    Supports multi-image composition and editing
    """
    # Set request headers
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }

    # Request data - multi-image composition example
    data = {
        "model": "gemini-2.5-flash-image",
        "stream": False,
        "messages": [
            {
                "role": "user",
                "content": [
                    {
                        "type": "text",
                        "text": "Combine 2 images and add a Corgi dog image"
                    },
                    {
                        "type": "image_url",
                        "image_url": {
                            "url": "https://github.com/dianping/cat/raw/master/cat-home/src/main/webapp/images/logo/cat_logo03.png"
                        }
                    },
                    {
                        "type": "image_url",
                        "image_url": {
                            "url": "https://raw.githubusercontent.com/leonindy/camel/master/camel-admin/src/main/webapp/assets/images/camel_logo_blue.png"
                        }
                    }
                ]
            }
        ]
    }

    print("🚀 Requesting image edit API...")
    print(f"📝 Edit instruction: {data['messages'][0]['content'][0]['text']}")
    print(f"🖼️  Input image count: {len([c for c in data['messages'][0]['content'] if c['type'] == 'image_url'])}")

    try:
        # Send request
        response = requests.post(API_URL, headers=headers, json=data)
        response.raise_for_status()

        print("✅ API request successful, processing response...")

        # Parse response
        result = response.json()

        # Extract content
        content = result['choices'][0]['message']['content']
        print(f"📄 Received content length: {len(content)} characters")
        print(f"📄 Content preview: {content[:200]}...")

        # Find Base64 image data
        base64_data = None

        # Method 1: Find standard format data:image/type;base64,data
        base64_match = re.search(r'data:image/[^;]+;base64,([A-Za-z0-9+/=]+)', content)

        if base64_match:
            base64_data = base64_match.group(1)
            print("✅ Found standard format Base64 data")
        else:
            # Method 2: Find pure Base64 data (long string)
            base64_match = re.search(r'([A-Za-z0-9+/=]{100,})', content)
            if base64_match:
                base64_data = base64_match.group(1)
                print("✅ Found pure Base64 data")
            else:
                print("❌ Error: Cannot find Base64 image data")
                print("📄 Full response content:")
                print(json.dumps(result, indent=2, ensure_ascii=False))
                return False

        # Generate filename
        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
        filename = f"edited_image_{timestamp}.png"

        print("💾 Saving edited image...")

        # Decode and save image
        try:
            image_data = base64.b64decode(base64_data)
            with open(filename, 'wb') as f:
                f.write(image_data)

            print(f"🎉 Image editing complete!")
            print(f"📁 File saved as: {filename}")
            print(f"📊 File size: {len(image_data):,} bytes")
            return True

        except Exception as e:
            print(f"❌ Error: Problem saving image: {e}")
            return False

    except requests.exceptions.RequestException as e:
        print(f"❌ Error: API request failed: {e}")
        return False
    except KeyError as e:
        print(f"❌ Error: Response format incorrect, missing field: {e}")
        if 'response' in locals():
            print("📄 Full response content:")
            print(json.dumps(response.json(), indent=2, ensure_ascii=False))
        return False
    except Exception as e:
        print(f"❌ Error: Unknown error: {e}")
        return False

def custom_edit_example():
    """
    Custom edit example
    """
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }

    # Custom edit task
    data = {
        "model": "gemini-2.5-flash-image",
        "stream": False,
        "messages": [
            {
                "role": "user",
                "content": [
                    {
                        "type": "text",
                        "text": "Please remove the background and make it transparent, then add a sunset background"
                    },
                    {
                        "type": "image_url",
                        "image_url": {
                            "url": "your_image_url_here"  # Replace with your image URL
                        }
                    }
                ]
            }
        ]
    }

    print("🎨 Executing custom edit task...")
    # Call edit logic...

def main():
    """
    Main function - Demonstrates different editing features
    """
    print("=" * 60)
    print("Nano Banana Image Editor - Python Version")
    print("=" * 60)
    print(f"🕐 Start time: {datetime.now()}")

    # Check API Key
    if API_KEY == "sk-":
        print("⚠️  Please set the correct API key first")
        return

    print("\n🔧 Available editing features:")
    print("1. Multi-image composition")
    print("2. Background replacement")
    print("3. Element addition")
    print("4. Style conversion")

    print("\n🚀 Executing multi-image composition example...")
    success = edit_images()

    print("\n" + "=" * 60)
    if success:
        print("🎉 Edit task executed successfully!")
        print("✅ Image saved locally")
    else:
        print("❌ Edit task execution failed!")
        print("💡 Please check API key and network connection")

    print(f"🕐 End time: {datetime.now()}")
    print("=" * 60)

if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)

# Combine two logos and add new elements
{
  "type": "text",
  "text": "Combine these two logos and add a cute animal"
}

# Replace image background
{
  "type": "text",
  "text": "Remove the background and add a sunset landscape"
}

# Change image style
{
  "type": "text",
  "text": "Convert this photo to anime/cartoon style"
}

# Add or remove elements
{
  "type": "text",
  "text": "Add sunglasses to the person and change hair color to blue"
}

def batch_edit_images(image_urls, edit_instruction):
    """Batch edit multiple images"""
    results = []
    for i, url in enumerate(image_urls):
        data = {
            "model": "gemini-2.5-flash-image",
            "messages": [{
                "role": "user",
                "content": [
                    {"type": "text", "text": edit_instruction},
                    {"type": "image_url", "image_url": {"url": url}}
                ]
            }]
        }
        # Execute edit...
        results.append(result)
    return results

def save_edit_history(original_url, instruction, result_path):
    """Save edit history record"""
    history = {
        "timestamp": datetime.now().isoformat(),
        "original_image": original_url,
        "edit_instruction": instruction,
        "result_path": result_path
    }
    # Save to file or database...