From 029f7c1acfcfc64a5fb4b539ca33177885cae99c Mon Sep 17 00:00:00 2001
From: Gianpaolo <gian.impronta@gmail.com>
Date: Thu, 22 Oct 2020 14:51:10 -0300
Subject: [PATCH] Fixed color KeyError

---
 websploit/core/utils/output.py |  6 +++++-
 websploit/core/utils/tui.py    | 15 ++++++++++++---
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/websploit/core/utils/output.py b/websploit/core/utils/output.py
index 5d0d89c..43491f9 100644
--- a/websploit/core/utils/output.py
+++ b/websploit/core/utils/output.py
@@ -16,7 +16,11 @@ def _output(self, text, status):
             "success": "[+]",
             "warning": "[!]",
             "error": "[-]",
-            "info": "[>]"
+            "info": "[>]",
+            "yellow": "",
+            "green": "",
+            "red": "",
+            "cyan": ""
         }
 
         if self._color_status:
diff --git a/websploit/core/utils/tui.py b/websploit/core/utils/tui.py
index 19b6401..03cb265 100644
--- a/websploit/core/utils/tui.py
+++ b/websploit/core/utils/tui.py
@@ -1,9 +1,18 @@
 from .version import version
-
+import platform
 
 def logo():
+    windows =  f"""
+    ____    __    ____
+    \   \  /  \  /   /     |    Welcome to Websploit 
+     \   \/    \/   /      |    Version : {version} 
+      \            /       |    https://github.com/websploit/websploit 
+       \    /\    /        |    Author : Fardin Allahverdinazhand 
+        \__/  \__/         |    Codename : Reborn 
 
-    return f"""\033[92m
+    
+    """
+    not_windows =  f"""\033[92m
     ____    __    ____
     \   \  /  \  /   /     |   \033[95m Welcome to Websploit \033[92m
      \   \/    \/   /      |   \033[95m Version : {version} \033[92m
@@ -13,4 +22,4 @@ def logo():
 
     \033[0m
     """
-
+    return windows if platform.system() == "Windows" else not_windows