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
55
56
57
58

use crate::error::Result;
use crate::gemini::GeminiClientTrait;
use crate::tts::TTSManager;
use async_trait::async_trait;
use std::path::PathBuf;
use std::fs;

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

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

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

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

pub async fn execute<T: GeminiClientTrait + Send + Sync>(
    client: T,
    file: PathBuf,
    language: Option<String>,
    tts_manager: Option<&TTSManager>,
) -> Result<()> {
    let code = fs::read_to_string(&file)?;
    let explain_command = ExplainCodeCommand::new(client);
    
    let prompt = if let Some(lang) = language {
        format!("Explain this {} code:\n\n{}", lang, code)
    } else {
        format!("Explain this code:\n\n{}", code)
    };
    
    let explanation = explain_command.generate_content(&prompt).await?;
    println!("{}", explanation);

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

    Ok(())
}