From 1a7016f9b76e021fe999704493e3647629172872 Mon Sep 17 00:00:00 2001
From: Craig Mayhew <craig@mayhew.io>
Date: Mon, 29 Jan 2024 12:00:20 +0000
Subject: [PATCH] Refactored part of main() into new fn
 markdown_plaintext_or_error()

---
 src/main.rs | 35 +++++++++++++++++++++--------------
 1 file changed, 21 insertions(+), 14 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 130b4ec..b32672d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -194,6 +194,26 @@ fn args_read (args_setup: Command) -> (std::string::String, i32, f32, f32, bool)
     (prepend.to_owned(),max_tokens.to_owned(),temperature.to_owned(),top_p.to_owned(),render_markdown.to_owned())
 }
 
+/// # Render in Markdown, Plaintext or Error
+/// 
+/// Takes a result from the Reqwest API call
+fn markdown_plaintext_or_error (gpt_result: Result<String, reqwest::Error>, render_markdown: bool) {
+    match gpt_result {
+        Ok(markdown) => {
+            if render_markdown {
+                let mut skin = MadSkin::default();
+                skin.code_block.left_margin = 4;
+                skin.code_block.set_fgbg(gray(17), gray(3));
+                skin.set_fg(Yellow);
+                skin.print_text(&markdown)
+            } else {
+                println!("{}", &markdown)
+            }
+        },
+        Err(e) => eprintln!("Error: {}", e),
+    }
+}
+
 /// # Entry Point for Application
 /// 
 /// - Initializes the application
@@ -221,18 +241,5 @@ async fn main() {
         io::stdin().read_to_string(&mut input).expect("Failed to read from stdin");
     }
 
-    match send_to_gpt4(&input, parsed_arguments).await {
-        Ok(markdown) => {
-            if render_markdown {
-                let mut skin = MadSkin::default();
-                skin.code_block.left_margin = 4;
-                skin.code_block.set_fgbg(gray(17), gray(3));
-                skin.set_fg(Yellow);
-                skin.print_text(&markdown)
-            } else {
-                println!("{}", &markdown)
-            }
-        },
-        Err(e) => eprintln!("Error: {}", e),
-    }
+    markdown_plaintext_or_error(send_to_gpt4(&input, parsed_arguments).await, render_markdown);
 }