Struct ImageHandler 

pub struct ImageHandler {
    pub config: Config,
    pub gcs: GcsClient,
    pub http: Client,
    pub auth: AuthProvider,
}

Image generation handler.

Handles image generation requests using the Vertex AI Imagen API.

Fields

config: Config

Application configuration.

gcs: GcsClient

GCS client for storage operations.

http: Client

HTTP client for API requests.

auth: AuthProvider

Authentication provider.

Implementations

impl ImageHandler

pub async fn new(config: Config) -> Result<Self, Error>

Create a new ImageHandler with the given configuration.

Errors

Returns an error if GCS client or auth provider initialization fails.

pub fn get_endpoint(&self, model: &str) -> String

Get the Vertex AI Imagen API endpoint for the given model.

pub async fn generate_image( &self, params: ImageGenerateParams, ) -> Result<ImageGenerateResult, Error>

Generate images from a text prompt.

Arguments

• params - Image generation parameters

Returns

• Ok(ImageGenerateResult) - Generated images with their data or paths
• Err(Error) - If validation fails, API call fails, or output handling fails

pub async fn upscale_image( &self, params: ImageUpscaleParams, ) -> Result<ImageUpscaleResult, Error>

Upscale an image using the Imagen Upscale API.

Arguments

• params - Image upscale parameters

Returns

• Ok(ImageUpscaleResult) - Upscaled image with data or path
• Err(Error) - If validation fails, API call fails, or output handling fails

pub fn get_upscale_endpoint(&self) -> String

Get the Vertex AI Imagen Upscale API endpoint.