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
44
45
46
47
48
49
50
51
52

use clap::Args;
use cmfy::{Client, Result};
use itertools::Itertools;

use super::Run;

/// Open images from completed prompts in a browser
#[derive(Debug, Args)]
pub struct View {
    /// Select a range of prompt indices,
    /// e.g. '1,2,3' or '4-5' or '1,3,4-6'
    #[clap(action, default_value = None)]
    range: Option<String>,

    /// Remove prompts from history after
    #[clap(short, long, action, default_value_t = false)]
    clear: bool,
}

impl Run for View {
    async fn run(self, client: Client) -> Result<()> {
        let history = client.history().await?;
        let entries = if let Some(range) = self.range {
            let range: Vec<u64> = range_parser::parse(&range)?;
            history
                .into_iter()
                .filter(|entry| range.contains(&entry.prompt.index))
                .collect_vec()
        } else {
            history.into_iter().collect_vec()
        };

        let mut set = tokio::task::JoinSet::new();
        for entry in &entries {
            for image in entry.outputs.images() {
                let url = client.url_for_image(image)?;
                set.spawn(async move {
                    println!("{}", url);
                    open::that(url.to_string())
                });
            }
        }
        set.join_all().await.into_iter().collect::<std::io::Result<Vec<_>>>()?;

        if self.clear {
            for entry in &entries {
                client.delete_from_history(&entry.prompt.uuid).await?;
            }
        }
        Ok(())
    }
}