try to use rich package for more structured console output!

predicators/args.py

@@ -43,6 +43,7 @@ def create_arg_parser(env_required: bool = True,
     parser.add_argument("--experiment_id", default="", type=str)
     parser.add_argument("--load_experiment_id", default="", type=str)
     parser.add_argument("--log_file", default="", type=str)
+    parser.add_argument("--log_rich", default="true", type=str)
     parser.add_argument("--use_gui", action="store_true")
     parser.add_argument('--debug',
                         action="store_const",

predicators/main.py

@@ -71,8 +71,14 @@ def main() -> None:
     args = utils.parse_args()
     utils.update_config(args)
     str_args = " ".join(sys.argv)
-    # Log to stderr.
-    handlers: List[logging.Handler] = [logging.StreamHandler()]
+    # Log to stderr or use `rich` package for more structured output.
+    handlers: List[logging.Handler] = []
+    if CFG.log_rich:
+        from rich.logging import RichHandler
+        handlers.append(RichHandler())
+    else:
+        handlers.append(logging.StreamHandler())
+
     if CFG.log_file:
         handlers.append(logging.FileHandler(CFG.log_file, mode='w'))
     logging.basicConfig(level=CFG.loglevel,

setup.py

@@ -42,7 +42,8 @@
         "opencv-python == 4.7.0.72",
         "pg3@git+https://github.com/tomsilver/pg3.git",
         "gym_sokoban@git+https://github.com/Learning-and-Intelligent-Systems/gym-sokoban.git",  # pylint: disable=line-too-long
-        "pbrspot@git+https://github.com/NishanthJKumar/pbrspot.git"
+        "pbrspot@git+https://github.com/NishanthJKumar/pbrspot.git",
+        "rich",
     ],
     include_package_data=True,
     extras_require={