nekoai-api 0.1.2

🎨 A lightweight async Python API for NovelAI image generation and director tools.

🐾 NekoAI-API

🎨 A lightweight async Python API for NovelAI image generation and director tools.

🌈 Introduction

🐾 NekoAI-API is a lightweight and easy-to-use Python wrapper for NovelAI's image generation capabilities. This project makes it simple to integrate NovelAI's powerful image generation and manipulation tools into your Python applications with minimal code overhead.

Built with asyncio for efficient performance, it provides full access to NovelAI's latest models (V3, V4, V4.5) and Director tools while maintaining a clean, pythonic interface. This project was heavily inspired by HanaokaYuzu's NovelAI-API, with a focus on providing more features support and enhanced usability.

🌟 Core Capabilities

Feature	Description
🚀 Lightweight	Focuses on image generation and Director tools, providing a simple and easy-to-use interface.
⚙️ Parameterized	Provides a Metadata class to easily set up generation parameters with type validation.
⚡ Asynchronous	Utilizes asyncio to run generating tasks and return outputs efficiently.
🔑 Multiple Authentication Methods	Supports both username/password and direct token authentication.
🌐 Custom Hosts	Allows specifying custom API hosts for flexibility.
✨ Latest Models	Full support for V3, V4, and V4.5 models including multi-character generation.
🛠️ Director Tools	Complete support for all NovelAI Director tools like line art, background removal, and emotion change.

📦 Installation

[!IMPORTANT]

This package requires Python 3.10 or higher.

Install/update with pip:

pip install -U nekoai-api

🚀 Usage

🔑 Initialization

Import required packages and initialize a client with your NovelAI credentials. You can use either username/password or a direct token.

import asyncio
from nekoai import NovelAI

# Option 1: Username and password
async def main_with_credentials():
    client = NovelAI(username="your_username", password="your_password")
    await client.init(timeout=30)

# Option 2: Direct token authentication
async def main_with_token():
    client = NovelAI(token="your_access_token")
    await client.init(timeout=30)

asyncio.run(main_with_token())  # Or main_with_credentials()

🖼️ Image Generation

After initializing successfully, you can generate images with the generate_image method. The method takes parameters directly or a Metadata object, and an optional host argument to specify the backend to use.

By passing verbose=True, the method will print the estimated Anlas cost each time a generating request is going to be made.

from nekoai import NovelAI, Metadata
from nekoai.constant import Model, Resolution, Sampler, Noise

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    # Generate using Metadata object
    metadata = Metadata(
        prompt="1girl, cute, anime style, detailed",
        model=Model.V4_5_CUR,  # Use the latest V4.5 model
        res_preset=Resolution.NORMAL_PORTRAIT,
        n_samples=1,
    )

    # Alternative: pass parameters directly
    images = await client.generate_image(
        prompt="1girl, cute, anime style, detailed",
        model=Model.V4_5_CUR,
        res_preset=Resolution.NORMAL_PORTRAIT,
        verbose=True,  # Show Anlas cost
        seed=1234567890  # Fixed seed for reproducibility
    )

    for image in images:
        image.save(path="output")
        print(f"Image saved: {image.filename}")

asyncio.run(main())

Multi-Character Generation (V4.5)

V4.5 models support generating multiple characters with character-specific prompts and positioning.

from nekoai import NovelAI
from nekoai.constant import Model, Resolution
from nekoai.types import CharacterPrompt, PositionCoords

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    # Create character prompts with positioning
    character_prompts = [
        CharacterPrompt(
            prompt="girl, red hair, red dress",
            uc="bad hands, bad anatomy",
            center=PositionCoords(x=0.3, y=0.3),
        ),
        CharacterPrompt(
            prompt="boy, blue hair, blue uniform",
            uc="bad hands, bad anatomy",
            center=PositionCoords(x=0.7, y=0.7),
        )
    ]
    
    # Generate image with multiple characters
    images = await client.generate_image(
        prompt="two people standing together, park background",
        model=Model.V4,
        res_preset=Resolution.NORMAL_LANDSCAPE,
        characterPrompts=character_prompts,
        verbose=True
    )
    
    for image in images:
        image.save("output")

asyncio.run(main())

Image to Image

To perform img2img action, set action parameter to Action.IMG2IMG, and provide a base64-encoded image.

