From 5b728ce80346eee82e3e765e2f794d640495f7c2 Mon Sep 17 00:00:00 2001
From: ZeroCool940711 <alejandrogilelias940711@gmail.com>
Date: Sat, 15 Oct 2022 02:17:31 -0700
Subject: [PATCH] Fixed logger.level() error when using the logger module with
 streamlit. It was getting loaded twice and so giving some errors because of
 this.

---
 scripts/logger.py | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/scripts/logger.py b/scripts/logger.py
index 6c94d30..087a611 100644
--- a/scripts/logger.py
+++ b/scripts/logger.py
@@ -70,15 +70,18 @@ genfmt = "<level>{level: <10}</level> @ <green>{time:YYYY-MM-DD HH:mm:ss}</green
 initfmt = "<magenta>INIT      </magenta> | <level>{extra[status]: <10}</level> | <magenta>{message}</magenta>"
 msgfmt = "<level>{level: <10}</level> | <level>{message}</level>"
 
-logger.level("GENERATION", no=24, color="<cyan>")
-logger.level("PROMPT", no=23, color="<yellow>")
-logger.level("INIT", no=31, color="<white>")
-logger.level("INIT_OK", no=31, color="<green>")
-logger.level("INIT_WARN", no=31, color="<yellow>")
-logger.level("INIT_ERR", no=31, color="<red>")
-# Messages contain important information without which this application might not be able to be used
-# As such, they have the highest priority
-logger.level("MESSAGE", no=61, color="<green>")
+try:
+    logger.level("GENERATION", no=24, color="<cyan>")
+    logger.level("PROMPT", no=23, color="<yellow>")
+    logger.level("INIT", no=31, color="<white>")
+    logger.level("INIT_OK", no=31, color="<green>")
+    logger.level("INIT_WARN", no=31, color="<yellow>")
+    logger.level("INIT_ERR", no=31, color="<red>")
+    # Messages contain important information without which this application might not be able to be used
+    # As such, they have the highest priority
+    logger.level("MESSAGE", no=61, color="<green>")
+except TypeError:
+    pass
 
 logger.__class__.generation = partialmethod(logger.__class__.log, "GENERATION")
 logger.__class__.prompt = partialmethod(logger.__class__.log, "PROMPT")