ngpt 2.3.4__py3-none-any.whl → 2.5.0__py3-none-any.whl

ngpt/cli.py

@@ -5,6 +5,23 @@ from .client import NGPTClient
 from .config import load_config, get_config_path, load_configs, add_config_entry, remove_config_entry
 from . import __version__
 
+# Try to import markdown rendering libraries
+try:
+    import rich
+    from rich.markdown import Markdown
+    from rich.console import Console
+    HAS_RICH = True
+except ImportError:
+    HAS_RICH = False
+
+# Try to import the glow command if available
+def has_glow_installed():
+    """Check if glow is installed in the system."""
+    import shutil
+    return shutil.which("glow") is not None
+
+HAS_GLOW = has_glow_installed()
+
 # ANSI color codes for terminal output
 COLORS = {
     "reset": "\033[0m",
@@ -68,6 +85,162 @@ if not HAS_COLOR:
     for key in COLORS:
         COLORS[key] = ""
 
+def has_markdown_renderer(renderer='auto'):
+    """Check if the specified markdown renderer is available.
+    
+    Args:
+        renderer (str): Which renderer to check: 'auto', 'rich', or 'glow'
+    
+    Returns:
+        bool: True if the renderer is available, False otherwise
+    """
+    if renderer == 'auto':
+        return HAS_RICH or HAS_GLOW
+    elif renderer == 'rich':
+        return HAS_RICH
+    elif renderer == 'glow':
+        return HAS_GLOW
+    else:
+        return False
+
+def show_available_renderers():
+    """Show which markdown renderers are available and their status."""
+    print(f"\n{COLORS['cyan']}{COLORS['bold']}Available Markdown Renderers:{COLORS['reset']}")
+    
+    if HAS_GLOW:
+        print(f"  {COLORS['green']}✓ Glow{COLORS['reset']} - Terminal-based Markdown renderer")
+    else:
+        print(f"  {COLORS['yellow']}✗ Glow{COLORS['reset']} - Not installed (https://github.com/charmbracelet/glow)")
+        
+    if HAS_RICH:
+        print(f"  {COLORS['green']}✓ Rich{COLORS['reset']} - Python library for terminal formatting (Recommended)")
+    else:
+        print(f"  {COLORS['yellow']}✗ Rich{COLORS['reset']} - Not installed (pip install rich)")
+        
+    if not HAS_GLOW and not HAS_RICH:
+        print(f"\n{COLORS['yellow']}To enable prettified markdown output, install one of the above renderers.{COLORS['reset']}")
+    else:
+        renderers = []
+        if HAS_RICH:
+            renderers.append("rich")
+        if HAS_GLOW:
+            renderers.append("glow")
+        print(f"\n{COLORS['green']}Usage examples:{COLORS['reset']}")
+        print(f"  ngpt --prettify \"Your prompt here\"                {COLORS['gray']}# Beautify markdown responses{COLORS['reset']}")
+        print(f"  ngpt -c --prettify \"Write a sort function\"        {COLORS['gray']}# Syntax highlight generated code{COLORS['reset']}")
+        if renderers:
+            renderer = renderers[0]
+            print(f"  ngpt --prettify --renderer={renderer} \"Your prompt\"  {COLORS['gray']}# Specify renderer{COLORS['reset']}")
+    
+    print("")
+
+def warn_if_no_markdown_renderer(renderer='auto'):
+    """Warn the user if the specified markdown renderer is not available.
+    
+    Args:
+        renderer (str): Which renderer to check: 'auto', 'rich', or 'glow'
+    
+    Returns:
+        bool: True if the renderer is available, False otherwise
+    """
+    if has_markdown_renderer(renderer):
+        return True
+    
+    if renderer == 'auto':
+        print(f"{COLORS['yellow']}Warning: No markdown rendering library available.{COLORS['reset']}")
+        print(f"{COLORS['yellow']}Install 'rich' package with: pip install rich{COLORS['reset']}")
+        print(f"{COLORS['yellow']}Or install 'glow' from https://github.com/charmbracelet/glow{COLORS['reset']}")
+    elif renderer == 'rich':
+        print(f"{COLORS['yellow']}Warning: Rich is not available.{COLORS['reset']}")
+        print(f"{COLORS['yellow']}Install with: pip install rich{COLORS['reset']}")
+    elif renderer == 'glow':
+        print(f"{COLORS['yellow']}Warning: Glow is not available.{COLORS['reset']}")
+        print(f"{COLORS['yellow']}Install from https://github.com/charmbracelet/glow{COLORS['reset']}")
+    else:
+        print(f"{COLORS['yellow']}Error: Invalid renderer '{renderer}'. Use 'auto', 'rich', or 'glow'.{COLORS['reset']}")
+    
+    return False
+
+def prettify_markdown(text, renderer='auto'):
+    """Render markdown text with beautiful formatting using either Rich or Glow.
+    
+    The function handles both general markdown and code blocks with syntax highlighting.
+    For code generation mode, it automatically wraps the code in markdown code blocks.
+    
+    Args:
+        text (str): Markdown text to render
+        renderer (str): Which renderer to use: 'auto', 'rich', or 'glow'
+        
+    Returns:
+        bool: True if rendering was successful, False otherwise
+    """
+    # For 'auto', prefer rich if available, otherwise use glow
+    if renderer == 'auto':
+        if HAS_RICH:
+            return prettify_markdown(text, 'rich')
+        elif HAS_GLOW:
+            return prettify_markdown(text, 'glow')
+        else:
+            return False
+    
+    # Use glow for rendering
+    elif renderer == 'glow':
+        if not HAS_GLOW:
+            print(f"{COLORS['yellow']}Warning: Glow is not available. Install from https://github.com/charmbracelet/glow{COLORS['reset']}")
+            # Fall back to rich if available
+            if HAS_RICH:
+                print(f"{COLORS['yellow']}Falling back to Rich renderer.{COLORS['reset']}")
+                return prettify_markdown(text, 'rich')
+            return False
+            
+        # Use glow
+        import tempfile
+        import subprocess
+        
+        with tempfile.NamedTemporaryFile(mode='w', suffix='.md', delete=False) as temp:
+            temp_filename = temp.name
+            temp.write(text)
+            
+        try:
+            # Execute glow on the temporary file
+            subprocess.run(["glow", temp_filename], check=True)
+            os.unlink(temp_filename)
+            return True
+        except Exception as e:
+            print(f"{COLORS['yellow']}Error using glow: {str(e)}{COLORS['reset']}")
+            os.unlink(temp_filename)
+            
+            # Fall back to rich if available
+            if HAS_RICH:
+                print(f"{COLORS['yellow']}Falling back to Rich renderer.{COLORS['reset']}")
+                return prettify_markdown(text, 'rich')
+            return False
+    
+    # Use rich for rendering
+    elif renderer == 'rich':
+        if not HAS_RICH:
+            print(f"{COLORS['yellow']}Warning: Rich is not available. Install with: pip install rich{COLORS['reset']}")
+            # Fall back to glow if available
+            if HAS_GLOW:
+                print(f"{COLORS['yellow']}Falling back to Glow renderer.{COLORS['reset']}")
+                return prettify_markdown(text, 'glow')
+            return False
+            
+        # Use rich
+        try:
+            console = Console()
+            md = Markdown(text)
+            console.print(md)
+            return True
+        except Exception as e:
+            print(f"{COLORS['yellow']}Error using rich for markdown: {str(e)}{COLORS['reset']}")
+            return False
+    
+    # Invalid renderer specified
+    else:
+        print(f"{COLORS['yellow']}Error: Invalid renderer '{renderer}'. Use 'auto', 'rich', or 'glow'.{COLORS['reset']}")
+        return False
+
 # Custom help formatter with color support
 class ColoredHelpFormatter(argparse.HelpFormatter):
     """Help formatter that properly handles ANSI color codes without breaking alignment."""
@@ -302,13 +475,20 @@ def show_config_help():
     print(f"  5. {COLORS['cyan']}Use --config-index to specify which configuration to use or edit:{COLORS['reset']}")
     print(f"     {COLORS['yellow']}ngpt --config-index 1 \"Your prompt\"{COLORS['reset']}")
     
-    print(f"  6. {COLORS['cyan']}Use --config without arguments to add a new configuration:{COLORS['reset']}")
+    print(f"  6. {COLORS['cyan']}Use --provider to specify which configuration to use by provider name:{COLORS['reset']}")
+    print(f"     {COLORS['yellow']}ngpt --provider Gemini \"Your prompt\"{COLORS['reset']}")
+    
+    print(f"  7. {COLORS['cyan']}Use --config without arguments to add a new configuration:{COLORS['reset']}")
     print(f"     {COLORS['yellow']}ngpt --config{COLORS['reset']}")
-    print(f"     Or specify an index to edit an existing configuration:")
+    print(f"     Or specify an index or provider to edit an existing configuration:")
     print(f"     {COLORS['yellow']}ngpt --config --config-index 1{COLORS['reset']}")
-    print(f"  7. {COLORS['cyan']}Remove a configuration at a specific index:{COLORS['reset']}")
+    print(f"     {COLORS['yellow']}ngpt --config --provider Gemini{COLORS['reset']}")
+
+    print(f"  8. {COLORS['cyan']}Remove a configuration by index or provider:{COLORS['reset']}")
     print(f"     {COLORS['yellow']}ngpt --config --remove --config-index 1{COLORS['reset']}")
-    print(f"  8. {COLORS['cyan']}List available models for the current configuration:{COLORS['reset']}")
+    print(f"     {COLORS['yellow']}ngpt --config --remove --provider Gemini{COLORS['reset']}")
+
+    print(f"  9. {COLORS['cyan']}List available models for the current configuration:{COLORS['reset']}")
     print(f"     {COLORS['yellow']}ngpt --list-models{COLORS['reset']}")
 
 def check_config(config):
@@ -325,7 +505,7 @@ def check_config(config):
     
     return True
 
-def interactive_chat_session(client, web_search=False, no_stream=False, temperature=0.7, top_p=1.0, max_tokens=None, log_file=None, preprompt=None):
+def interactive_chat_session(client, web_search=False, no_stream=False, temperature=0.7, top_p=1.0, max_tokens=None, log_file=None, preprompt=None, prettify=False, renderer='auto'):
     """Run an interactive chat session with conversation history."""
     # Get terminal width for better formatting
     try:
@@ -467,6 +647,7 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
             
             # Skip empty messages but don't raise an error
             if not user_input.strip():
+                print(f"{COLORS['yellow']}Empty message skipped. Type 'exit' to quit.{COLORS['reset']}")
                 continue
             
             # Add user message to conversation
@@ -484,11 +665,19 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
             else:
                 print(f"\n{ngpt_header()}: {COLORS['reset']}", flush=True)
             
+            # If prettify is enabled, we need to disable streaming to collect the full response
+            should_stream = not no_stream and not prettify
+            
+            # If prettify is enabled with streaming, inform the user
+            if prettify and not no_stream:
+                print(f"\n{COLORS['yellow']}Note: Streaming disabled to enable markdown rendering.{COLORS['reset']}")
+                print(f"\n{ngpt_header()}: {COLORS['reset']}", flush=True)
+            
             # Get AI response with conversation history
             response = client.chat(
                 prompt=user_input,
                 messages=conversation,
-                stream=not no_stream,
+                stream=should_stream,
                 web_search=web_search,
                 temperature=temperature,
                 top_p=top_p,
@@ -500,9 +689,12 @@ def interactive_chat_session(client, web_search=False, no_stream=False, temperat
                 assistant_message = {"role": "assistant", "content": response}
                 conversation.append(assistant_message)
                 
-                # Print response if not streamed
-                if no_stream:
-                    print(response)
+                # Print response if not streamed (either due to no_stream or prettify)
+                if no_stream or prettify:
+                    if prettify:
+                        prettify_markdown(response, renderer)
+                    else:
+                        print(response)
                 
                 # Log assistant response if logging is enabled
                 if log_handle:
@@ -554,10 +746,12 @@ def main():
     config_group = parser.add_argument_group('Configuration Options')
     config_group.add_argument('--config', nargs='?', const=True, help='Path to a custom config file or, if no value provided, enter interactive configuration mode to create a new config')
     config_group.add_argument('--config-index', type=int, default=0, help='Index of the configuration to use or edit (default: 0)')
+    config_group.add_argument('--provider', help='Provider name to identify the configuration to use')
     config_group.add_argument('--remove', action='store_true', help='Remove the configuration at the specified index (requires --config and --config-index)')
     config_group.add_argument('--show-config', action='store_true', help='Show the current configuration(s) and exit')
     config_group.add_argument('--all', action='store_true', help='Show details for all configurations (requires --show-config)')
     config_group.add_argument('--list-models', action='store_true', help='List all available models for the current configuration and exit')
+    config_group.add_argument('--list-renderers', action='store_true', help='Show available markdown renderers for use with --prettify')
     
     # Global options
     global_group = parser.add_argument_group('Global Options')
@@ -578,6 +772,10 @@ def main():
                       help='Set filepath to log conversation to (For interactive modes)')
     global_group.add_argument('--preprompt', 
                       help='Set custom system prompt to control AI behavior')
+    global_group.add_argument('--prettify', action='store_const', const='auto',
+                      help='Render markdown responses and code with syntax highlighting and formatting')
+    global_group.add_argument('--renderer', choices=['auto', 'rich', 'glow'], default='auto',
+                      help='Select which markdown renderer to use with --prettify (auto, rich, or glow)')
     
     # Mode flags (mutually exclusive)
     mode_group = parser.add_argument_group('Modes (mutually exclusive)')
@@ -599,6 +797,15 @@ def main():
     # Validate --all usage
     if args.all and not args.show_config:
         parser.error("--all can only be used with --show-config")
+    
+    # Handle --renderers flag to show available markdown renderers
+    if args.list_renderers:
+        show_available_renderers()
+        return
+    
+    # Check for mutual exclusivity between --config-index and --provider
+    if args.config_index != 0 and args.provider:
+        parser.error("--config-index and --provider cannot be used together")
 
     # Handle interactive configuration mode
     if args.config is True:  # --config was used without a value
@@ -607,20 +814,46 @@ def main():
         # Handle configuration removal if --remove flag is present
         if args.remove:
             # Validate that config_index is explicitly provided
-            if '--config-index' not in sys.argv:
-                parser.error("--remove requires explicitly specifying --config-index")
+            if '--config-index' not in sys.argv and not args.provider:
+                parser.error("--remove requires explicitly specifying --config-index or --provider")
             
             # Show config details before asking for confirmation
             configs = load_configs(str(config_path))
             
+            # Determine the config index to remove
+            config_index = args.config_index
+            if args.provider:
+                # Find config index by provider name
+                matching_configs = [i for i, cfg in enumerate(configs) if cfg.get('provider', '').lower() == args.provider.lower()]
+                if not matching_configs:
+                    print(f"Error: No configuration found for provider '{args.provider}'")
+                    return
+                elif len(matching_configs) > 1:
+                    print(f"Multiple configurations found for provider '{args.provider}':")
+                    for i, idx in enumerate(matching_configs):
+                        print(f"  [{i}] Index {idx}: {configs[idx].get('model', 'Unknown model')}")
+                    
+                    try:
+                        choice = input("Choose a configuration to remove (or press Enter to cancel): ")
+                        if choice and choice.isdigit() and 0 <= int(choice) < len(matching_configs):
+                            config_index = matching_configs[int(choice)]
+                        else:
+                            print("Configuration removal cancelled.")
+                            return
+                    except (ValueError, IndexError, KeyboardInterrupt):
+                        print("\nConfiguration removal cancelled.")
+                        return
+                else:
+                    config_index = matching_configs[0]
+            
             # Check if index is valid
-            if args.config_index < 0 or args.config_index >= len(configs):
-                print(f"Error: Configuration index {args.config_index} is out of range. Valid range: 0-{len(configs)-1}")
+            if config_index < 0 or config_index >= len(configs):
+                print(f"Error: Configuration index {config_index} is out of range. Valid range: 0-{len(configs)-1}")
                 return
             
             # Show the configuration that will be removed
-            config = configs[args.config_index]
-            print(f"Configuration to remove (index {args.config_index}):")
+            config = configs[config_index]
+            print(f"Configuration to remove (index {config_index}):")
             print(f"  Provider: {config.get('provider', 'N/A')}")
             print(f"  Model: {config.get('model', 'N/A')}")
             print(f"  Base URL: {config.get('base_url', 'N/A')}")
@@ -631,7 +864,7 @@ def main():
                 print("\nAre you sure you want to remove this configuration? [y/N] ", end='')
                 response = input().lower()
                 if response in ('y', 'yes'):
-                    remove_config_entry(config_path, args.config_index)
+                    remove_config_entry(config_path, config_index)
                 else:
                     print("Configuration removal cancelled.")
             except KeyboardInterrupt:
@@ -643,20 +876,51 @@ def main():
         # If --config-index was not explicitly specified, create a new entry by passing None
         # This will cause add_config_entry to create a new entry at the end of the list
         # Otherwise, edit the existing config at the specified index
-        config_index = None if args.config_index == 0 and '--config-index' not in sys.argv else args.config_index
+        config_index = None
         
-        # Load existing configs to determine the new index if creating a new config
-        configs = load_configs(str(config_path))
-        if config_index is None:
-            print(f"Creating new configuration at index {len(configs)}")
-        else:
+        # Determine if we're editing an existing config or creating a new one
+        if args.provider:
+            # Find config by provider name
+            configs = load_configs(str(config_path))
+            matching_configs = [i for i, cfg in enumerate(configs) if cfg.get('provider', '').lower() == args.provider.lower()]
+            
+            if not matching_configs:
+                print(f"No configuration found for provider '{args.provider}'. Creating a new configuration.")
+            elif len(matching_configs) > 1:
+                print(f"Multiple configurations found for provider '{args.provider}':")
+                for i, idx in enumerate(matching_configs):
+                    print(f"  [{i}] Index {idx}: {configs[idx].get('model', 'Unknown model')}")
+                
+                try:
+                    choice = input("Choose a configuration to edit (or press Enter for the first one): ")
+                    if choice and choice.isdigit() and 0 <= int(choice) < len(matching_configs):
+                        config_index = matching_configs[int(choice)]
+                    else:
+                        config_index = matching_configs[0]
+                except (ValueError, IndexError, KeyboardInterrupt):
+                    config_index = matching_configs[0]
+            else:
+                config_index = matching_configs[0]
+                
             print(f"Editing existing configuration at index {config_index}")
+        elif args.config_index != 0 or '--config-index' in sys.argv:
+            # Check if the index is valid
+            configs = load_configs(str(config_path))
+            if args.config_index >= 0 and args.config_index < len(configs):
+                config_index = args.config_index
+                print(f"Editing existing configuration at index {config_index}")
+            else:
+                print(f"Configuration index {args.config_index} is out of range. Creating a new configuration.")
+        else:
+            # Creating a new config
+            configs = load_configs(str(config_path))
+            print(f"Creating new configuration at index {len(configs)}")
         
         add_config_entry(config_path, config_index)
         return
     
-    # Load configuration using the specified index (needed for active config display)
-    active_config = load_config(args.config, args.config_index)
+    # Load configuration using the specified index or provider (needed for active config display)
+    active_config = load_config(args.config, args.config_index, args.provider)
     
     # Command-line arguments override config settings for active config display
     # This part is kept to ensure the active config display reflects potential overrides,
@@ -675,31 +939,57 @@ def main():
         
         print(f"Configuration file: {config_path}")
         print(f"Total configurations: {len(configs)}")
-        print(f"Active configuration index: {args.config_index}")
+        
+        # Determine active configuration and display identifier
+        active_identifier = f"index {args.config_index}"
+        if args.provider:
+            active_identifier = f"provider '{args.provider}'"
+        print(f"Active configuration: {active_identifier}")
 
         if args.all:
             # Show details for all configurations
             print("\nAll configuration details:")
             for i, cfg in enumerate(configs):
-                active_str = '(Active)' if i == args.config_index else ''
-                print(f"\n--- Configuration Index {i} {active_str} ---")
+                provider = cfg.get('provider', 'N/A')
+                active_str = '(Active)' if (
+                    (args.provider and provider.lower() == args.provider.lower()) or 
+                    (not args.provider and i == args.config_index)
+                ) else ''
+                print(f"\n--- Configuration Index {i} / Provider: {COLORS['green']}{provider}{COLORS['reset']} {active_str} ---")
                 print(f"  API Key: {'[Set]' if cfg.get('api_key') else '[Not Set]'}")
                 print(f"  Base URL: {cfg.get('base_url', 'N/A')}")
-                print(f"  Provider: {cfg.get('provider', 'N/A')}")
                 print(f"  Model: {cfg.get('model', 'N/A')}")
         else:
             # Show active config details and summary list
             print("\nActive configuration details:")
+            print(f"  Provider: {COLORS['green']}{active_config.get('provider', 'N/A')}{COLORS['reset']}")
             print(f"  API Key: {'[Set]' if active_config.get('api_key') else '[Not Set]'}")
             print(f"  Base URL: {active_config.get('base_url', 'N/A')}")
-            print(f"  Provider: {active_config.get('provider', 'N/A')}")
             print(f"  Model: {active_config.get('model', 'N/A')}")
             
             if len(configs) > 1:
                 print("\nAvailable configurations:")
+                # Check for duplicate provider names for warning
+                provider_counts = {}
+                for cfg in configs:
+                    provider = cfg.get('provider', 'N/A').lower()
+                    provider_counts[provider] = provider_counts.get(provider, 0) + 1
+                
                 for i, cfg in enumerate(configs):
-                    active_marker = "*" if i == args.config_index else " "
-                    print(f"[{i}]{active_marker} {cfg.get('provider', 'N/A')} - {cfg.get('model', 'N/A')} ({'[API Key Set]' if cfg.get('api_key') else '[API Key Not Set]'})")
+                    provider = cfg.get('provider', 'N/A')
+                    provider_display = provider
+                    # Add warning for duplicate providers
+                    if provider_counts.get(provider.lower(), 0) > 1:
+                        provider_display = f"{provider} {COLORS['yellow']}(duplicate){COLORS['reset']}"
+                    
+                    active_marker = "*" if (
+                        (args.provider and provider.lower() == args.provider.lower()) or 
+                        (not args.provider and i == args.config_index)
+                    ) else " "
+                    print(f"[{i}]{active_marker} {COLORS['green']}{provider_display}{COLORS['reset']} - {cfg.get('model', 'N/A')} ({'[API Key Set]' if cfg.get('api_key') else '[API Key Not Set]'})")
+                
+                # Show instruction for using --provider
+                print(f"\nTip: Use {COLORS['yellow']}--provider NAME{COLORS['reset']} to select a configuration by provider name.")
         
         return
     
@@ -712,6 +1002,17 @@ def main():
     if not args.show_config and not args.list_models and not check_config(active_config):
         return
     
+    # Check if --prettify is used but no markdown renderer is available
+    # This will warn the user immediately if they request prettify but don't have the tools
+    has_renderer = True
+    if args.prettify:
+        has_renderer = warn_if_no_markdown_renderer(args.renderer)
+        if not has_renderer:
+            # Set a flag to disable prettify since we already warned the user
+            print(f"{COLORS['yellow']}Continuing without markdown rendering.{COLORS['reset']}")
+            show_available_renderers()
+            args.prettify = False
+        
     # Initialize client using the potentially overridden active_config
     client = NGPTClient(**active_config)
     
@@ -738,7 +1039,7 @@ def main():
             # Interactive chat mode
             interactive_chat_session(client, web_search=args.web_search, no_stream=args.no_stream,
                                    temperature=args.temperature, top_p=args.top_p, 
-                                   max_tokens=args.max_tokens, log_file=args.log, preprompt=args.preprompt)
+                                   max_tokens=args.max_tokens, log_file=args.log, preprompt=args.preprompt, prettify=args.prettify, renderer=args.renderer)
         elif args.shell:
             if args.prompt is None:
                 try:
@@ -792,7 +1093,13 @@ def main():
                                               temperature=args.temperature, top_p=args.top_p,
                                               max_tokens=args.max_tokens)
             if generated_code:
