Improved Running and Initializing banner

Author: UtkarshTheDev

CHANGELOG.md

@@ -2,6 +2,14 @@
 
 All notable changes to LocalLab will be documented in this file.
 
+## [0.5.5] - 2024-04-30
+
+### Fixed
+
+- Fixed extra spacing in the boundary of status banners
+- Improved alignment of INITIALIZING and RUNNING status boxes
+- Enhanced visual consistency across all UI elements
+
 ## [0.5.4] - 2024-04-30
 
 ### Improved

locallab/__init__.py

@@ -2,7 +2,7 @@
 LocalLab - A lightweight AI inference server for running LLMs locally
 """
 
-__version__ = "0.5.4"  # Updated to match setup.py
+__version__ = "0.5.5"  # Updated to match setup.py
 
 # Only import what's necessary initially, lazy-load the rest
 from .logger import get_logger

locallab/ui/banners.py

@@ -28,14 +28,13 @@ def print_initializing_banner(version: str = "0.4.25"):
   ╚══════╝ ╚═════╝  ╚═════╝╚═╝  ╚═╝╚══════╝╚══════╝╚═╝  ╚═╝╚═════╝
 {Style.RESET_ALL}
 
-{Fore.YELLOW}
-  ╔═══════════════════════════════════════════════════════════════╗
-  ║                      ⚠️  INITIALIZING  ⚠️                      ║
-  ║                                                               ║
-  ║  • Server is starting up - please wait                        ║
-  ║  • Do not make API requests yet                               ║
-  ║  • Wait for the "RUNNING" banner to appear                    ║
-  ╚═══════════════════════════════════════════════════════════════╝
+{Fore.YELLOW}▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄
+                      ⚠️  INITIALIZING  ⚠️
+
+  • Server is starting up - please wait
+  • Do not make API requests yet
+  • Wait for the "RUNNING" banner to appear
+▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀
 {Style.RESET_ALL}
 
 {Fore.CYAN}════════════════════════════════════════════════════════════════════════{Style.RESET_ALL}
@@ -67,15 +66,14 @@ def print_running_banner(version: str):
   ███████╗╚██████╔╝╚██████╗██║  ██║███████╗███████╗██║  ██║██████╔╝
   ╚══════╝ ╚═════╝  ╚═════╝╚═╝  ╚═╝╚══════╝╚══════╝╚═╝  ╚═╝╚═════╝
 {Style.RESET_ALL}
-{Fore.GREEN}
-  ╔═══════════════════════════════════════════════════════════════╗
-  ║                        ✅  RUNNING  ✅                         ║
-  ║                                                               ║
-  ║  • Server is ready - you can now make API requests            ║
-  ║  • Prefer to use the client packages for easier interaction   ║
-  ║  • Model loading will continue in the background              ║
-  ║  • API documentation is available below                       ║
-  ╚═══════════════════════════════════════════════════════════════╝
+{Fore.GREEN}▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄
+                        ✅  RUNNING  ✅
+
+  • Server is ready - you can now make API requests
+  • Prefer to use the client packages for easier interaction
+  • Model loading will continue in the background
+  • API documentation is available below
+▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀▀
 {Style.RESET_ALL}
 
 {Fore.CYAN}════════════════════════════════════════════════════════════════════════{Style.RESET_ALL}

locallab/utils/networking.py

@@ -53,24 +53,24 @@ def setup_ngrok(port: int) -> Optional[str]:
         url_length = len(public_url)
         banner_width = max(80, url_length + 20)  # Add padding for aesthetics
 
-        # Create dynamic width horizontal lines
-        h_line = "═" * (banner_width - 2)
+        # Create dynamic width horizontal lines with thicker borders
+        h_line_top = "▄" * banner_width
+        h_line_bottom = "▀" * banner_width
 
-        # Create centered title with proper padding
+        # Create centered title
         title = "✨ NGROK TUNNEL ACTIVE ✨"
-        title_padding = (banner_width - len(title) - 2) // 2
+        title_padding = (banner_width - len(title)) // 2
         padded_title = " " * title_padding + title + " " * title_padding
         # Adjust if odd number
-        if len(padded_title) < banner_width - 2:
+        if len(padded_title) < banner_width:
             padded_title += " "
 
-        # Display banner
+        # Display banner with no side borders and prominent top/bottom borders
         logger.info(f"""
-{Fore.CYAN}╔{h_line}╗{Style.RESET_ALL}
-{Fore.CYAN}║{padded_title}║{Style.RESET_ALL}
-{Fore.CYAN}╠{h_line}╣{Style.RESET_ALL}
-{Fore.CYAN}║{Style.RESET_ALL} {Fore.GREEN}Public URL:{Style.RESET_ALL} {Fore.YELLOW}{public_url}{Style.RESET_ALL}{" " * (banner_width - len(public_url) - 14)}{Fore.CYAN}║{Style.RESET_ALL}
-{Fore.CYAN}╚{h_line}╝{Style.RESET_ALL}
+{Fore.CYAN}{h_line_top}{Style.RESET_ALL}
+{Fore.CYAN}{padded_title}{Style.RESET_ALL}
+{Fore.GREEN}Public URL: {Fore.YELLOW}{public_url}{Style.RESET_ALL}
+{Fore.CYAN}{h_line_bottom}{Style.RESET_ALL}
 """)
         return public_url
 

setup.py

@@ -47,7 +47,7 @@
 
 setup(
     name="locallab",
-    version="0.5.4",
+    version="0.5.5",
     packages=find_packages(include=["locallab", "locallab.*"]),
     install_requires=install_requires,
     extras_require={