Update package versions in Cargo.lock

Cargo.toml

@@ -19,6 +19,7 @@ futures = "0.3.30"
 shellexpand = "3.1.0"
 tokio-stream = "0.1.14"
 clap-stdin = "0.4.0"
+termimad = "0.29.0"
 
 [dependencies.google-generative-ai-rs]
 

src/cli.rs

@@ -67,6 +67,11 @@ pub fn create_cli_app() -> App<'static>{
                 .long("verbose")
                 .help("output more logs"),
         )
+        .arg(
+            Arg::with_name("rich")
+                .long("rich")
+                .help("output the response in rich terminal"),
+        )
         .arg(
             Arg::with_name("stream")
                 .long("stream")

src/main.rs

@@ -46,6 +46,7 @@ async fn run(matches: ArgMatches) -> Result<(), Box<dyn std::error::Error>> {
         .unwrap_or("~/.config/gemini-cli.toml");
 
     let is_stream = matches.contains_id("stream");
+    let is_rich = matches.contains_id("rich");
     let config = cli::read_config(config_path).await?;
 
     let token = matches
@@ -100,7 +101,11 @@ async fn run(matches: ArgMatches) -> Result<(), Box<dyn std::error::Error>> {
             .first()
             .and_then(|c| c.content.parts.first().and_then(|p| p.text.as_ref()))
         {
-            print!("{}", text);
+            if is_rich {
+                termimad::print_inline(text);
+            } else {
+                println!("{}", text);
+            }
         }
     }