-                print(f"\nGenerated code:\n{generated_code}")
+                if args.prettify:
+                    # Format code as markdown with proper syntax highlighting
+                    markdown_code = f"```{args.language}\n{generated_code}\n```"
+                    print("\nGenerated code:")
+                    prettify_markdown(markdown_code, args.renderer)
+                else:
+                    print(f"\nGenerated code:\n{generated_code}")
             
         elif args.text:
             if args.prompt is not None:
@@ -910,12 +1217,24 @@ def main():
                     {"role": "system", "content": args.preprompt},
                     {"role": "user", "content": prompt}
                 ]
+            
+            # If prettify is enabled, we need to disable streaming to collect the full response
+            should_stream = not args.no_stream and not args.prettify
+            
+            # If prettify is enabled with streaming, inform the user
+            if args.prettify and not args.no_stream:
+                print(f"{COLORS['yellow']}Note: Streaming disabled to enable markdown rendering.{COLORS['reset']}")
                 
-            response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search,
+            response = client.chat(prompt, stream=should_stream, web_search=args.web_search,
                                temperature=args.temperature, top_p=args.top_p,
                                max_tokens=args.max_tokens, messages=messages)
-            if args.no_stream and response:
-                print(response)
+            
+            # Handle non-stream response (either because no_stream was set or prettify forced it)
+            if (args.no_stream or args.prettify) and response:
+                if args.prettify:
+                    prettify_markdown(response, args.renderer)
+                else:
+                    print(response)
             
         else:
             # Default to chat mode
