From 23270e8da94d22c5cf02a0552b6887ad8ffb4be5 Mon Sep 17 00:00:00 2001
From: Richard Mwewa <74001397+rly0nheart@users.noreply.github.com>
Date: Thu, 14 Apr 2022 11:15:18 +0200
Subject: [PATCH] Update octosuite

---
 octosuite | 25 +++++++++++--------------
 1 file changed, 11 insertions(+), 14 deletions(-)

diff --git a/octosuite b/octosuite
index 73f0c9b..b5bfd97 100644
--- a/octosuite
+++ b/octosuite
@@ -2,19 +2,16 @@
 
 import logging
 from src.main import *
-from lib.colors import red,white,green,reset
+from lib.colors import red,red_bg,white,white_bg,green_bg,green,reset
 
 if __name__ == '__main__':
-    while True:
-        try:
-        	octosuite().on_start()
-        	
-        except KeyboardInterrupt:
-        	logging.info('Process interrupted with Ctrl+C')
-        	exit(f'\n{white}[{red}x{white}] Process interrupted with {red}Ctrl{white}+{red}C{reset}')
-        	
-        	
-        except Exception as e:
-        	logging.error(f'An error occured: {e}')
-        	print(f'{white}[{red}!{white}] An error occured: {red}{e}{reset}')
-        input(f'\n{white}^ Press any key to continue{reset} ')
+    try:
+        octosuite().on_start()
+       
+    except KeyboardInterrupt:
+        logging.warning('Process interrupted with Ctrl+C')
+        exit(f'\n{white}[{red} x {white}] Process interrupted with {red_bg}Ctrl+C{reset}')
+        
+    except Exception as e:
+        logging.error(f'Session terminated on error: {e}')
+        exit(f'\n{white}[{red} ! {white}] Session {red_bg}terminated{reset}{white} on error: {red}{e}{reset}')