Fine-tune display_image() in main.rs

README.md

@@ -1,5 +1,10 @@
 # tnap - Let's take a nap 💤
 
+## What's `tnap`?
+
+`tnap` is the screen save for TUI.
+You can use sample themes for tnap and generate image with default prompts or your own prompts.
+
 ## Examples
 
 ### Prompt
@@ -17,19 +22,21 @@
 ## Usage
 
 ```
-Generate image with DALL-E and print it
+You can use sample themes for tnap and generate image with default prompts or your own prompts
 
 Usage: tnap [OPTIONS]
 
 Options:
-  -p, --prompt <PROMPT>  Prompt to pass to DALL-E
+  -t, --theme <THEME>    Use the sample theme without generating images
+  -k, --key <KEY>        Generate Image by looking up the corresponding value in config.toml using the subsequent string as a key and using it as a prompt
+  -p, --prompt <PROMPT>  Generate images with user-considered prompt
   -a, --ascii            Convert an image to ASCII art
   -h, --help             Print help
   -V, --version          Print version
 ```
 
 ```sh
-$ cargo run 
+cargo run
 ```
 
 ## License

src/main.rs

@@ -1,5 +1,6 @@
 use anyhow::Result;
 use clap::Parser;
+use convert_image_to_ascii::convert_image_to_ascii;
 use dotenv::dotenv;
 use generate_image::{download_image, generate_image};
 use std::fs;
@@ -85,7 +86,11 @@ fn display_generated_image_from_prompt(prompt: &str, ascii: bool) -> Result<()>
 
 fn display_image(path: &str, ascii: bool) -> Result<()> {
     if ascii {
-        convert_image_to_ascii(&path)?;
+        let ascii_art = convert_image_to_ascii(Path::new(&path));
+        match ascii_art {
+            Ok(art) => println!("{}", art), // 成功した場合、ASCII アートを出力
+            Err(e) => println!("Error converting image to ASCII: {:?}", e), // エラーが発生した場合、エラーメッセージを出力
+        }
     } else {
         println!("Displaying image: {}", path);
         // render_image(&path)?;