use std::sync::Arc;

use gemini_rs::prelude::*;

use gcp_auth::AuthenticationManager;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let authentication_manager = Arc::new(AuthenticationManager::new().await?);
    let api_endpoint = std::env::var("API_ENDPOINT")?;
    let project_id = std::env::var("PROJECT_ID")?;
    let location_id = std::env::var("LOCATION_ID")?;

    let gemini = GeminiClient::new(
        authentication_manager,
        api_endpoint,
        project_id,
        location_id,
    );

    let prompt = "Tell me the story of the genesis of the universe as a bedtime story.";
    let request = GenerateContentRequest::from_prompt(prompt, None);
    let queue = gemini
        .stream_generate_content(&request, Model::GeminiPro)
        .await;

    while let Some(response) = queue.pop().await {
        if let GenerateContentResponse::Ok {
            candidates,
            usage_metadata: _,
        } = response
        {
            let text = candidates
                .iter()
                .filter_map(|c| c.get_text())
                .collect::<String>();
            print!("{}", text);
        }
    }

    Ok(())
}