Improved Color Contrast and Ngrok banner

Author: UtkarshTheDev

CHANGELOG.md

@@ -2,6 +2,16 @@
 
 All notable changes to LocalLab will be documented in this file.
 
+## [0.5.4] - 2024-04-30
+
+### Improved
+
+- Enhanced log coloring with lighter shades for better readability
+- Redesigned ngrok tunnel banner with dynamic width to accommodate long URLs
+- Improved visual aesthetics of the ngrok tunnel banner with modern styling
+- Added automatic width adjustment for banners based on content length
+- Fine-tuned color scheme to ensure all logs remain visible while not competing with important banners
+
 ## [0.5.3] - 2024-04-30
 
 ### Improved

locallab/__init__.py

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

locallab/logger/__init__.py

@@ -57,12 +57,15 @@ class Colors:
     BRIGHT_YELLOW = Fore.YELLOW
     BRIGHT_RED = Fore.RED
 
-    # Subdued colors for regular logs
-    SUBDUED_DEBUG = Fore.BLUE + Style.DIM  # Dimmed blue for debug
-    SUBDUED_INFO = Fore.WHITE + Style.DIM   # Dimmed white for info
+    # Subdued colors for regular logs - using lighter shades
+    SUBDUED_DEBUG = Fore.CYAN + Style.DIM  # Dimmed cyan for debug
+    SUBDUED_INFO = Fore.WHITE  # White for info (not dimmed)
     SUBDUED_WARNING = Fore.YELLOW + Style.DIM  # Dimmed yellow for warnings
     SUBDUED_ERROR = Fore.RED + Style.DIM  # Dimmed red for errors
 
+    # Special colors for very unimportant logs
+    VERY_SUBDUED = Fore.LIGHTBLACK_EX  # Light gray for very unimportant logs
+
     # Reset
     RESET = Style.RESET_ALL
 
@@ -122,8 +125,8 @@ def format(self, record):
 
         # Determine the appropriate color based on log level and importance
         if is_ngrok_log or is_uvicorn_log:
-            # Make ngrok and uvicorn logs very subdued (dark gray)
-            color = Fore.BLACK + Style.BRIGHT  # This creates a dark gray color on most terminals
+            # Make ngrok and uvicorn logs subdued but still readable (light gray)
+            color = Colors.VERY_SUBDUED
         elif record.levelno == logging.DEBUG:
             color = Colors.BRIGHT_CYAN if is_important else Colors.SUBDUED_DEBUG
         elif record.levelno == logging.INFO:

locallab/utils/networking.py

@@ -22,43 +22,58 @@ def setup_ngrok(port: int) -> Optional[str]:
     try:
         from pyngrok import ngrok, conf
         from colorama import Fore, Style
-        
+
         # Get ngrok token using the standardized function
         auth_token = get_ngrok_token()
-        
+
         if not auth_token:
             logger.error(f"{Fore.RED}Ngrok auth token not found. Please configure it using 'locallab config'{Style.RESET_ALL}")
             return None
-            
+
         # Ensure token is properly set in environment
         set_env_var(NGROK_TOKEN_ENV, auth_token)
-        
+
         # Configure ngrok
         ngrok.set_auth_token(auth_token)
         conf.get_default().auth_token = auth_token
-        
+
         # Start tunnel with simplified configuration
         tunnel = ngrok.connect(
             addr=port,
             proto="http",
             bind_tls=True  # Enable HTTPS
         )
-        
+
         public_url = tunnel.public_url
-        
+
         # Store the URL in environment for clients
         os.environ["LOCALLAB_NGROK_URL"] = public_url
-        
+
+        # Calculate banner width based on URL length (minimum 80 characters)
+        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 centered title with proper padding
+        title = "✨ NGROK TUNNEL ACTIVE ✨"
+        title_padding = (banner_width - len(title) - 2) // 2
+        padded_title = " " * title_padding + title + " " * title_padding
+        # Adjust if odd number
+        if len(padded_title) < banner_width - 2:
+            padded_title += " "
+
         # Display banner
         logger.info(f"""
-{Fore.GREEN}╔═══════════════════════════════════════════════════════════════════╗{Style.RESET_ALL}
-{Fore.GREEN}                      ✨ NGROK TUNNEL ACTIVE ✨                      {Style.RESET_ALL}
-{Fore.GREEN}╠═══════════════════════════════════════════════════════════════════╣{Style.RESET_ALL}
-{Fore.CYAN}  Public URL: {Fore.YELLOW}{public_url}{Style.RESET_ALL}
-{Fore.GREEN}╚═══════════════════════════════════════════════════════════════════╝{Style.RESET_ALL}
+{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}
 """)
         return public_url
-        
+
     except Exception as e:
         logger.error(f"{Fore.RED}Failed to setup ngrok: {str(e)}{Style.RESET_ALL}")
         logger.info(f"{Fore.YELLOW}Please check your ngrok token using 'locallab config'{Style.RESET_ALL}")
@@ -91,7 +106,7 @@ def get_network_interfaces() -> List[Dict[str, str]]:
             })
         except Exception:
             pass
-            
+
     return interfaces
 
 def get_public_ip() -> Optional[str]:
@@ -105,15 +120,15 @@ def get_public_ip() -> Optional[str]:
 def get_network_interfaces() -> dict:
     """
     Get information about available network interfaces
-    
+
     Returns:
         A dictionary with interface names as keys and their addresses as values
     """
     interfaces = {}
     try:
         import socket
         import netifaces
-        
+
         for interface in netifaces.interfaces():
             addresses = netifaces.ifaddresses(interface)
             # Get IPv4 addresses if available
@@ -143,13 +158,13 @@ def get_network_interfaces() -> dict:
     except Exception as e:
         logger.warning(f"Failed to get network interface information: {str(e)}")
         interfaces["error"] = str(e)
-    
+
     return interfaces
 
 async def get_public_ip() -> str:
     """
     Get the public IP address of this machine
-    
+
     Returns:
         The public IP address as a string, or an empty string if it cannot be determined
     """
@@ -159,7 +174,7 @@ async def get_public_ip() -> str:
         "https://checkip.amazonaws.com",
         "https://ipinfo.io/ip"
     ]
-    
+
     try:
         # Try to use httpx for async requests
         import httpx
@@ -184,6 +199,6 @@ async def get_public_ip() -> str:
                     continue
         except ImportError:
             logger.warning(f"{Fore.YELLOW}Neither httpx nor requests packages found. Cannot determine public IP.{Style.RESET_ALL}")
-    
+
     # If we couldn't get the IP from any service, return empty string
     return ""

setup.py

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