From 9a6127d6bc33e5a3b44aeeaf88f5842d38b32ecc Mon Sep 17 00:00:00 2001
From: Richard Mwewa <74001397+rly0nheart@users.noreply.github.com>
Date: Fri, 25 Nov 2022 02:35:37 +0200
Subject: [PATCH] Update colors.py

---
 octosuite/colors.py | 24 ++++++++++--------------
 1 file changed, 10 insertions(+), 14 deletions(-)

diff --git a/octosuite/colors.py b/octosuite/colors.py
index e3bfd8e..0d6edad 100644
--- a/octosuite/colors.py
+++ b/octosuite/colors.py
@@ -20,31 +20,27 @@ first_banner = f"""
             
 """
 
-print(first_banner)
-system_tree = Tree(platform.system())
-for system_key, system_value in system_info:
-    system_tree.add(f"{system_key}: {system_value}")
-xprint(system_tree)
+print(banner)
+print(f"\t{platform.system()}")
+for key, value in system_info:
+    print(f"\t├─ {key}: {value()}")
 print("\n")
 while True:
     try:
-        color_chooser = input(f"[?] Welcome, would you like to enable colors for this session? (yes/no) ").lower()
-        if color_chooser == "yes":
+        color_chooser = input(f"[ ? ] Welcome, would you like to enable colors for this session? (Y/n) ").lower()
+        if color_chooser == "y":
             header_title = "bold white"
             red = "[red]"
             white = "[white]"
-            green = "[green]"
-            red_bold = "[white bold]"
-            white_bold = "[white bold]"
+@@ -33,11 +38,11 @@
             green_bold = "[green bold]"
             reset = "[/]"
             break
-        elif color_chooser == "no":
+        elif color_chooser == "n":
             header_title = red = white = green = red_bold = white_bold = green_bold = reset = ""
             break
         else:
-            print(f"\n[!] Your response '{color_chooser}' is invalid (expected yes or no)")
+            print(f"\n[ ! ] Your response '{color_chooser}' is invalid (expected y or n)")
 
     except KeyboardInterrupt:
-        exit(f"[!] Process interrupted with Ctrl+C.")
-      
+        exit(f"[ ! ] Process interrupted with [Ctrl+C].")