fal 0.3.6

A Rust crate for the fal.ai API, including generated, typed functions for all public models
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

use std::{future::Future, io::Cursor};

use base64::prelude::*;
use image::DynamicImage;

use crate::FalError;

#[deprecated(note = "use `reqwest::get` and `image::load_from_memory` instead")]
#[allow(deprecated)]
impl crate::File {
    /// Convert the File to an image
    pub fn into_image(self) -> impl Future<Output = Result<image::DynamicImage, FalError>> {
        async move {
            let image_bytes = reqwest::get(&self.url).await?.bytes().await?;
            let output = image::load_from_memory(&image_bytes)?;
            Ok(output)
        }
    }

    /// Convert the file into raw bytes
    pub fn into_raw_image(self) -> impl Future<Output = Result<Vec<u8>, FalError>> {
        async move {
            let image_bytes = reqwest::get(&self.url).await?.bytes().await?;
            Ok(image_bytes.to_vec())
        }
    }
}

pub trait ToDataUrl {
    fn to_data_url(&self) -> String;
}

impl ToDataUrl for DynamicImage {
    /// Convert the image into a PNG data URL.
    /// This is a convenience function to help with image-to-X endpoints.
    fn to_data_url(&self) -> String {
        image_to_data_url(self)
    }
}

fn image_to_data_url(image: &image::DynamicImage) -> String {
    let mut buf = Vec::new();
    let mut writer = Cursor::new(&mut buf);
    image
        .write_to(&mut writer, image::ImageFormat::Png)
        .unwrap();
    let data = BASE64_STANDARD.encode(&buf);
    format!("data:image/png;base64,{}", data)
}