feat: Rich formatting (#174)

* Use Rich to pretty-format the announcements in CLI

Author: nokchalatte

changes/174.feature

@@ -0,0 +1 @@
+Utilize Rich as a formatting library for server announcements

setup.py

@@ -19,6 +19,7 @@
     'multidict>=5.1.0',
     'python-dateutil>=2.8.2',
     'PyYAML~=5.4.1',
+    'rich~=10.5.0',
     'tabulate>=0.8.9',
     'tqdm>=4.61',
     'typing-extensions>=3.10.0',

src/ai/backend/client/cli/announcement.py

@@ -5,8 +5,7 @@
 
 import appdirs
 
-from .pretty import print_info
-
+from rich import print as rprint
 
 _printed_announcement = False
 
@@ -27,10 +26,10 @@ def announce(msg: str, only_once: bool = True) -> None:
     msg_hash = hasher.hexdigest()
 
     if not (last_state['hash'] == msg_hash and last_state['dismissed']):
-        print_info("The server has an announcement!", file=sys.stderr)
-        print('----------', file=sys.stderr)
-        print(msg, file=sys.stderr)
-        print('----------', file=sys.stderr)
+        rprint('[bold cyan]The server has an announcement![/]', file=sys.stderr)
+        rprint('[bold cyan]----------[/]', file=sys.stderr)
+        rprint(f'[bold]{msg}[/]', file=sys.stderr)
+        rprint('[bold cyan]----------[/]', file=sys.stderr)
     _printed_announcement = True
 
     last_state['hash'] = msg_hash