image/
image.rs

1use openai_gpt_rs::{args::ImageSize, client::Client, response::Content};
2use std::env;
3use std::io::{stdin, stdout, Write};
4
5#[tokio::main]
6async fn main() {
7    let mut prompt = String::new();
8
9    print!("Enter a prompt: ");
10    let _ = stdout().flush();
11
12    stdin().read_line(&mut prompt).unwrap();
13
14    println!("Generating image...\n");
15
16    let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());
17
18    let resp = client
19        .create_image(|args| args.prompt(prompt).size(ImageSize::Medium).n(1))
20        .await
21        .unwrap();
22
23    let url = resp.get_content(0).unwrap();
24
25    println!("Url: {}", url);
26}