bug fixes

Author: ppeuchin

option_validator.py

@@ -1,3 +1,5 @@
+from colorama.ansi import Fore, Style
+from colours import print_with_color
 import time
 import sys
 from extras import load_a, load_b,clear
@@ -19,7 +21,7 @@ def options():
         key = int(input("Enter the key: "))
         print()
         time.sleep(1)
-        print("--------------------------")
+        print_with_color("--------------------------", color=Fore.YELLOW, brightness=Style.BRIGHT)
         load_a()
         ce.caesar_encryption(plaintext, key)
 
@@ -35,21 +37,26 @@ def options():
         key = int(input("Enter the key: "))
         print()
         time.sleep(1)
-        print("--------------------------")
+        print_with_color("--------------------------", color=Fore.YELLOW, brightness=Style.BRIGHT)
         load_b()
         cd.caesar_decryption(ciphertext, key)
 
     elif option == '3':
         time.sleep(1.5)
         clear()
-        msg = "Thank you! :)\nSee you next time!"
-        sys.exit(msg)
+        print_with_color('[~~~~~~~~~~~~~~~~~~~~~~~~]',color=Fore.MAGENTA)
+        time.sleep(1)
+        print("  Thank you! :)")
+        time.sleep(0.7)
+        print("      See you next time!")
+        time.sleep(0.5)
+        sys.exit()
 
     else:
         time.sleep(1.5)
-        print("Invalid Option.")
+        print_with_color("Invalid Option.", color=Fore.RED)
         time.sleep(0.5)
-        print("Try again!...")
+        print_with_color("Try again!...", color=Fore.RED)
         print()
         time.sleep(1)
         options()