cmfy 0.3.0

A CLI companion app for Comfy UI
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use super::Run;
use crate::io::{Input, JsonRead};
use clap::Args;
use cmfy::{dto, nodes::KSampler};
use colored::Colorize;

/// Submits a batch of prompts to the server.
///
/// Reads a batch of prompts from a JSON file and submits it
/// to the server.
#[derive(Debug, Args)]
pub struct Submit {
    /// Input file containing the prompts in json format
    #[clap(default_value = "-")]
    input: Input,

    /// Reseeds the prompts before submission
    /// (assumes a KSampler node)
    #[clap(long, short, action, default_value_t = false)]
    reseed: bool,

    /// Allows to specify the number of times each prompt
    /// will be submitted.
    #[clap(long, short = 'n', action, default_value_t = 1)]
    count: usize,
}

impl Run for Submit {
    async fn run(mut self, client: cmfy::Client) -> cmfy::Result<()> {
        let prompts: Vec<dto::PromptNodes> = self.input.read_json()?;
        for mut prompt in prompts {
            for _ in 0..self.count {
                if self.reseed {
                    prompt.reseed()?;
                }
                let response = client.submit(&prompt).await?;
                let index = format!("[{}]", response.number.to_string().bright_blue());
                println!("{:<15}{}", index, response.prompt_id);
            }
        }
        Ok(())
    }
}