smartui 0.2.0

A terminal utility that integrates with Google Gemini API
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
53
54

use crate::error::Result;
use crate::gemini::GeminiClientTrait;
use crate::tts::TTSManager;
use async_trait::async_trait;

#[async_trait]
pub trait GenerateContent {
    async fn generate_content(&self, prompt: &str) -> Result<String>;
}

pub struct GenerateTestCommand<T: GeminiClientTrait> {
    client: T,
}

impl<T: GeminiClientTrait> GenerateTestCommand<T> {
    pub fn new(client: T) -> Self {
        Self { client }
    }
}

#[async_trait]
impl<T: GeminiClientTrait + Send + Sync> GenerateContent for GenerateTestCommand<T> {
    async fn generate_content(&self, prompt: &str) -> Result<String> {
        let test_prompt = format!(
            "Generate test cases for the following code:\n\n{}",
            prompt
        );
        
        self.client.generate_content(test_prompt).await
    }
}

pub async fn execute<T: GeminiClientTrait + Send + Sync>(
    client: T,
    code: String,
    language: Option<String>,
    tts_manager: Option<&TTSManager>,
) -> Result<()> {
    let generate_command = GenerateTestCommand::new(client);
    let prompt = if let Some(lang) = language {
        format!("Language: {}\n\nCode:\n{}", lang, code)
    } else {
        code
    };
    let tests = generate_command.generate_content(&prompt).await?;
    println!("{}", tests);

    if let Some(tts) = tts_manager {
        tts.speak(&tests).await?;
    }

    Ok(())
}