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

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 ExplainErrorCommand<T: GeminiClientTrait> {
    client: T,
}

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

#[async_trait]
impl<T: GeminiClientTrait + Send + Sync> GenerateContent for ExplainErrorCommand<T> {
    async fn generate_content(&self, prompt: &str) -> Result<String> {
        let error_prompt = format!(
            "Explain the following error message and suggest possible solutions:\n\n{}",
            prompt
        );
        
        self.client.generate_content(error_prompt).await
    }
}

pub async fn execute<T: GeminiClientTrait + Send + Sync>(
    client: T,
    error: String,
    stdin: bool,
    tts_manager: Option<&TTSManager>,
) -> Result<()> {
    let explain_command = ExplainErrorCommand::new(client);
    let prompt = if stdin {
        error
    } else {
        format!("Error: {}", error)
    };
    let explanation = explain_command.generate_content(&prompt).await?;
    println!("{}", explanation);

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

    Ok(())
}