import base64
from nekoai import NovelAI
from nekoai.constant import Action
from nekoai.utils import parse_image

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    width, height, base64_image = parse_image('image.png')

    images = await client.generate_image(
        prompt="1girl, fantasy outfit",
        action=Action.IMG2IMG,
        width=width,
        height=height,
        image=base64_image ,
        strength=0.5,  # Lower = more similar to original
        noise=0.1,
        verbose=True
    )

    for image in images:
        image.save("output")

asyncio.run(main())

Inpainting

To perform inpainting, set action to Action.INPAINT, and provide both a base image and a mask.

import base64
from nekoai import NovelAI
from nekoai.constant import Model, Action, Resolution

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    with open("input/portrait.jpg", "rb") as f:
        base_image = base64.b64encode(f.read()).decode("utf-8")

    with open("input/mask.jpg", "rb") as f:
        mask = base64.b64encode(f.read()).decode("utf-8")

    images = await client.generate_image(
        prompt="1girl, detailed background",
        model=Model.V3INP,  # Use inpainting model
        action=Action.INPAINT,
        res_preset=Resolution.NORMAL_PORTRAIT,
        image=base_image,
        mask=mask,
        add_original_image=True,  # Overlay original image
        verbose=True
    )

    for image in images:
        image.save("output")

asyncio.run(main())

Vibe Transfer

Vibe transfer allows using a reference image's style or mood in your generated image.

import base64
from nekoai import NovelAI
from nekoai.constant import Model, Resolution

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    with open("input/style_reference.jpg", "rb") as f:
        ref_image = base64.b64encode(f.read()).decode("utf-8")

    images = await client.generate_image(
        prompt="landscape, mountains, sunset",
        model=Model.V4,
        res_preset=Resolution.NORMAL_LANDSCAPE,
        reference_image_multiple=[ref_image],
        reference_information_extracted_multiple=[1],  # Max information extracted
        reference_strength_multiple=[0.7],  # Strong style transfer
        verbose=True
    )

    for image in images:
        image.save("output")

asyncio.run(main())

Director Tools

NovelAI offers several Director tools for image manipulation, all accessible through dedicated methods.

Line Art

Convert an image to line art:

import base64
import asyncio
from nekoai import NovelAI

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    result = await client.line_art('image.png')
    result.save("output")

    print(f"Line art saved as {result.filename}")

asyncio.run(main())

Background Removal

Remove the background from an image:

import base64
import asyncio
from nekoai import NovelAI

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    result = await client.background_removal('image.png')
    result.save("output")

    print(f"Background has been removed and saved as {result.filename}")

asyncio.run(main())

Change Emotion

Change the emotion of a character in an image:

import base64
import asyncio
from nekoai import NovelAI
from nekoai.types import EmotionOptions, EmotionLevel

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    result = await client.change_emotion(
        image="image.png"
        target_emotion=EmotionOptions.HAPPY,
        original_emotion=EmotionOptions.NEUTRAL,
        emotion_level=EmotionLevel.NORMAL
    )
    
    result.save("output")

asyncio.run(main())

Other Director Tools

Additional tools include:

# Declutter an image
result = await client.declutter(image='image.png')

# Colorize a sketch or line art
result = await client.colorize(image='image.png')

Custom Hosts

You can specify custom API hosts for flexibility:

from nekoai import NovelAI
from nekoai.constant import Host
from nekoai.types.host import HostInstance

async def main():
    client = NovelAI(token="your_access_token")
    await client.init()
    
    # Use predefined hosts
    images1 = await client.generate_image(
        prompt="1girl",
        host=Host.API  # Use API host instead of default WEB host
    )
    
    # Use custom host
    custom_host = Host.custom(
        url="https://your-custom-host.com",
        accept="binary/octet-stream",
        name="custom-host"
    )
    
    images2 = await client.generate_image(
        prompt="1girl",
        host=custom_host
    )

    # Or create HostInstance directly
    custom_host2 = HostInstance(
        url="https://another-custom-host.com",
        accept="binary/octet-stream",
        name="another-custom"
    )
    
    images3 = await client.generate_image(
        prompt="1girl",
        host=custom_host2
    )

asyncio.run(main())

CLI Token Generation

You can generate an access token from the command line:

# Replace with your actual account credentials
nekoai login <username> <password>

Example Scripts

The package includes several example scripts in the examples/requests/ directory:

• Generation with different models (V3, V4, V4.5)
• Multi-character generation
• All director tools (line art, background removal, emotion change, etc.)

References

NovelAI Documentation

NovelAI Backend API

NovelAI Unofficial Knowledgebase

Aedial/novelai-api

NovelAI-API