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

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

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

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

pub async fn execute<T: GeminiClientTrait + Send + Sync>(
    client: T,
    pattern: String,
    tts_manager: Option<&TTSManager>,
) -> Result<()> {
    let generate_command = GenerateRegexCommand::new(client);
    let regex = generate_command.generate_content(&pattern).await?;
    println!("{}", regex);

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

    Ok(())
}