cmfy 0.1.1

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

use super::Run;
use crate::io::Input;
use clap::Args;
use cmfy::dto;
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_t=Input::default())]
    input: Input,
}

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