chat-console 0.4.2__py3-none-any.whl → 0.4.6__py3-none-any.whl

app/console_chat.py

@@ -0,0 +1,816 @@
+#!/usr/bin/env python3
+"""
+Pure Console Chat CLI - No Textual Dependencies
+A true terminal interface following Dieter Rams principles
+"""
+
+import os
+import sys
+import asyncio
+import argparse
+import signal
+import threading
+import time
+from datetime import datetime
+from typing import List, Optional, Dict, Any
+import shutil
+
+from .models import Message, Conversation
+from .database import ChatDatabase
+from .config import CONFIG, save_config
+from .utils import resolve_model_id, generate_conversation_title
+from .console_utils import console_streaming_response, apply_style_prefix
+from .api.base import BaseModelClient
+
+class ConsoleUI:
+    """Pure console UI following Rams design principles"""
+    
+    def __init__(self):
+        self.width = min(shutil.get_terminal_size().columns, 120)
+        self.height = shutil.get_terminal_size().lines
+        self.db = ChatDatabase()
+        self.current_conversation: Optional[Conversation] = None
+        self.messages: List[Message] = []
+        self.selected_model = resolve_model_id(CONFIG["default_model"])
+        self.selected_style = CONFIG["default_style"]
+        self.running = True
+        self.generating = False
+        self.input_mode = "text"  # "text" or "menu"
+        
+        # Suppress verbose logging for console mode
+        self._setup_console_logging()
+    
+    def _setup_console_logging(self):
+        """Setup logging to minimize disruption to console UI"""
+        import logging
+        
+        # Set root logger to ERROR to suppress all INFO messages
+        logging.getLogger().setLevel(logging.ERROR)
+        
+        # Suppress all app module logging
+        logging.getLogger('app').setLevel(logging.ERROR)
+        logging.getLogger('app.api').setLevel(logging.ERROR)
+        logging.getLogger('app.api.base').setLevel(logging.ERROR)
+        logging.getLogger('app.api.ollama').setLevel(logging.ERROR)
+        logging.getLogger('app.utils').setLevel(logging.ERROR)
+        logging.getLogger('app.console_utils').setLevel(logging.ERROR)
+        
+        # Suppress third-party library logging
+        logging.getLogger('aiohttp').setLevel(logging.ERROR)
+        logging.getLogger('urllib3').setLevel(logging.ERROR)
+        logging.getLogger('httpx').setLevel(logging.ERROR)
+        logging.getLogger('asyncio').setLevel(logging.ERROR)
+        logging.getLogger('root').setLevel(logging.ERROR)
+        
+        # Completely disable all handlers to prevent any output
+        logging.basicConfig(
+            level=logging.CRITICAL,  # Only show CRITICAL messages
+            format='',  # Empty format
+            handlers=[logging.NullHandler()]  # Null handler suppresses all output
+        )
+        
+        # Clear any existing handlers
+        for handler in logging.root.handlers[:]:
+            logging.root.removeHandler(handler)
+        
+        # Add only NullHandler
+        logging.root.addHandler(logging.NullHandler())
+        
+        # Redirect stdout/stderr for subprocess calls (if any)
+        self._dev_null = open(os.devnull, 'w')
+        
+    def _suppress_output(self):
+        """Context manager to suppress all output during sensitive operations"""
+        import sys
+        import contextlib
+        
+        @contextlib.contextmanager
+        def suppress():
+            with open(os.devnull, "w") as devnull:
+                old_stdout = sys.stdout
+                old_stderr = sys.stderr
+                try:
+                    sys.stdout = devnull
+                    sys.stderr = devnull
+                    yield
+                finally:
+                    sys.stdout = old_stdout
+                    sys.stderr = old_stderr
+        
+        return suppress()
+        
+    def clear_screen(self):
+        """Clear the terminal screen"""
+        os.system('cls' if os.name == 'nt' else 'clear')
+    
+    def get_border_chars(self):
+        """Get clean ASCII border characters"""
+        return {
+            'horizontal': '─',
+            'vertical': '│',
+            'top_left': '┌',
+            'top_right': '┐',
+            'bottom_left': '└',
+            'bottom_right': '┘',
+            'tee_down': '┬',
+            'tee_up': '┴',
+            'tee_right': '├',
+            'tee_left': '┤'
+        }
+    
+    def draw_border_line(self, width: int, position: str = 'top') -> str:
+        """Draw a clean border line"""
+        chars = self.get_border_chars()
+        
+        if position == 'top':
+            return chars['top_left'] + chars['horizontal'] * (width - 2) + chars['top_right']
+        elif position == 'bottom':
+            return chars['bottom_left'] + chars['horizontal'] * (width - 2) + chars['bottom_right']
+        elif position == 'middle':
+            return chars['tee_right'] + chars['horizontal'] * (width - 2) + chars['tee_left']
+        else:
+            return chars['horizontal'] * width
+    
+    def draw_header(self) -> List[str]:
+        """Draw the application header"""
+        from . import __version__
+        chars = self.get_border_chars()
+        
+        lines = []
+        
+        # Top border with title and model info
+        title = f" Chat Console v{__version__} "
+        model_info = f" Model: {self.selected_model} "
+        
+        # Calculate spacing
+        used_space = len(title) + len(model_info)
+        remaining = self.width - used_space - 2
+        spacing = chars['horizontal'] * max(0, remaining)
+        
+        header_line = chars['top_left'] + title + spacing + model_info + chars['top_right']
+        lines.append(header_line)
+        
+        # Conversation title
+        conv_title = self.current_conversation.title if self.current_conversation else "New Conversation"
+        title_line = chars['vertical'] + f" {conv_title} ".ljust(self.width - 2) + chars['vertical']
+        lines.append(title_line)
+        
+        # Separator
+        lines.append(self.draw_border_line(self.width, 'middle'))
+        
+        return lines
+    
+    def draw_footer(self) -> List[str]:
+        """Draw the footer with controls"""
+        chars = self.get_border_chars()
+        
+        controls = "[Tab] Menu Mode  [q] Quit  [n] New  [h] History  [s] Settings"
+        footer_line = chars['vertical'] + f" {controls} ".ljust(self.width - 2) + chars['vertical']
+        
+        return [
+            self.draw_border_line(self.width, 'middle'),
+            footer_line,
+            self.draw_border_line(self.width, 'bottom')
+        ]
+    
+    def format_message(self, message: Message) -> List[str]:
+        """Format a message for console display"""
+        timestamp = datetime.now().strftime("%H:%M")
+        chars = self.get_border_chars()
+        
+        # Calculate available width for content
+        content_width = self.width - 10  # Account for borders and timestamp
+        
+        # Word wrap content
+        words = message.content.split()
+        lines = []
+        current_line = ""
+        
+        for word in words:
+            if len(current_line) + len(word) + 1 <= content_width:
+                if current_line:
+                    current_line += " "
+                current_line += word
+            else:
+                if current_line:
+                    lines.append(current_line)
+                current_line = word
+        
+        if current_line:
+            lines.append(current_line)
+        
+        # Format lines with proper spacing
+        formatted_lines = []
+        for i, line in enumerate(lines):
+            if i == 0:
+                # First line with timestamp
+                prefix = f"  {timestamp}  " if message.role == "user" else f"  {timestamp}  "
+                formatted_line = chars['vertical'] + prefix + line.ljust(content_width) + chars['vertical']
+            else:
+                # Continuation lines
+                prefix = "        "  # Align with content
+                formatted_line = chars['vertical'] + prefix + line.ljust(content_width) + chars['vertical']
+            formatted_lines.append(formatted_line)
+        
+        # Add empty line for spacing
+        empty_line = chars['vertical'] + " " * (self.width - 2) + chars['vertical']
+        formatted_lines.append(empty_line)
+        
+        return formatted_lines
+    
+    def draw_messages(self) -> List[str]:
+        """Draw all messages in the conversation"""
+        lines = []
+        chars = self.get_border_chars()
+        
+        if not self.messages:
+            # Empty state
+            empty_line = chars['vertical'] + " " * (self.width - 2) + chars['vertical']
+            lines.extend([empty_line] * 3)
+            center_text = "Start a conversation by typing a message below"
+            centered_line = chars['vertical'] + center_text.center(self.width - 2) + chars['vertical']
+            lines.append(centered_line)
+            lines.extend([empty_line] * 3)
+        else:
+            # Display messages
+            for message in self.messages[-10:]:  # Show last 10 messages
+                lines.extend(self.format_message(message))
+        
+        return lines
+    
+    def draw_input_area(self, current_input: str = "", prompt: str = "Type your message") -> List[str]:
+        """Draw the input area with mode indicator"""
+        chars = self.get_border_chars()
+        lines = []
+        
+        # Input prompt with mode indicator
+        mode_indicator = "📝" if self.input_mode == "text" else "⚡"
+        mode_text = "TEXT" if self.input_mode == "text" else "MENU"
+        prompt_with_mode = f"{mode_indicator} {prompt} ({mode_text} mode - Tab to switch)"
+        prompt_line = chars['vertical'] + f" {prompt_with_mode}: ".ljust(self.width - 2) + chars['vertical']
+        lines.append(prompt_line)
+        
+        # Input field
+        if self.input_mode == "text":
+            input_content = current_input
+            if len(input_content) > self.width - 6:
+                input_content = input_content[-(self.width - 9):] + "..."
+            input_line = chars['vertical'] + f" > {input_content}".ljust(self.width - 2) + chars['vertical']
+        else:
+            # Menu mode - show available hotkeys
+            menu_help = "n)ew  h)istory  s)ettings  q)uit"
+            input_line = chars['vertical'] + f" {menu_help}".ljust(self.width - 2) + chars['vertical']
+        
+        lines.append(input_line)
+        
+        # Show generating indicator if needed
+        if self.generating:
+            status_line = chars['vertical'] + " ● Generating response...".ljust(self.width - 2) + chars['vertical']
+            lines.append(status_line)
+        
+        return lines
+    
+    def draw_screen(self, current_input: str = "", input_prompt: str = "Type your message"):
+        """Draw the complete screen"""
+        self.clear_screen()
+        
+        # Calculate layout
+        header_lines = self.draw_header()
+        footer_lines = self.draw_footer()
+        input_lines = self.draw_input_area(current_input, input_prompt)
+        
+        # Calculate available space for messages
+        used_lines = len(header_lines) + len(footer_lines) + len(input_lines)
+        available_lines = self.height - used_lines - 2
+        
+        # Draw header
+        for line in header_lines:
+            print(line)
+        
+        # Draw messages
+        message_lines = self.draw_messages()
+        chars = self.get_border_chars()
+        
+        # Pad or truncate message area
+        if len(message_lines) < available_lines:
+            # Pad with empty lines
+            empty_line = chars['vertical'] + " " * (self.width - 2) + chars['vertical']
+            message_lines.extend([empty_line] * (available_lines - len(message_lines)))
+        else:
+            # Truncate to fit
+            message_lines = message_lines[-available_lines:]
+        
+        for line in message_lines:
+            print(line)
+        
+        # Draw input area
+        for line in input_lines:
+            print(line)
+        
+        # Draw footer
+        for line in footer_lines:
+            print(line)
+        
+        # Position cursor
+        print("\033[A" * (len(footer_lines) + len(input_lines) - 1), end="")
+        print(f"\033[{len(current_input) + 4}C", end="")
+        sys.stdout.flush()
+    
+    def get_input(self, prompt: str = "Type your message") -> str:
+        """Enhanced input with tab navigation and hotkey support"""
+        current_input = ""
+        
+        while True:
+            self.draw_screen(current_input, prompt)
+            
+            # Get single character
+            if os.name == 'nt':
+                import msvcrt
+                char = msvcrt.getch().decode('utf-8', errors='ignore')
+            else:
+                import termios, tty
+                fd = sys.stdin.fileno()
+                old_settings = termios.tcgetattr(fd)
+                try:
+                    tty.setraw(sys.stdin.fileno())
+                    char = sys.stdin.read(1)
+                finally:
+                    termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+            
+            # Handle special keys first
+            if char == '\t':
+                # Tab - switch between text and menu mode
+                self.input_mode = "menu" if self.input_mode == "text" else "text"
+                continue
+            elif char == '\r' or char == '\n':
+                # Enter
+                if self.input_mode == "text":
+                    # Submit text input
+                    if current_input.strip():
+                        return current_input.strip()
+                    # If empty input in text mode, switch to menu mode
+                    self.input_mode = "menu"
+                    continue
+                else:
+                    # In menu mode, Enter does nothing
+                    continue
+            elif char == '\x03':
+                # Ctrl+C
+                if self.generating:
+                    self.generating = False
+                    return ""
+                else:
+                    raise KeyboardInterrupt
+            
+            # Mode-specific handling
+            if self.input_mode == "text":
+                # Text input mode
+                if char == '\x7f' or char == '\x08':
+                    # Backspace
+                    current_input = current_input[:-1]
+                elif ord(char) >= 32:
+                    # Printable character
+                    current_input += char
+            else:
+                # Menu mode - handle hotkeys
+                if char.lower() == 'q':
+                    return "##QUIT##"
+                elif char.lower() == 'n':
+                    return "##NEW##"
+                elif char.lower() == 'h':
+                    return "##HISTORY##"
+                elif char.lower() == 's':
+                    return "##SETTINGS##"
+                elif char == '\x1b':  # Escape - back to text mode
+                    self.input_mode = "text"
+                    continue
+    
+    async def create_new_conversation(self):
+        """Create a new conversation"""
+        title = "New Conversation"
+        conversation_id = self.db.create_conversation(title, self.selected_model, self.selected_style)
+        conversation_data = self.db.get_conversation(conversation_id)
+        self.current_conversation = Conversation.from_dict(conversation_data)
+        self.messages = []
+        
+    async def add_message(self, role: str, content: str):
+        """Add a message to the current conversation"""
+        message = Message(role=role, content=content)
+        self.messages.append(message)
+        
+        if self.current_conversation:
+            self.db.add_message(self.current_conversation.id, role, content)
+    
+    async def _generate_title_background(self, first_message: str):
+        """Generate conversation title in background after first user message"""
+        if not CONFIG.get("generate_dynamic_titles", True):
+            return
+            
+        try:
+            # Get client for title generation
+            with self._suppress_output():
+                client = await BaseModelClient.get_client_for_model(self.selected_model)
+            
+            # Generate title
+            new_title = await generate_conversation_title(first_message, self.selected_model, client)
+            
+            # Update conversation title in database and UI
+            if self.current_conversation and new_title and new_title != "New Conversation":
+                self.db.update_conversation_title(self.current_conversation.id, new_title)
+                self.current_conversation.title = new_title
+                
+        except Exception as e:
+            # Silently fail - title generation is not critical
+            pass
+    
+    async def generate_response(self, user_message: str):
+        """Generate AI response"""
+        self.generating = True
+        
+        try:
+            # Add user message
+            await self.add_message("user", user_message)
+            
+            # Generate title for first user message if this is a new conversation
+            if (self.current_conversation and 
+                self.current_conversation.title == "New Conversation" and 
+                len([msg for msg in self.messages if msg.role == "user"]) == 1):
+                # Generate title in background (non-blocking)
+                import asyncio
+                asyncio.create_task(self._generate_title_background(user_message))
+            
+            # Prepare messages for API
+            api_messages = []
+            for msg in self.messages:
+                api_messages.append({
+                    "role": msg.role,
+                    "content": msg.content
+                })
+            
+            # Get client with appropriate output suppression
+            model_info = CONFIG["available_models"].get(self.selected_model, {})
+            is_ollama = (model_info.get("provider") == "ollama" or 
+                        "ollama" in self.selected_model.lower() or 
+                        self.selected_model in ["gemma:2b", "gemma:7b", "llama3:8b", "mistral:7b"])
+            
+            if is_ollama:
+                with self._suppress_output():
+                    client = await BaseModelClient.get_client_for_model(self.selected_model)
+            else:
+                client = await BaseModelClient.get_client_for_model(self.selected_model)
+            
+            # Add assistant message
+            assistant_message = Message(role="assistant", content="")
+            self.messages.append(assistant_message)
+            
+            # Stream response
+            full_response = ""
+            
+            def update_callback(content: str):
+                nonlocal full_response
+                full_response = content
+                assistant_message.content = content
+                # Redraw screen periodically
+                self.draw_screen("", "Generating response")
+            
+            # Apply style to messages
+            styled_messages = apply_style_prefix(api_messages, self.selected_style)
+            
+            # Generate streaming response with output suppression
+            with self._suppress_output():
+                async for chunk in console_streaming_response(
+                    styled_messages, self.selected_model, self.selected_style, client, update_callback
+                ):
+                    if not self.generating:
+                        break
+                    if chunk:
+                        full_response += chunk
+            
+            # Update final message content
+            assistant_message.content = full_response
+            
+            # Save final response
+            if self.current_conversation and full_response:
+                self.db.add_message(self.current_conversation.id, "assistant", full_response)
+                
+        except Exception as e:
+            # Handle errors
+            error_msg = f"Error: {str(e)}"
+            if self.messages and self.messages[-1].role == "assistant":
+                self.messages[-1].content = error_msg
+            else:
+                await self.add_message("assistant", error_msg)
+        finally:
+            self.generating = False
+    
+    def show_history(self):
+        """Show conversation history"""
+        conversations = self.db.get_all_conversations(limit=20)
+        if not conversations:
+            input("No conversations found. Press Enter to continue...")
+            return
+        
+        self.clear_screen()
+        print("=" * self.width)
+        print("CONVERSATION HISTORY".center(self.width))
+        print("=" * self.width)
+        
+        for i, conv in enumerate(conversations):
+            print(f"{i+1:2d}. {conv['title'][:60]} ({conv['model']})")
+        
+        print("\nEnter conversation number to load (or press Enter to cancel):")
+        
+        try:
+            choice = input("> ").strip()
+            if choice and choice.isdigit():
+                idx = int(choice) - 1
+                if 0 <= idx < len(conversations):
+                    # Load conversation
+                    conv_data = self.db.get_conversation(conversations[idx]['id'])
+                    self.current_conversation = Conversation.from_dict(conv_data)
+                    self.messages = [Message(**msg) for msg in self.current_conversation.messages]
+        except (ValueError, KeyboardInterrupt):
+            pass
+    
+    async def show_settings(self):
+        """Show enhanced settings menu with dynamic model detection"""
+        while True:
+            self.clear_screen()
+            print("=" * self.width)
+            print("SETTINGS".center(self.width))
+            print("=" * self.width)
+            
+            print(f"Current Model: {self.selected_model}")
+            print(f"Current Style: {self.selected_style}")
+            print()
+            
+            print("What would you like to change?")
+            print("1. Select Model")
+            print("2. Response Style")
+            print("3. Detect Ollama Models")
+            print("0. Back to Chat")
+            
+            try:
+                choice = input("\n> ").strip()
+                
+                if choice == "1":
+                    await self._select_model()
+                elif choice == "2":
+                    self._select_style()
+                elif choice == "3":
+                    await self._detect_ollama_models()
+                elif choice == "0" or choice == "":
+                    break
+                    
+            except (ValueError, KeyboardInterrupt):
+                break
+    
+    async def _select_model(self):
+        """Enhanced model selection with all providers"""
+        self.clear_screen()
+        print("=" * self.width)
+        print("MODEL SELECTION".center(self.width))
+        print("=" * self.width)
+        
+        # Group models by provider
+        providers = {}
+        for model_id, model_info in CONFIG["available_models"].items():
+            provider = model_info["provider"]
+            if provider not in providers:
+                providers[provider] = []
+            providers[provider].append((model_id, model_info))
+        
+        # Add dynamically detected Ollama models
+        try:
+            with self._suppress_output():
+                from .api.ollama import OllamaClient
+                client = await OllamaClient.create()
+                local_models = await client.get_available_models()
+                
+                if local_models:
+                    if "ollama" not in providers:
+                        providers["ollama"] = []
+                    
+                    for model in local_models:
+                        model_id = model.get("id", "unknown")
+                        # Only add if not already in config
+                        if model_id not in CONFIG["available_models"]:
+                            providers["ollama"].append((model_id, {
+                                "provider": "ollama",
+                                "display_name": model_id,
+                                "max_tokens": 4096
+                            }))
+        except Exception as e:
+            pass  # Ollama not available
+        
+        # Display models by provider
+        model_list = []
+        print("Available Models by Provider:\n")
+        
+        for provider, models in providers.items():
+            if models:  # Only show providers with available models
+                print(f"=== {provider.upper()} ===")
+                for model_id, model_info in models:
+                    marker = "►" if model_id == self.selected_model else " "
+                    display_name = model_info.get("display_name", model_id)
+                    model_list.append(model_id)
+                    print(f"{marker} {len(model_list):2d}. {display_name}")
+                print()
+        
+        if not model_list:
+            print("No models available. Please check your API keys or Ollama installation.")
+            input("Press Enter to continue...")
+            return
+        
+        print("Enter model number to select (or press Enter to cancel):")
+        
+        try:
+            choice = input("> ").strip()
+            if choice and choice.isdigit():
+                idx = int(choice) - 1
+                if 0 <= idx < len(model_list):
+                    old_model = self.selected_model
+                    self.selected_model = model_list[idx]
+                    print(f"Model changed from {old_model} to {self.selected_model}")
+                    input("Press Enter to continue...")
+        except (ValueError, KeyboardInterrupt):
+            pass
+    
+    def _select_style(self):
+        """Style selection submenu"""
+        self.clear_screen()
+        print("=" * self.width)
+        print("RESPONSE STYLE SELECTION".center(self.width))
+        print("=" * self.width)
+        
+        styles = list(CONFIG["user_styles"].keys())
+        for i, style in enumerate(styles):
+            marker = "►" if style == self.selected_style else " "
+            name = CONFIG["user_styles"][style]["name"]
+            description = CONFIG["user_styles"][style]["description"]
+            print(f"{marker} {i+1:2d}. {name}")
+            print(f"     {description}")
+            print()
+        
+        print("Enter style number to select (or press Enter to cancel):")
+        
+        try:
+            choice = input("> ").strip()
+            if choice and choice.isdigit():
+                idx = int(choice) - 1
+                if 0 <= idx < len(styles):
+                    old_style = self.selected_style
+                    self.selected_style = styles[idx]
+                    print(f"Style changed from {old_style} to {self.selected_style}")
+                    input("Press Enter to continue...")
+        except (ValueError, KeyboardInterrupt):
+            pass
+    
+    async def _detect_ollama_models(self):
+        """Detect and add locally available Ollama models"""
+        self.clear_screen()
+        print("=" * self.width)
+        print("OLLAMA MODEL DETECTION".center(self.width))
+        print("=" * self.width)
+        
+        print("Checking for local Ollama models...")
+        
+        try:
+            with self._suppress_output():
+                from .api.ollama import OllamaClient
+                client = await OllamaClient.create()
+                local_models = await client.get_available_models()
+            
+            if not local_models:
+                print("No local Ollama models found.")
+                print("Use the model browser ('m' key) to download models.")
+            else:
+                print(f"Found {len(local_models)} local Ollama models:")
+                print()
+                
+                new_models = 0
+                for model in local_models:
+                    model_id = model.get("id", "unknown")
+                    print(f"  • {model_id}")
+                    
+                    # Add to config if not already present
+                    if model_id not in CONFIG["available_models"]:
+                        CONFIG["available_models"][model_id] = {
+                            "provider": "ollama",
+                            "display_name": model_id,
+                            "max_tokens": 4096
+                        }
+                        new_models += 1
+                
+                if new_models > 0:
+                    save_config(CONFIG)
+                    print(f"\nAdded {new_models} new models to configuration.")
+                else:
+                    print("\nAll models already in configuration.")
+                    
+        except Exception as e:
+            print(f"Error detecting Ollama models: {str(e)}")
+            print("Make sure Ollama is running and accessible.")
+        
+        input("\nPress Enter to continue...")
+    
+    async def run(self):
+        """Main application loop"""
+        # Create initial conversation
+        await self.create_new_conversation()
+        
+        # Welcome message
+        self.draw_screen("", "Type your message (or 'q' to quit)")
+        
+        while self.running:
+            try:
+                user_input = self.get_input("Type your message")
+                
+                if not user_input:
+                    continue
+                
+                # Handle special command tokens from enhanced input
+                if user_input == "##QUIT##":
+                    self.running = False
+                    break
+                elif user_input == "##NEW##":
+                    await self.create_new_conversation()
+                    continue
+                elif user_input == "##HISTORY##":
+                    self.show_history()
+                    continue
+                elif user_input == "##SETTINGS##":
+                    await self.show_settings()
+                    continue
+                
+                # Handle legacy single-letter commands for backward compatibility
+                if user_input.lower() == 'q':
+                    self.running = False
+                    break
+                elif user_input.lower() == 'n':
+                    await self.create_new_conversation()
+                    continue
+                elif user_input.lower() == 'h':
+                    self.show_history()
+                    continue
+                elif user_input.lower() == 's':
+                    await self.show_settings()
+                    continue
+                
+                # Generate response
+                await self.generate_response(user_input)
+                
+            except KeyboardInterrupt:
+                if self.generating:
+                    self.generating = False
+                    print("\nGeneration cancelled.")
+                    time.sleep(1)
+                else:
+                    self.running = False
+                    break
+            except Exception as e:
+                print(f"\nError: {e}")
+                input("Press Enter to continue...")
+
+def setup_signal_handlers():
+    """Setup signal handlers for graceful shutdown"""
+    def signal_handler(signum, frame):
+        print("\n\nShutting down gracefully...")
+        sys.exit(0)
+    
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+
+async def main():
+    """Main entry point for console version"""
+    parser = argparse.ArgumentParser(description="Chat Console - Pure Terminal Version")
+    parser.add_argument("--model", help="Initial model to use")
+    parser.add_argument("--style", help="Response style")
+    parser.add_argument("message", nargs="?", help="Initial message to send")
+    
+    args = parser.parse_args()
+    
+    # Setup signal handling
+    setup_signal_handlers()
+    
+    # Create console UI
+    console = ConsoleUI()
+    
+    if args.model:
+        console.selected_model = resolve_model_id(args.model)
+    if args.style:
+        console.selected_style = args.style
+    
+    # Run the application
+    await console.run()
+    
+    print("\nGoodbye!")
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("\nGoodbye!")
+    except Exception as e:
+        print(f"Error: {e}")
+        sys.exit(1)