From 3baabf5bc0b04347af85dafbb056f134258d9715 Mon Sep 17 00:00:00 2001
From: Matthias Bisping <matthias.bisping@iqser.com>
Date: Wed, 16 Mar 2022 14:20:30 +0100
Subject: [PATCH] added banner

---
 image_prediction/utils.py | 34 ++++++++++++++++++++++++++++++++++
 src/serve.py              |  4 +++-
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/image_prediction/utils.py b/image_prediction/utils.py
index f03ff69..b48af9a 100644
--- a/image_prediction/utils.py
+++ b/image_prediction/utils.py
@@ -33,3 +33,37 @@ def make_logger_getter():
 
 
 get_logger = make_logger_getter()
+
+
+def show_banner():
+    banner = \
+    '''
+    .....     .                            ...                ..          
+  .d88888Neu. 'L                        xH88"`~ .x8X    x .d88"    oec :  
+  F""""*8888888F    ..    .     :     :8888   .f"8888Hf  5888R    @88888  
+ *      `"*88*"   .888: x888  x888.  :8888>  X8L  ^""`   '888R    8"*88%  
+  -....    ue=:. ~`8888~'888X`?888f` X8888  X888h         888R    8b.     
+         :88N  `   X888  888X '888>  88888  !88888.       888R   u888888> 
+         9888L     X888  888X '888>  88888   %88888       888R    8888R   
+  uzu.   `8888L    X888  888X '888>  88888 '> `8888>      888R    8888P   
+,""888i   ?8888    X888  888X '888>  `8888L %  ?888   !   888R    *888>   
+4  9888L   %888>  "*88%""*88" '888!`  `8888  `-*""   /   .888B .  4888    
+'  '8888   '88%     `~    "    `"`      "888.      :"    ^*888%   '888    
+     "*8Nu.z*"                            `""***~"`        "%      88R    
+                                                                   88>    
+                                                                   48     
+                                                                   '8     
+    '''
+
+    logger = logging.getLogger(__name__)
+    logger.propagate = False
+
+    handler = logging.StreamHandler()
+    handler.setLevel(logging.INFO)
+
+    formatter = logging.Formatter("")
+
+    handler.setFormatter(formatter)
+    logger.addHandler(handler)
+
+    logger.info(banner)
diff --git a/src/serve.py b/src/serve.py
index 39afd1c..f44b632 100644
--- a/src/serve.py
+++ b/src/serve.py
@@ -6,7 +6,7 @@ from image_prediction.config import CONFIG
 from image_prediction.flask import make_prediction_server
 from image_prediction.predictor import Predictor
 from image_prediction.response import build_response
-from image_prediction.utils import get_logger
+from image_prediction.utils import get_logger, show_banner
 
 logger = get_logger()
 
@@ -43,4 +43,6 @@ if __name__ == "__main__":
     logging.getLogger("PIL").setLevel(logging.ERROR)
     logging.getLogger("h5py").setLevel(logging.ERROR)
 
+    show_banner()
+
     main()