use openai_gpt_rs::models::EditModels;
use openai_gpt_rs::{client::Client, response::Content};
use std::env;
use std::io::{stdin, stdout, Write};
#[tokio::main]
async fn main() {
let mut prompt = String::new();
let mut instruction = String::new();
print!("Enter a prompt: ");
let _ = stdout().flush();
stdin().read_line(&mut prompt).unwrap();
print!("Enter the instruction: ");
let _ = stdout().flush();
stdin().read_line(&mut instruction).unwrap();
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().as_str());
let resp = client
.create_edit(|args| {
args.input(prompt)
.instruction(instruction)
.model(EditModels::TextDavinciEdit1)
.n(1)
.temperature(1.0)
})
.await
.unwrap();
let text = resp.get_content(0).unwrap();
println!("{}", text);
}