added verbosity controls

aiserver.py

@@ -17,7 +17,7 @@ os.environ['TOKENIZERS_PARALLELISM'] = 'false'
 from eventlet import tpool
 
 import logging
-from logger import logger
+from logger import logger, set_logger_verbosity, quiesce_logger
 
 logging.getLogger("urllib3").setLevel(logging.ERROR)
 
@@ -1325,6 +1325,9 @@ def general_startup(override_args=None):
     parser.add_argument("--savemodel", action='store_true', help="Saves the model to the models folder even if --colab is used (Allows you to save models to Google Drive)")
     parser.add_argument("--customsettings", help="Preloads arguements from json file. You only need to provide the location of the json file. Use customsettings.json template file. It can be renamed if you wish so that you can store multiple configurations. Leave any settings you want as default as null. Any values you wish to set need to be in double quotation marks")
     parser.add_argument("--no_ui", action='store_true', default=False, help="Disables the GUI and Socket.IO server while leaving the API server running.")
+    parser.add_argument('-v', '--verbosity', action='count', default=0, help="The default logging level is ERROR or higher. This value increases the amount of logging seen in your screen")
+    parser.add_argument('-q', '--quiesce', action='count', default=0, help="The default logging level is ERROR or higher. This value decreases the amount of logging seen in your screen")
+
     #args: argparse.Namespace = None
     if "pytest" in sys.modules and override_args is None:
         args = parser.parse_args([])
@@ -1338,6 +1341,8 @@ def general_startup(override_args=None):
     else:
         args = parser.parse_args()
 
+    set_logger_verbosity(args.verbosity)
+    quiesce_logger(args.quiesce)
     if args.customsettings:
         f = open (args.customsettings)
         importedsettings = json.load(f)

logger.py

@@ -6,7 +6,7 @@ STDOUT_LEVELS = ["GENERATION", "PROMPT"]
 INIT_LEVELS = ["INIT", "INIT_OK", "INIT_WARN", "INIT_ERR"]
 MESSAGE_LEVELS = ["MESSAGE"]
 # By default we're at error level or higher
-verbosity = 40
+verbosity = 20
 quiet = 0
 
 def set_logger_verbosity(count):
@@ -14,7 +14,7 @@ def set_logger_verbosity(count):
     # The count comes reversed. So count = 0 means minimum verbosity
     # While count 5 means maximum verbosity
     # So the more count we have, the lowe we drop the versbosity maximum
-    verbosity = 40 - (count * 10)
+    verbosity = 20 - (count * 10)
 
 def quiesce_logger(count):
     global quiet