@@ -936,12 +1255,24 @@ def main():
                     {"role": "system", "content": args.preprompt},
                     {"role": "user", "content": prompt}
                 ]
+            
+            # If prettify is enabled, we need to disable streaming to collect the full response
+            should_stream = not args.no_stream and not args.prettify
+            
+            # If prettify is enabled with streaming, inform the user
+            if args.prettify and not args.no_stream:
+                print(f"{COLORS['yellow']}Note: Streaming disabled to enable markdown rendering.{COLORS['reset']}")
                 
-            response = client.chat(prompt, stream=not args.no_stream, web_search=args.web_search,
+            response = client.chat(prompt, stream=should_stream, web_search=args.web_search,
                                temperature=args.temperature, top_p=args.top_p,
                                max_tokens=args.max_tokens, messages=messages)
-            if args.no_stream and response:
-                print(response)
+            
+            # Handle non-stream response (either because no_stream was set or prettify forced it)
+            if (args.no_stream or args.prettify) and response:
+                if args.prettify:
+                    prettify_markdown(response, args.renderer)
+                else:
+                    print(response)
     
     except KeyboardInterrupt:
         print("\nOperation cancelled by user. Exiting gracefully.")

ngpt/config.py

@@ -75,9 +75,29 @@ def add_config_entry(config_path: Path, config_index: Optional[int] = None) -> N
         if user_input:
             entry["base_url"] = user_input
         
-        user_input = input(f"Provider [{entry['provider']}]: ")
-        if user_input:
-            entry["provider"] = user_input
+        # For provider, check for uniqueness when creating new config
+        provider_unique = False
+        original_provider = entry['provider']
+        while not provider_unique:
+            user_input = input(f"Provider [{entry['provider']}]: ")
+            if user_input:
+                provider = user_input
+            else:
+                provider = entry['provider']
+            
+            # When creating new config or changing provider, check uniqueness
+            if is_existing_config and provider.lower() == original_provider.lower():
+                # No change in provider name, so keep it
+                provider_unique = True
+            elif is_provider_unique(configs, provider, config_index if is_existing_config else None):
+                provider_unique = True
+            else:
+                print(f"Error: Provider '{provider}' already exists. Please choose a unique provider name.")
+                # If it's the existing provider, allow keeping it (for existing configs)
+                if is_existing_config and provider.lower() == original_provider.lower():
+                    provider_unique = True
+        
+        entry["provider"] = provider
         
         user_input = input(f"Model [{entry['model']}]: ")
         if user_input:
@@ -127,13 +147,46 @@ def load_configs(custom_path: Optional[str] = None) -> List[Dict[str, Any]]:
     
     return configs
 
-def load_config(custom_path: Optional[str] = None, config_index: int = 0) -> Dict[str, Any]:
+def load_config(custom_path: Optional[str] = None, config_index: int = 0, provider: Optional[str] = None) -> Dict[str, Any]:
     """
-    Load a specific configuration by index and apply environment variables.
+    Load a specific configuration by index or provider name and apply environment variables.
     Environment variables take precedence over the config file.
+    
+    Args:
+        custom_path: Optional path to a custom config file
+        config_index: Index of the configuration to use (default: 0)
+        provider: Provider name to identify the configuration
+        
+    Returns:
+        The selected configuration with environment variables applied
     """
     configs = load_configs(custom_path)
     
+    # If provider is specified, try to find a matching config
+    if provider:
+        matching_configs = [i for i, cfg in enumerate(configs) if cfg.get('provider', '').lower() == provider.lower()]
+        
+        if not matching_configs:
+            print(f"Warning: No configuration found for provider '{provider}'. Using default configuration.")
+            config_index = 0
+        elif len(matching_configs) > 1:
+            print(f"Warning: Multiple configurations found for provider '{provider}'.")
+            for i, idx in enumerate(matching_configs):
+                print(f"  [{i}] Index {idx}: {configs[idx].get('model', 'Unknown model')}")
+            
+            try:
+                choice = input("Choose a configuration (or press Enter for the first one): ")
+                if choice and choice.isdigit() and 0 <= int(choice) < len(matching_configs):
+                    config_index = matching_configs[int(choice)]
+                else:
+                    config_index = matching_configs[0]
+                    print(f"Using first matching configuration (index {config_index}).")
+            except (ValueError, IndexError, KeyboardInterrupt):
+                config_index = matching_configs[0]
+                print(f"Using first matching configuration (index {config_index}).")
+        else:
+            config_index = matching_configs[0]
+    
     # If config_index is out of range, use the first config
     if config_index < 0 or config_index >= len(configs):
         if len(configs) > 0:
@@ -157,7 +210,7 @@ def load_config(custom_path: Optional[str] = None, config_index: int = 0) -> Dic
         if env_var in os.environ and os.environ[env_var]:
             config[config_key] = os.environ[env_var]
     
-    return config 
+    return config
 
 def remove_config_entry(config_path: Path, config_index: int) -> bool:
     """
@@ -182,4 +235,24 @@ def remove_config_entry(config_path: Path, config_index: int) -> bool:
         return True
     except Exception as e:
         print(f"Error saving configuration: {e}")
-        return False 
+        return False
+
+def is_provider_unique(configs: List[Dict[str, Any]], provider: str, exclude_index: Optional[int] = None) -> bool:
+    """
+    Check if a provider name is unique among configurations.
+    
+    Args:
+        configs: List of configuration dictionaries
+        provider: Provider name to check
+        exclude_index: Optional index to exclude from the check (for updating existing config)
+        
+    Returns:
+        True if the provider name is unique, False otherwise
+    """
+    provider = provider.lower()
+    for i, cfg in enumerate(configs):
+        if i == exclude_index:
+            continue
+        if cfg.get('provider', '').lower() == provider:
+            return False
+    return True 

{ngpt-2.3.4.dist-info → ngpt-2.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ngpt
-Version: 2.3.4
+Version: 2.5.0
 Summary: A lightweight Python CLI and library for interacting with OpenAI-compatible APIs, supporting both official and self-hosted LLM endpoints.
 Project-URL: Homepage, https://github.com/nazdridoy/ngpt
 Project-URL: Repository, https://github.com/nazdridoy/ngpt
@@ -30,6 +30,9 @@ Classifier: Topic :: Utilities
 Requires-Python: >=3.8
 Requires-Dist: prompt-toolkit>=3.0.0
 Requires-Dist: requests>=2.31.0
+Requires-Dist: rich>=14.0.0
+Provides-Extra: prettify
+Requires-Dist: rich>=10.0.0; extra == 'prettify'
 Description-Content-Type: text/markdown
 
 # nGPT
@@ -76,9 +79,18 @@ ngpt -n "Tell me about quantum computing"
 # Generate code
 ngpt --code "function to calculate the Fibonacci sequence"
 
+# Generate code with syntax highlighting
+ngpt --code --prettify "function to calculate the Fibonacci sequence"
+
 # Generate and execute shell commands
 ngpt --shell "list all files in the current directory"
 
+# Display markdown responses with beautiful formatting
+ngpt --prettify "Explain markdown syntax with examples"
+
+# Use a specific markdown renderer
+ngpt --prettify --renderer=rich "Create a markdown table"
+
 # Use multiline editor for complex prompts
 ngpt --text
 
@@ -99,6 +111,7 @@ For more examples and detailed usage, visit the [CLI Usage Guide](https://nazdri
 - 💬 **Interactive Chat**: Continuous conversation with memory in modern UI
 - 📊 **Streaming Responses**: Real-time output for better user experience
 - 🔍 **Web Search**: Integrated with compatible API endpoints
+- 🎨 **Markdown Rendering**: Beautiful formatting of markdown and code with syntax highlighting
 - ⚙️ **Multiple Configurations**: Cross-platform config system supporting different profiles
 - 💻 **Shell Command Generation**: OS-aware command execution
 - 🧩 **Clean Code Generation**: Output code without markdown or explanations
@@ -267,8 +280,12 @@ You can configure the client using the following options:
 | `--max_tokens` | Set maximum response length in tokens |
 | `--preprompt` | Set custom system prompt to control AI behavior |
 | `--log` | Set filepath to log conversation to (for interactive modes) |
+| `--prettify` | Render markdown responses and code with syntax highlighting |
+| `--renderer` | Select which markdown renderer to use with --prettify (auto, rich, or glow) |
+| `--list-renderers` | Show available markdown renderers for use with --prettify |
 | `--config` | Path to a custom configuration file or, when used without a value, enters interactive configuration mode |
 | `--config-index` | Index of the configuration to use (default: 0) |
+| `--provider` | Provider name to identify the configuration to use (alternative to --config-index) |
 | `--remove` | Remove the configuration at the specified index (requires --config and --config-index) |
 | `--show-config` | Show configuration details and exit |
 | `--all` | Used with `--show-config` to display all configurations |
@@ -291,8 +308,17 @@ ngpt --config
 # Edit an existing configuration at index 1
 ngpt --config --config-index 1
 
+# Edit an existing configuration by provider name
+ngpt --config --provider Gemini
+
 # Remove a configuration at index 2
 ngpt --config --remove --config-index 2
+
+# Remove a configuration by provider name
+ngpt --config --remove --provider Gemini
+
+# Use a specific configuration by provider name
+ngpt --provider OpenAI "Tell me about quantum computing"
 ```
 
 In interactive mode:

ngpt-2.5.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
+ngpt/cli.py,sha256=VE-Zp7mx-GC6p0o9CcKAnAJCDbLjwiV_UKxwikceDaM,59632
+ngpt/client.py,sha256=lJfyLONeBU7YqMVJJys6zPay7gcJTq108_rJ4wvOtok,13067
+ngpt/config.py,sha256=WYOk_b1eiYjo6hpV3pfXr2RjqhOnmKqwZwKid1T41I4,10363
+ngpt-2.5.0.dist-info/METADATA,sha256=Guk41uYFdF-_hWqi6QgUHhWXXuhgU2CMai0GvIOnkHA,14685
+ngpt-2.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ngpt-2.5.0.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
+ngpt-2.5.0.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
+ngpt-2.5.0.dist-info/RECORD,,

ngpt-2.3.4.dist-info/RECORD

@@ -1,9 +0,0 @@
-ngpt/__init__.py,sha256=ehInP9w0MZlS1vZ1g6Cm4YE1ftmgF72CnEddQ3Le9n4,368
-ngpt/cli.py,sha256=pmgUK-vrMAAsALKnTxVAcoFSGZ4DM89d43bXKuiLbN0,43532
-ngpt/client.py,sha256=lJfyLONeBU7YqMVJJys6zPay7gcJTq108_rJ4wvOtok,13067
-ngpt/config.py,sha256=BF0G3QeiPma8l7EQyc37bR7LWZog7FHJQNe7uj9cr4w,6896
-ngpt-2.3.4.dist-info/METADATA,sha256=H78mhW758iTL9wnB2lgMGYUzivmHzDQU7IdEXqC4X4Y,13535
-ngpt-2.3.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ngpt-2.3.4.dist-info/entry_points.txt,sha256=1cnAMujyy34DlOahrJg19lePSnb08bLbkUs_kVerqdk,39
-ngpt-2.3.4.dist-info/licenses/LICENSE,sha256=mQkpWoADxbHqE0HRefYLJdm7OpdrXBr3vNv5bZ8w72M,1065
-ngpt-2.3.4.dist-info/RECORD,,