Struct stable_diffusion_bot::StableDiffusionBotBuilder

pub struct StableDiffusionBotBuilder { /* private fields */ }

Struct that builds a StableDiffusionBot instance.

Implementations

impl StableDiffusionBotBuilder

pub fn new( api_key: String, allowed_users: Vec<i64>, sd_api_url: String, api_type: ApiType, allow_all_users: bool, ) -> Self

Constructor that returns a new StableDiffusionBotBuilder instance.

pub fn db_path(self, path: Option<String>) -> Self

Builder function that sets the path of the storage database for the bot.

Arguments

• path - An optional String representing the path to the storage database.

Examples

let builder = StableDiffusionBotBuilder::new(api_key, allowed_users, sd_api_url, allow_all_users);

let bot = builder.db_path(Some("database.sqlite".to_string())).build().await.unwrap();

pub fn txt2img_defaults(self, request: Txt2ImgRequest) -> Self

Builder function that sets the defaults for text to image requests.

Arguments

• request - A Txt2ImgRequest representing the default settings for text to image conversion.

Examples

let builder = StableDiffusionBotBuilder::new(api_key, allowed_users, sd_api_url, api_type, allow_all_users);

let bot = builder.txt2img_defaults(Txt2ImgRequest::default()).build().await.unwrap();

pub fn clear_txt2img_defaults(self) -> Self

Builder function that clears the defaults for text to image requests.

pub fn img2img_defaults(self, request: Img2ImgRequest) -> Self

Builder function that sets the defaults for image to image requests.

Arguments

• request - An Img2ImgRequest representing the default settings for image to image conversion.

Examples

let builder = StableDiffusionBotBuilder::new(api_key, allowed_users, sd_api_url, api_type, allow_all_users);

let bot = builder.img2img_defaults(Img2ImgRequest::default()).build().await.unwrap();

pub fn clear_img2img_defaults(self) -> Self

Builder function that clears the defaults for image to image requests.

pub fn comfyui_config(self, _: ComfyUIConfig) -> Self

pub async fn build(self) -> Result<StableDiffusionBot>

Consumes the builder and builds a StableDiffusionBot instance.

Examples

let builder = StableDiffusionBotBuilder::new(api_key, allowed_users, sd_api_url, api_type, allow_all_users);

let bot = builder.build().await.unwrap();