234 lines
9.3 KiB
PHP
234 lines
9.3 KiB
PHP
<?php
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
* phpcs:ignore WordPress.Files.FileName.InvalidClassFileName -- class naming convention differs from WordPress standard
|
|
*
|
|
* Static registry of available image generation models.
|
|
*
|
|
* What: Returns a hardcoded list of image models for the admin model picker.
|
|
* No API discovery — updated manually when providers add/remove models.
|
|
* As of 2026-04-26, registry includes 21 models across Runware (15) and
|
|
* OpenRouter (6), ordered cheapest-to-most-expensive within each provider.
|
|
* Who calls it: PRAutoBlogger_Admin_Page (model picker UI + save-time provider
|
|
* derivation) and PRAutoBlogger_Image_Pipeline (provider lookup).
|
|
* Dependencies: None.
|
|
*
|
|
* @see admin/class-settings-fields-extended.php — Image settings reference this registry.
|
|
* @see core/class-image-pipeline.php — Derives provider from picked model.
|
|
* @see providers/class-runware-image-provider.php — Runware FLUX backend.
|
|
*/
|
|
class PRAutoBlogger_Image_Model_Registry {
|
|
|
|
/**
|
|
* Get all available image generation models.
|
|
*
|
|
* Each entry contains:
|
|
* - id: Model identifier used in API calls.
|
|
* - name: Human-readable display name.
|
|
* - provider: 'runware' | 'openrouter'.
|
|
* - cost_per_image: Estimated USD cost per generation.
|
|
* - capabilities: Array of capability tags.
|
|
* - description: Short description for the admin UI.
|
|
*
|
|
* Ordered cheapest-to-most-expensive within each provider, with the
|
|
* recommended default (Runware schnell) at the top.
|
|
*
|
|
* Last verified: 2026-04-26. To update: check https://runware.ai/models (text-to-image section)
|
|
* and https://openrouter.ai/api/v1/models filtering output_modalities containing 'image'.
|
|
* Exclude: inpainting (needs mask), img2img (needs seedImage), video, upscalers, background removal.
|
|
*
|
|
* @return array<int, array<string, mixed>>
|
|
*/
|
|
public static function get_models(): array {
|
|
return array(
|
|
// RUNWARE MODELS
|
|
array(
|
|
'id' => 'runware:100@1',
|
|
'name' => 'FLUX.1 schnell (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0006,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Default. Fast, very cheap, comic-friendly fidelity.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:twinflow-z-image-turbo@0',
|
|
'name' => 'TwinFlow Z-Image-Turbo (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0006,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Fast text-to-image, same cost as schnell.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:400@2',
|
|
'name' => 'FLUX.2 klein 9B (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.00078,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'FLUX.2 compact 9B. Stronger than schnell at similar cost.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:400@4',
|
|
'name' => 'FLUX.2 klein 4B (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0006,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'FLUX.2 compact 4B. Newer architecture at schnell price.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:5@1',
|
|
'name' => 'Stable Diffusion 3 (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0013,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Stable Diffusion 3. Sharp text, complex scenes.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:z-image@turbo',
|
|
'name' => 'Z-Image Turbo (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0013,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Fast photorealistic generation.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:101@1',
|
|
'name' => 'FLUX.1 dev (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.02,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Higher fidelity FLUX.1. ~30x schnell cost.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:97@3',
|
|
'name' => 'HiDream-I1 Fast (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0038,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'HiDream-I1 Fast. Low latency, 17B quality.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:97@2',
|
|
'name' => 'HiDream-I1 Dev (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0045,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'HiDream-I1 Dev. Strong prompt alignment.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:z-image@0',
|
|
'name' => 'Z-Image (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0045,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'High-quality Z-Image foundation model.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:400@1',
|
|
'name' => 'FLUX.2 dev (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0051,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'FLUX.2 dev. Controllable open text-to-image. From $0.0051.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:108@1',
|
|
'name' => 'Qwen-Image (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0058,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Alibaba Qwen-Image. Strong text rendering in generated images.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:97@1',
|
|
'name' => 'HiDream-I1 Full (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.009,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'HiDream-I1 Full. 17B high-fidelity, LoRA support.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:107@1',
|
|
'name' => 'FLUX.1 Krea dev (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0098,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'FLUX.1 Krea Dev. Photorealistic open-weight generation.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'runware:glm-image@0',
|
|
'name' => 'GLM-Image (Runware)',
|
|
'provider' => 'runware',
|
|
'cost_per_image' => 0.0225,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'GLM-Image. Hybrid autoregressive+diffusion, excellent text rendering.', 'prautoblogger' ),
|
|
),
|
|
// OPENROUTER MODELS
|
|
array(
|
|
'id' => 'google/gemini-2.5-flash-image',
|
|
'name' => 'Gemini 2.5 Flash Image (OpenRouter)',
|
|
'provider' => 'openrouter',
|
|
'cost_per_image' => 0.005,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Mid-tier quality, proven.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'google/gemini-3.1-flash-image-preview',
|
|
'name' => 'Gemini 3.1 Flash Image (OpenRouter)',
|
|
'provider' => 'openrouter',
|
|
'cost_per_image' => 0.008,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Latest Google Flash image. Better quality.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'openai/gpt-5-image-mini',
|
|
'name' => 'GPT-5 Image Mini (OpenRouter)',
|
|
'provider' => 'openrouter',
|
|
'cost_per_image' => 0.02,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'OpenAI budget image model.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'openai/gpt-5.4-image-2',
|
|
'name' => 'GPT-5.4 Image 2 (OpenRouter)',
|
|
'provider' => 'openrouter',
|
|
'cost_per_image' => 0.02,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'OpenAI GPT-5.4 image generation.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'google/gemini-3-pro-image-preview',
|
|
'name' => 'Gemini 3 Pro Image (OpenRouter)',
|
|
'provider' => 'openrouter',
|
|
'cost_per_image' => 0.03,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'Highest quality Google image. Premium.', 'prautoblogger' ),
|
|
),
|
|
array(
|
|
'id' => 'openai/gpt-5-image',
|
|
'name' => 'GPT-5 Image (OpenRouter)',
|
|
'provider' => 'openrouter',
|
|
'cost_per_image' => 0.08,
|
|
'capabilities' => array( 'image_generation' ),
|
|
'description' => __( 'OpenAI premium. Photorealistic.', 'prautoblogger' ),
|
|
),
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Return the provider id for a known model, or empty string if the
|
|
* model id is not in the registry.
|
|
*
|
|
* @param string $model_id Model slug from the admin UI.
|
|
* @return string Provider id ('runware' | 'openrouter' | '').
|
|
*/
|
|
public static function provider_for( string $model_id ): string {
|
|
foreach ( self::get_models() as $model ) {
|
|
if ( ( $model['id'] ?? '' ) === $model_id ) {
|
|
return (string) ( $model['provider'] ?? '' );
|
|
}
|
|
}
|
|
return '';
|
|
}
|
|
}
|