diff --git a/aiserver.py b/aiserver.py index 45ab66d0..460cac02 100644 --- a/aiserver.py +++ b/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) diff --git a/logger.py b/logger.py index 2557d730..8da2aa7e 100644 --- a/logger.py +++ b/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