astraagent 2.25.6

package/astra/chat.py

@@ -0,0 +1,763 @@
+"""Chat Mode - Enhanced with keyboard shortcuts, file access, and settings."""
+
+import asyncio
+import json
+import os
+from pathlib import Path
+from datetime import datetime
+from typing import List, Optional, Dict, Any
+
+from astra.llm.providers import (
+    Message, PROVIDERS, create_provider, LLMProvider
+)
+from astra.search import DeviceSearcher, format_search_results
+from astra.tasks import TaskExecutor, format_task_result
+from astra.updater import UpdateManager, format_update_info
+
+
+class ChatHistory:
+    """Manages chat history and logging."""
+    
+    def __init__(self, logs_dir: str = "./logs/chats"):
+        self.logs_dir = Path(logs_dir)
+        self.logs_dir.mkdir(parents=True, exist_ok=True)
+        self.messages: List[dict] = []
+        self.provider_name: str = ""
+        self.model_name: str = ""
+        self.session_start = datetime.now()
+    
+    def add_message(self, role: str, content: str):
+        """Add message to history."""
+        self.messages.append({
+            "timestamp": datetime.now().isoformat(),
+            "role": role,
+            "content": content
+        })
+    
+    def save(self):
+        """Save chat history to file."""
+        timestamp = self.session_start.strftime("%Y%m%d_%H%M%S")
+        filename = f"chat_{self.provider_name}_{self.model_name}_{timestamp}.json"
+        filepath = self.logs_dir / filename
+        
+        data = {
+            "provider": self.provider_name,
+            "model": self.model_name,
+            "session_start": self.session_start.isoformat(),
+            "session_end": datetime.now().isoformat(),
+            "total_messages": len(self.messages),
+            "messages": self.messages
+        }
+        
+        with open(filepath, "w", encoding="utf-8") as f:
+            json.dump(data, f, indent=2, ensure_ascii=False)
+        
+        return filepath
+
+
+class APIKeyManager:
+    """Manage API keys across providers."""
+    
+    ENV_FILE = Path(".env")
+    
+    @staticmethod
+    def get_all_keys() -> Dict[str, Optional[str]]:
+        """Get all API keys from environment."""
+        keys = {}
+        for provider_key, provider_info in PROVIDERS.items():
+            env_keys = provider_info.get("env_keys", [])
+            if env_keys:
+                key = os.getenv(env_keys[0])
+                if key:
+                    # Mask most of the key
+                    masked = key[:4] + "*" * (len(key) - 8) + key[-4:]
+                    keys[provider_key] = masked
+                else:
+                    keys[provider_key] = "NOT SET"
+        return keys
+    
+    @staticmethod
+    def set_key(provider_key: str, api_key: str):
+        """Set API key for provider."""
+        provider_info = PROVIDERS.get(provider_key)
+        if not provider_info:
+            print(f"❌ Unknown provider: {provider_key}")
+            return False
+        
+        env_key = provider_info["env_keys"][0]
+        
+        # Update .env file
+        content = APIKeyManager.ENV_FILE.read_text() if APIKeyManager.ENV_FILE.exists() else ""
+        
+        if f"{env_key}=" in content:
+            # Update existing
+            lines = content.split("\n")
+            content = "\n".join(f"{env_key}={api_key}" if line.startswith(f"{env_key}=") else line for line in lines)
+        else:
+            # Add new
+            if content and not content.endswith("\n"):
+                content += "\n"
+            content += f"{env_key}={api_key}\n"
+        
+        APIKeyManager.ENV_FILE.write_text(content)
+        os.environ[env_key] = api_key
+        print(f"✅ API key saved for {provider_key}")
+        return True
+    
+    @staticmethod
+    def view_keys():
+        """View all API keys (masked)."""
+        keys = APIKeyManager.get_all_keys()
+        print("\n🔑 API Keys Status:\n")
+        for provider, status in keys.items():
+            check = "✅" if status != "NOT SET" else "❌"
+            print(f"  {check} {provider:15} {status}")
+
+
+def show_help():
+    """Show help and keyboard shortcuts."""
+    print("\n" + "="*70)
+    print("⌨️  KEYBOARD SHORTCUTS & COMMANDS")
+    print("="*70)
+    print("""
+CHAT COMMANDS:
+  /help or ?              Show this help
+  /exit or /quit or /q    Close chat and save
+  /close                  Close chat (without saving)
+  /clear                  Clear conversation history
+  /history or /h          Show chat history
+  /settings or /s         Open settings menu
+  /file or /f             Read and send file contents
+  /files or /ls           List files in current directory
+  /docs                   List documentation files
+  /images or /img         List image files
+  /view <filename>        View file contents
+  /keys or /k             View API keys status
+  /models or /m           Show available models for current provider
+  /provider or /p         Show current provider info
+  /tokens or /t           Show token estimate
+
+SEARCH COMMANDS (Device Search):
+  /search <name>          Search for file/folder by name
+  /find <name>            Alias for search
+  /search-ext <ext>       Search files by extension
+  /search-type <type>     Search by file type (code, image, doc, etc)
+  /search-recent          Find recently modified files
+  /search-size <kb>       Find files by size
+  /search-content <text>  Search file contents
+  /browse <path>          Browse directory
+
+TASK COMMANDS (File Operations):
+  /open <path>            Open file with default app
+  /folder <path>          Open folder in file explorer
+  /copy <src> <dest>      Copy file
+  /move <src> <dest>      Move file
+  /delete <path>          Delete file
+  /rename <path> <name>   Rename file
+  /mkdir <path>           Create folder
+  /info <path>            Get file/folder info
+  /ls <path>              List directory contents
+
+UPDATE COMMANDS:
+  /updates                Check for available updates
+  /update-check           Force update check
+  /update npm             Update via npm
+  /update pip             Update via pip
+  /update github          Update via github
+  /version                Show current version
+  /rollback <version>     Rollback to previous version
+
+SETTINGS MENU (/settings):
+  - Change provider
+  - Change model
+  - Update API key
+  - Adjust temperature
+  - Set max tokens
+  - View current settings
+
+FILE ACCESS (/file):
+  - Load .txt, .md, .py, .json, .html files
+  - Files automatically included in AI context
+  - Max 100KB per file
+
+IMAGE SUPPORT (Paste):
+  - Paste images from clipboard (Ctrl+V)
+  - Send image URLs directly
+  - Supported: PNG, JPG, WebP, GIF
+  - Provider must support vision
+
+TIPS:
+  • Type normally to send messages
+  • Use /help anytime for this reference
+  • Search works across entire device
+  • Tasks handle file operations safely
+  • Updates check npm and PyPI simultaneously
+""")
+    print("="*70 + "\n")
+  • Use Ctrl+C to interrupt response
+  • Use Tab for auto-completion (if available)
+  • Multi-line input: Use Shift+Enter
+  • Type /help anytime for quick reference
+""")
+    print("="*70 + "\n")
+
+
+def read_file_content(filepath: str, max_size: int = 100000) -> Optional[str]:
+    """Safely read file content."""
+    try:
+        path = Path(filepath)
+        if not path.exists():
+            print(f"❌ File not found: {filepath}")
+            return None
+        
+        if path.stat().st_size > max_size:
+            print(f"❌ File too large ({path.stat().st_size} bytes, max {max_size})")
+            return None
+        
+        # Allow safe file types
+        safe_extensions = {'.txt', '.md', '.py', '.js', '.json', '.html', '.xml', '.csv', '.log'}
+        if path.suffix not in safe_extensions:
+            print(f"❌ File type not supported. Safe types: {', '.join(safe_extensions)}")
+            return None
+        
+        with open(path, 'r', encoding='utf-8', errors='ignore') as f:
+            content = f.read()
+        
+        return content
+    except Exception as e:
+        print(f"❌ Error reading file: {e}")
+        return None
+
+
+def list_files(pattern: str = "*", directory: str = ".") -> List[str]:
+    """List files matching pattern."""
+    try:
+        path = Path(directory)
+        files = sorted([f.name for f in path.glob(pattern) if f.is_file()])
+        return files[:20]  # Max 20 files
+    except Exception:
+        return []
+
+
+def list_docs() -> List[str]:
+    """List documentation files."""
+    docs = list_files("*.md", ".")
+    return docs
+
+
+def list_images() -> List[str]:
+    """List image files."""
+    img_extensions = ['*.png', '*.jpg', '*.jpeg', '*.webp', '*.gif', '*.bmp']
+    images = []
+    for ext in img_extensions:
+        images.extend(list_files(ext, "."))
+    return sorted(list(set(images)))
+
+
+def select_provider() -> tuple[str, str]:
+    """Interactive provider selection."""
+    print("\n📦 Available LLM Providers:\n")
+    
+    providers_list = list(PROVIDERS.keys())
+    for i, provider_key in enumerate(providers_list, 1):
+        info = PROVIDERS[provider_key]
+        free_tag = "💰 Free" if info.get("free") else "💳 Paid"
+        print(f"  {i}. {info['name']} [{free_tag}]")
+    
+    print(f"  0. View API key status")
+    
+    while True:
+        try:
+            choice = input(f"\nSelect provider (0-{len(providers_list)}): ").strip()
+            idx = int(choice)
+            
+            if idx == 0:
+                APIKeyManager.view_keys()
+                continue
+            
+            if 1 <= idx <= len(providers_list):
+                selected = providers_list[idx - 1]
+                return selected, PROVIDERS[selected]["name"]
+            
+            print("❌ Invalid choice")
+        except (ValueError, IndexError):
+            print("❌ Invalid input")
+
+
+def select_model(provider_key: str) -> str:
+    """Interactive model selection."""
+    provider_info = PROVIDERS[provider_key]
+    models = provider_info["models"]
+    
+    print(f"\n🤖 Available models for {provider_info['name']}:\n")
+    
+    for i, model in enumerate(models, 1):
+        print(f"  {i}. {model}")
+    
+    print(f"  {len(models) + 1}. Custom model (enter name)")
+    
+    while True:
+        try:
+            choice = input(f"\nSelect model (1-{len(models) + 1}): ").strip()
+            idx = int(choice) - 1
+            
+            if 0 <= idx < len(models):
+                return models[idx]
+            elif idx == len(models):
+                custom = input("Enter custom model name: ").strip()
+                if custom:
+                    return custom
+                print("❌ Model name cannot be empty")
+            else:
+                print("❌ Invalid choice")
+        except ValueError:
+            print("❌ Invalid input")
+
+
+def get_api_key(provider_key: str) -> str:
+    """Get API key for provider."""
+    provider_info = PROVIDERS[provider_key]
+    env_keys = provider_info["env_keys"]
+    
+    print(f"\n🔑 API Key for {provider_info['name']}")
+    
+    # Check if already set
+    for env_key in env_keys:
+        existing = os.getenv(env_key)
+        if existing:
+            use_existing = input(f"Found {env_key} in environment. Use it? (y/n): ").lower()
+            if use_existing == 'y':
+                return existing
+    
+    # Get from user
+    print(f"Enter your API key (or paste it):")
+    api_key = input("> ").strip()
+    
+    if not api_key:
+        raise ValueError("API key cannot be empty")
+    
+    # Offer to save
+    save_to_env = input("\nSave to environment variable? (y/n): ").lower()
+    if save_to_env == 'y':
+        APIKeyManager.set_key(provider_key, api_key)
+    
+    return api_key
+
+
+async def settings_menu(provider_key: str, model_name: str, provider_info: dict) -> tuple[str, str]:
+    """Interactive settings menu."""
+    while True:
+        print("\n" + "="*50)
+        print("⚙️  SETTINGS MENU")
+        print("="*50)
+        print(f"\nCurrent Settings:")
+        print(f"  Provider: {provider_key}")
+        print(f"  Model: {model_name}")
+        print(f"  Temperature: 0.7 (default)")
+        print(f"  Max Tokens: 4096 (default)")
+        print(f"\nOptions:")
+        print(f"  1. Change provider")
+        print(f"  2. Change model")
+        print(f"  3. Update/set API key")
+        print(f"  4. View all API keys")
+        print(f"  5. Back to chat")
+        
+        choice = input(f"\nSelect option (1-5): ").strip()
+        
+        if choice == '1':
+            provider_key, _ = select_provider()
+            print(f"✅ Provider changed to: {provider_key}")
+            model_name = select_model(provider_key)
+            print(f"✅ Model changed to: {model_name}")
+        
+        elif choice == '2':
+            model_name = select_model(provider_key)
+            print(f"✅ Model changed to: {model_name}")
+        
+        elif choice == '3':
+            new_key = input("Enter new API key: ").strip()
+            if new_key:
+                APIKeyManager.set_key(provider_key, new_key)
+        
+        elif choice == '4':
+            APIKeyManager.view_keys()
+            input("\nPress Enter to continue...")
+        
+        elif choice == '5':
+            return provider_key, model_name
+        
+        else:
+            print("❌ Invalid option")
+
+
+async def chat_with_provider(provider: LLMProvider, history: ChatHistory, provider_key: str):
+    """Interactive chat loop with enhanced features."""
+    print(f"\n💬 Chat Mode")
+    print(f"Provider: {history.provider_name} | Model: {history.model_name}")
+    print(f"Type '/help' for commands or just start typing\n")
+    
+    conversation = []
+    should_close = False
+    
+    while True:
+        try:
+            user_input = input("You: ").strip()
+        except (EOFError, KeyboardInterrupt):
+            print("\n[*] Chat interrupted")
+            break
+        
+        if not user_input:
+            continue
+        
+        # Handle shortcuts
+        lower_input = user_input.lower()
+        
+        if lower_input in ['/exit', '/quit', '/q', 'exit', 'quit']:
+            should_close = True
+            break
+        
+        if lower_input == '/close':
+            should_close = False
+            break
+        
+        if lower_input in ['/help', '?', '/h']:
+            show_help()
+            continue
+        
+        if lower_input == '/clear':
+            conversation.clear()
+            history.messages.clear()
+            print("✅ Conversation cleared\n")
+            continue
+        
+        if lower_input in ['/history', '/h']:
+            if not conversation:
+                print("No messages yet.\n")
+            else:
+                print("\n📜 Chat History:\n")
+                for i, msg in enumerate(conversation, 1):
+                    role = msg["role"].upper()
+                    content = msg["content"][:80]
+                    print(f"  {i}. [{role}] {content}...")
+                print()
+            continue
+        
+        if lower_input in ['/keys', '/k']:
+            APIKeyManager.view_keys()
+            print()
+            continue
+        
+        if lower_input in ['/models', '/m']:
+            provider_info = PROVIDERS[provider_key]
+            print(f"\n🤖 Available models for {provider_info['name']}:")
+            for model in provider_info["models"]:
+                print(f"  • {model}")
+            print()
+            continue
+        
+        if lower_input in ['/provider', '/p']:
+            provider_info = PROVIDERS[provider_key]
+            print(f"\n📦 Current Provider: {provider_info['name']}")
+            print(f"   Key: {provider_key}")
+            print(f"   Models: {len(provider_info['models'])} available")
+            print()
+            continue
+        
+        if lower_input in ['/files', '/ls']:
+            files = list_files()
+            print(f"\n📁 Files in current directory:\n")
+            for f in files:
+                print(f"  • {f}")
+            print()
+            continue
+        
+        if lower_input == '/docs':
+            docs = list_docs()
+            print(f"\n📖 Documentation files:\n")
+            for doc in docs:
+                print(f"  • {doc}")
+            print()
+            continue
+        
+        if lower_input in ['/images', '/img']:
+            images = list_images()
+            if not images:
+                print("❌ No image files found\n")
+            else:
+                print(f"\n🖼️  Image files:\n")
+                for img in images:
+                    print(f"  • {img}")
+                print()
+            continue
+        
+        if lower_input.startswith('/view '):
+            filename = user_input[6:].strip()
+            content = read_file_content(filename)
+            if content:
+                print(f"\n📄 {filename}:\n{content}\n")
+            continue
+        
+        if lower_input.startswith('/file ') or lower_input.startswith('/f '):
+            filename = user_input.split(' ', 1)[1].strip() if ' ' in user_input else ""
+            if not filename:
+                print("Usage: /file <filename>\n")
+                continue
+            
+            content = read_file_content(filename)
+            if content:
+                user_input = f"Please review this file and provide insights:\n\n```\n{content}\n```"
+                print(f"✅ File loaded ({len(content)} bytes)\n")
+            else:
+                continue
+        
+        if lower_input in ['/settings', '/s']:
+            # Need to update provider and model through settings
+            print("Settings menu - returning to main chat\n")
+            continue
+        
+        # SEARCH COMMANDS
+        if lower_input.startswith('/search ') or lower_input.startswith('/find '):
+            search_term = user_input.split(' ', 1)[1].strip()
+            searcher = DeviceSearcher(max_results=20)
+            results = searcher.search_by_name(search_term)
+            print(format_search_results(results, f"Search for '{search_term}'"))
+            continue
+        
+        if lower_input.startswith('/search-ext '):
+            ext = user_input.split(' ', 1)[1].strip()
+            searcher = DeviceSearcher(max_results=30)
+            results = searcher.search_by_extension(ext, limit=30)
+            print(format_search_results(results, f"Files with extension '.{ext}'"))
+            continue
+        
+        if lower_input.startswith('/search-type '):
+            file_type = user_input.split(' ', 1)[1].strip().lower()
+            searcher = DeviceSearcher(max_results=30)
+            results = searcher.search_by_category(file_type, limit=30)
+            print(format_search_results(results, f"Files of type '{file_type}'"))
+            continue
+        
+        if lower_input == '/search-recent':
+            searcher = DeviceSearcher(max_results=20)
+            results = searcher.search_by_date(days_ago=7)
+            print(format_search_results(results, "Files modified in last 7 days"))
+            continue
+        
+        if lower_input.startswith('/search-size '):
+            try:
+                size_kb = int(user_input.split(' ', 1)[1].strip())
+                size_bytes = size_kb * 1024
+                searcher = DeviceSearcher(max_results=20)
+                results = searcher.search_by_size(min_size=size_bytes, limit=20)
+                print(format_search_results(results, f"Files larger than {size_kb} KB"))
+            except:
+                print("❌ Usage: /search-size <kb>\n")
+            continue
+        
+        if lower_input.startswith('/search-content '):
+            search_text = user_input.split(' ', 1)[1].strip()
+            searcher = DeviceSearcher(max_results=20)
+            results = searcher.search_by_content(search_text, limit=20)
+            print(format_search_results(results, f"Files containing '{search_text}'"))
+            continue
+        
+        if lower_input.startswith('/browse '):
+            path = user_input.split(' ', 1)[1].strip()
+            searcher = DeviceSearcher()
+            result = searcher.get_directory_contents(path)
+            if 'error' in result:
+                print(f"❌ Error: {result['error']}\n")
+            else:
+                output = f"\n📁 Directory: {result['path']}\n"
+                output += f"   Total folders: {result['total_dirs']}\n"
+                output += f"   Total files: {result['total_files']}\n\n"
+                
+                if result['directories']:
+                    output += "   📂 Folders:\n"
+                    for d in result['directories']:
+                        output += f"      {d['name']}\n"
+                
+                if result['files']:
+                    output += "\n   📄 Files:\n"
+                    for f in result['files']:
+                        output += f"      {f['name']} ({f['size_kb']} KB)\n"
+                
+                print(output)
+            continue
+        
+        # TASK COMMANDS
+        if lower_input.startswith('/open '):
+            path = user_input.split(' ', 1)[1].strip()
+            result = TaskExecutor.open_file(path)
+            print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/folder '):
+            path = user_input.split(' ', 1)[1].strip()
+            result = TaskExecutor.open_folder(path)
+            print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/copy '):
+            parts = user_input.split(' ')
+            if len(parts) < 3:
+                print("❌ Usage: /copy <source> <destination>\n")
+                continue
+            source = parts[1]
+            dest = ' '.join(parts[2:])
+            result = TaskExecutor.copy_file(source, dest)
+            print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/move '):
+            parts = user_input.split(' ')
+            if len(parts) < 3:
+                print("❌ Usage: /move <source> <destination>\n")
+                continue
+            source = parts[1]
+            dest = ' '.join(parts[2:])
+            result = TaskExecutor.move_file(source, dest)
+            print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/delete '):
+            path = user_input.split(' ', 1)[1].strip()
+            confirm = input(f"⚠️  Delete '{path}'? (y/N): ").strip().lower()
+            if confirm == 'y':
+                result = TaskExecutor.delete_file(path)
+                print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/rename '):
+            parts = user_input.split(' ')
+            if len(parts) < 3:
+                print("❌ Usage: /rename <path> <new_name>\n")
+                continue
+            path = parts[1]
+            new_name = ' '.join(parts[2:])
+            result = TaskExecutor.rename_file(path, new_name)
+            print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/mkdir '):
+            path = user_input.split(' ', 1)[1].strip()
+            result = TaskExecutor.create_folder(path)
+            print(format_task_result(result))
+            continue
+        
+        if lower_input.startswith('/info '):
+            path = user_input.split(' ', 1)[1].strip()
+            result = TaskExecutor.get_file_info(path)
+            print(format_task_result(result))
+            continue
+        
+        # UPDATE COMMANDS
+        if lower_input in ['/updates', '/update-check']:
+            updater = UpdateManager()
+            force = lower_input == '/update-check'
+            updates = updater.check_for_updates(force=force)
+            print(format_update_info(updates))
+            continue
+        
+        if lower_input.startswith('/update '):
+            source = user_input.split(' ', 1)[1].strip().lower()
+            if source not in ['npm', 'pip', 'github']:
+                print("❌ Usage: /update <npm|pip|github>\n")
+                continue
+            
+            confirm = input(f"⚠️  Update via {source}? (y/N): ").strip().lower()
+            if confirm == 'y':
+                updater = UpdateManager()
+                result = updater.download_update(source)
+                print(format_task_result(result))
+            continue
+        
+        if lower_input == '/version':
+            updater = UpdateManager()
+            version_info = updater.get_version_info()
+            output = f"\n📦 Version Information:\n"
+            output += f"   Project: {version_info['project']}\n"
+            output += f"   Version: {version_info['current_version']}\n"
+            output += f"   Python: {version_info['python_version']}\n"
+            output += f"   Platform: {version_info['platform']}\n"
+            output += f"   npm available: {'✅' if version_info['npm_available'] else '❌'}\n"
+            output += f"   pip available: {'✅' if version_info['pip_available'] else '❌'}\n\n"
+            print(output)
+            continue
+        
+        if lower_input.startswith('/rollback '):
+            version = user_input.split(' ', 1)[1].strip()
+            confirm = input(f"⚠️  Rollback to {version}? (y/N): ").strip().lower()
+            if confirm == 'y':
+                updater = UpdateManager()
+                result = updater.rollback_version(version)
+                print(format_task_result(result))
+            continue
+        
+        # Add user message
+        conversation.append({"role": "user", "content": user_input})
+        history.add_message("user", user_input)
+        
+        # Get AI response
+        try:
+            print("🤖 ", end="", flush=True)
+            messages = [Message(role=msg["role"], content=msg["content"]) for msg in conversation]
+            response = await provider.generate(messages, temperature=0.7)
+            
+            ai_response = response.content
+            print(ai_response, end="\n\n", flush=True)
+            
+            # Add to history
+            conversation.append({"role": "assistant", "content": ai_response})
+            history.add_message("assistant", ai_response)
+            
+        except RuntimeError as e:
+            print(f"\n❌ Error: {str(e)}\n")
+        except Exception as e:
+            print(f"\n❌ Unexpected error: {str(e)}\n")
+    
+    return should_close
+
+
+async def start_chat_mode():
+    """Start interactive chat mode with enhanced features."""
+    try:
+        # Select provider
+        provider_key, provider_name = select_provider()
+        
+        # Select model
+        model_name = select_model(provider_key)
+        
+        # Get API key
+        try:
+            api_key = get_api_key(provider_key)
+        except ValueError as e:
+            print(f"❌ {str(e)}")
+            return
+        
+        # Create provider
+        try:
+            provider = create_provider(provider_key, model=model_name, api_key=api_key)
+        except ValueError as e:
+            print(f"❌ {str(e)}")
+            return
+        
+        # Create history
+        history = ChatHistory()
+        history.provider_name = provider_key
+        history.model_name = model_name.replace("/", "_")[:30]
+        
+        # Start chat
+        should_save = await chat_with_provider(provider, history, provider_key)
+        
+        # Save history if not explicitly closed
+        if should_save:
+            filepath = history.save()
+            print(f"\n✅ Chat saved to: {filepath}")
+        else:
+            print(f"\n⚠️  Chat closed without saving")
+        
+    except KeyboardInterrupt:
+        print("\n[*] Chat interrupted")
+    except Exception as e:
+        print(f"❌ Error: {str(e)}")