diff --git a/INSTALL.md b/INSTALL.md
new file mode 100644
index 0000000..b035c75
--- /dev/null
+++ b/INSTALL.md
@@ -0,0 +1,314 @@
+# 🚀 Heidi CLI One-Click Installer
+
+Install Heidi CLI with a single command! The installer will automatically:
+- Clone the latest version from GitHub
+- Install all dependencies
+- Build and install Heidi CLI
+- Verify the installation
+- Clean up temporary files
+
+## 📋 **System Requirements**
+
+### **Required**
+- **Python 3.10+** - [Download Python](https://python.org)
+- **Git** - [Download Git](https://git-scm.com)
+- **pip** - Usually included with Python
+
+### **Optional**
+- **CUDA** - For GPU acceleration (automatic if available)
+- **Admin/root access** - For system-wide installation (optional)
+
+---
+
+## 🖥️ **Installation**
+
+### **Linux/macOS**
+
+```bash
+# Download and run the installer
+curl -fsSL https://raw.githubusercontent.com/heidi-dang/heidi-cli/main/install | bash
+
+# Or download first, then run
+wget https://raw.githubusercontent.com/heidi-dang/heidi-cli/main/install
+chmod +x install
+./install
+```
+
+### **Windows (PowerShell)**
+
+```powershell
+# Download and run the installer
+Invoke-WebRequest -Uri "https://raw.githubusercontent.com/heidi-dang/heidi-cli/main/install.ps1" -OutFile "install.ps1"
+Set-ExecutionPolicy -ExecutionPolicy RemoteSigned -Scope CurrentUser
+.\install.ps1
+```
+
+### **From Local Clone**
+
+```bash
+# Clone the repository
+git clone https://github.com/heidi-dang/heidi-cli.git
+cd heidi-cli
+
+# Run the installer
+./install  # Linux/macOS
+# or
+.\install.ps1  # Windows
+```
+
+---
+
+## 🎯 **What the Installer Does**
+
+### **✅ Automatic Steps**
+
+1. **🔍 Check Requirements** - Verifies Python, Git, and pip
+2. **📥 Clone Repository** - Downloads latest Heidi CLI from main branch
+3. **🏗️ Setup Environment** - Creates virtual environment
+4. **📦 Install Dependencies** - Installs all required packages
+5. **🔨 Build Heidi CLI** - Compiles and builds the application
+6. **⚙️ Install Heidi CLI** - Installs system-wide or for current user
+7. **✅ Verify Installation** - Tests that everything works
+8. **🧹 Clean Up** - Removes temporary files
+
+### **🎉 Installation Results**
+
+After successful installation, you'll have:
+- ✅ **Heidi CLI command** available system-wide
+- ✅ **All dependencies** installed and configured
+- ✅ **API key system** ready to use
+- ✅ **Model hosting** capabilities
+- ✅ **Complete documentation** accessible
+
+---
+
+## 🚀 **Post-Installation**
+
+### **Quick Start**
+
+```bash
+# Verify installation
+heidi --version
+
+# Run setup wizard
+heidi setup
+
+# Generate API key
+heidi api generate --name "My First Key"
+
+# Start model server
+heidi model serve
+
+# View help
+heidi --help
+```
+
+### **Next Steps**
+
+1. **🔑 Generate API Key**
+   ```bash
+   heidi api generate --name "Production Key" --user "your-username"
+   ```
+
+2. **🤖 Download Models**
+   ```bash
+   heidi hf search "text-generation" --limit 5
+   heidi hf download "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+   ```
+
+3. **🌐 Start API Server**
+   ```bash
+   heidi api server
+   ```
+
+4. **📚 Explore Documentation**
+   ```bash
+   # View comprehensive guide
+   cat docs/how-to-use.md
+   ```
+
+---
+
+## 🔧 **Installation Options**
+
+### **User vs System Installation**
+
+**User Installation (Default)**
+- Installs to `~/.local/bin` (Linux/macOS) or `%USERPROFILE%\AppData\Local\Programs` (Windows)
+- No admin privileges required
+- Available only for current user
+
+**System Installation (Admin Required)**
+- Installs to `/usr/local/bin` (Linux/macOS) or system-wide (Windows)
+- Requires admin/root privileges
+- Available for all users
+
+### **Custom Installation Path**
+
+```bash
+# Linux/macOS - set custom prefix
+export HEIDI_INSTALL_PREFIX="/opt/heidi-cli"
+./install
+
+# Windows - custom path
+.\install.ps1 -InstallPath "C:\HeidiCLI"
+```
+
+---
+
+## 🛠️ **Troubleshooting**
+
+### **Common Issues**
+
+#### **Python Not Found**
+```bash
+# Install Python 3.10+
+# Ubuntu/Debian:
+sudo apt update && sudo apt install python3 python3-pip python3-venv
+
+# macOS (with Homebrew):
+brew install python3
+
+# Windows: Download from python.org
+```
+
+#### **Git Not Found**
+```bash
+# Ubuntu/Debian:
+sudo apt install git
+
+# macOS (with Homebrew):
+brew install git
+
+# Windows: Download from git-scm.com
+```
+
+#### **Permission Denied**
+```bash
+# Make installer executable
+chmod +x install
+
+# Or use bash directly
+bash install
+```
+
+#### **Command Not Found After Installation**
+```bash
+# Add to PATH (Linux/macOS)
+echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.bashrc
+source ~/.bashrc
+
+# Or restart your terminal
+```
+
+#### **Virtual Environment Issues**
+```bash
+# Clean installation
+rm -rf ~/.heidi
+./install --force
+```
+
+### **Get Help**
+
+```bash
+# Check system requirements
+heidi doctor
+
+# View installation logs
+# Installer shows detailed error messages
+
+# Get community help
+# GitHub Issues: https://github.com/heidi-dang/heidi-cli/issues
+# Discord: https://discord.gg/heidi-cli
+```
+
+---
+
+## 📊 **Installation Verification**
+
+### **Check Installation Status**
+
+```bash
+# Verify Heidi CLI is installed
+heidi --version
+
+# Check all systems
+heidi doctor
+
+# Test API key system
+heidi api config
+
+# List available models
+heidi api models
+```
+
+### **Expected Output**
+
+```
+🔑 Heidi API Key: heidik_abc123...
+📋 User Information:
+┌─────────────────┬─────────────────┐
+│ User ID:        │ default         │
+│ Key Name:       │ Demo Key        │
+│ Rate Limit:     │ 100 requests/min │
+│ Usage Count:    │ 0 requests      │
+└─────────────────┴─────────────────┘
+
+🤖 Available Models:
+• local://opencode-gpt-4
+• hf://TinyLlama/TinyLlama-1.1B-Chat-v1.0
+• opencode://gpt-4
+```
+
+---
+
+## 🔄 **Update Heidi CLI**
+
+### **Update to Latest Version**
+
+```bash
+# Re-run the installer (will update existing installation)
+./install
+
+# Or manually update
+pip install --upgrade heidi-cli
+```
+
+### **Uninstall Heidi CLI**
+
+```bash
+# Uninstall with pip
+pip uninstall heidi-cli
+
+# Remove configuration (optional)
+rm -rf ~/.heidi
+```
+
+---
+
+## 🎉 **Success!**
+
+**🚀 Heidi CLI is now installed and ready to use!**
+
+### **What You Can Do Now**
+
+- ✅ **Generate API keys** for unified model access
+- ✅ **Host models locally** with automatic management
+- ✅ **Access HuggingFace models** with smart integration
+- ✅ **Track usage and costs** with built-in analytics
+- ✅ **Scale to production** with enterprise features
+
+### **Learn More**
+
+- 📖 **Complete Guide**: [docs/how-to-use.md](docs/how-to-use.md)
+- 🔑 **API Keys**: [docs/api-keys.md](docs/api-keys.md)
+- 🤖 **Model Management**: [docs/model-management.md](docs/model-management.md)
+- 💬 **Community**: [Discord Server](https://discord.gg/heidi-cli)
+
+---
+
+**🎯 Installation completed successfully! Welcome to Heidi CLI!** 🚀
+
+*Last updated: March 2026*  
+*Installer version: 1.0.0*  
+*Heidi CLI version: 0.1.1*
diff --git a/install b/install
new file mode 100755
index 0000000..787de90
--- /dev/null
+++ b/install
@@ -0,0 +1,294 @@
+#!/bin/bash
+#
+# Heidi CLI One-Click Installer
+# 
+# This script will:
+# 1. Clone the latest Heidi CLI from main branch
+# 2. Build and install it system-wide
+# 3. Verify installation
+# 4. Clean up temporary files
+#
+# Usage: ./install
+#
+
+set -e  # Exit on any error
+
+# Colors for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+PURPLE='\033[0;35m'
+CYAN='\033[0;36m'
+NC='\033[0m' # No Color
+
+# Print functions
+print_header() {
+    echo -e "${PURPLE}================================${NC}"
+    echo -e "${PURPLE}🚀 Heidi CLI One-Click Installer${NC}"
+    echo -e "${PURPLE}================================${NC}"
+    echo
+}
+
+print_success() {
+    echo -e "${GREEN}✅ $1${NC}"
+}
+
+print_info() {
+    echo -e "${BLUE}ℹ️  $1${NC}"
+}
+
+print_warning() {
+    echo -e "${YELLOW}⚠️  $1${NC}"
+}
+
+print_error() {
+    echo -e "${RED}❌ $1${NC}"
+}
+
+print_step() {
+    echo -e "${CYAN}🔄 $1${NC}"
+}
+
+# Check if running as root (for system-wide installation)
+check_permissions() {
+    print_step "Checking permissions..."
+    
+    if [[ $EUID -eq 0 ]]; then
+        print_warning "Running as root - installing system-wide"
+        INSTALL_PREFIX="/usr/local"
+        PIP_INSTALL_ARGS="--user"
+    else
+        print_info "Running as user - installing in user directory"
+        INSTALL_PREFIX="$HOME/.local"
+        PIP_INSTALL_ARGS="--user"
+        
+        # Add ~/.local/bin to PATH if not already there
+        if [[ ":$PATH:" != *":$HOME/.local/bin:"* ]]; then
+            export PATH="$HOME/.local/bin:$PATH"
+            echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.bashrc
+            print_info "Added ~/.local/bin to PATH"
+        fi
+    fi
+    
+    print_success "Permissions checked - installing to $INSTALL_PREFIX"
+}
+
+# Check system requirements
+check_requirements() {
+    print_step "Checking system requirements..."
+    
+    # Check Python
+    if ! command -v python3 &> /dev/null; then
+        print_error "Python 3 is required but not installed"
+        print_info "Please install Python 3.10 or higher"
+        exit 1
+    fi
+    
+    python_version=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
+    required_version="3.10"
+    
+    if python3 -c "import sys; exit(0 if sys.version_info >= (3, 10) else 1)"; then
+        print_success "Python $python_version found (>= $required_version)"
+    else
+        print_error "Python $python_version found but >= $required_version required"
+        exit 1
+    fi
+    
+    # Check pip
+    if ! command -v pip3 &> /dev/null && ! command -v pip &> /dev/null; then
+        print_error "pip is required but not installed"
+        print_info "Please install pip"
+        exit 1
+    fi
+    
+    # Check git
+    if ! command -v git &> /dev/null; then
+        print_error "git is required but not installed"
+        print_info "Please install git"
+        exit 1
+    fi
+    
+    print_success "All requirements satisfied"
+}
+
+# Create temporary directory
+create_temp_dir() {
+    TEMP_DIR=$(mktemp -d)
+    print_info "Created temporary directory: $TEMP_DIR"
+    cd "$TEMP_DIR"
+}
+
+# Clone Heidi CLI
+clone_heidi() {
+    print_step "Cloning Heidi CLI from main branch..."
+    
+    REPO_URL="https://github.com/heidi-dang/heidi-cli.git"
+    
+    if ! git clone "$REPO_URL" heidi-cli; then
+        print_error "Failed to clone Heidi CLI repository"
+        exit 1
+    fi
+    
+    cd heidi-cli
+    print_success "Heidi CLI cloned successfully"
+    
+    # Show current commit info
+    COMMIT_HASH=$(git rev-parse HEAD)
+    COMMIT_DATE=$(git log -1 --format="%cd" --date=short)
+    print_info "Installing commit: $COMMIT_HASH ($COMMIT_DATE)"
+}
+
+# Setup virtual environment
+setup_venv() {
+    print_step "Setting up virtual environment..."
+    
+    if ! python3 -m venv venv; then
+        print_error "Failed to create virtual environment"
+        exit 1
+    fi
+    
+    source venv/bin/activate
+    print_success "Virtual environment created and activated"
+    
+    # Upgrade pip
+    pip install --upgrade pip
+    print_success "pip upgraded"
+}
+
+# Install dependencies
+install_dependencies() {
+    print_step "Installing dependencies..."
+    
+    if ! pip install -e .; then
+        print_error "Failed to install Heidi CLI dependencies"
+        exit 1
+    fi
+    
+    print_success "Dependencies installed"
+}
+
+# Build Heidi CLI
+build_heidi() {
+    print_step "Building Heidi CLI..."
+    
+    # Check if build tools are available
+    if command -v python3 -m build &> /dev/null; then
+        # Use modern build system
+        python3 -m build
+        print_success "Heidi CLI built with modern build system"
+    else
+        # Fallback to setup.py
+        python3 setup.py build
+        print_success "Heidi CLI built with setup.py"
+    fi
+}
+
+# Install Heidi CLI
+install_heidi() {
+    print_step "Installing Heidi CLI..."
+    
+    # Install in development mode with user permissions
+    if ! pip install -e . $PIP_INSTALL_ARGS; then
+        print_error "Failed to install Heidi CLI"
+        exit 1
+    fi
+    
+    print_success "Heidi CLI installed successfully"
+}
+
+# Verify installation
+verify_installation() {
+    print_step "Verifying installation..."
+    
+    # Check if heidi command is available
+    if command -v heidi &> /dev/null; then
+        print_success "heidi command is available"
+        
+        # Show version
+        VERSION=$(heidi --version 2>/dev/null || echo "unknown")
+        print_success "Heidi CLI version: $VERSION"
+        
+        # Test basic functionality
+        if heidi --help &> /dev/null; then
+            print_success "Heidi CLI help command works"
+        else
+            print_warning "Heidi CLI help command failed - installation may have issues"
+        fi
+    else
+        print_error "heidi command not found after installation"
+        print_info "Try running: export PATH=\"$HOME/.local/bin:\$PATH\""
+        exit 1
+    fi
+}
+
+# Run post-install setup
+post_install_setup() {
+    print_step "Running post-install setup..."
+    
+    # Initialize Heidi CLI
+    if heidi doctor &> /dev/null; then
+        print_success "Heidi CLI initialization successful"
+    else
+        print_warning "Heidi CLI initialization had issues - this is normal for first run"
+    fi
+    
+    # Show next steps
+    echo
+    print_info "🎉 Installation completed successfully!"
+    echo
+    print_info "Next steps:"
+    echo "  1. Run: heidi setup"
+    echo "  2. Generate API key: heidi api generate --name \"My Key\""
+    echo "  3. Start model server: heidi model serve"
+    echo "  4. View help: heidi --help"
+    echo
+    print_info "Documentation: https://github.com/heidi-dang/heidi-cli/blob/main/docs/how-to-use.md"
+}
+
+# Cleanup temporary files
+cleanup() {
+    print_step "Cleaning up temporary files..."
+    
+    if [[ -n "$TEMP_DIR" && -d "$TEMP_DIR" ]]; then
+        cd /
+        rm -rf "$TEMP_DIR"
+        print_success "Temporary files cleaned up"
+    fi
+}
+
+# Error handling
+handle_error() {
+    print_error "Installation failed!"
+    print_info "Check the error messages above for details"
+    print_info "You can try running the installer again"
+    cleanup
+    exit 1
+}
+
+# Main installation function
+main() {
+    print_header
+    
+    # Set up error handling
+    trap handle_error ERR
+    trap cleanup EXIT
+    
+    # Run installation steps
+    check_permissions
+    check_requirements
+    create_temp_dir
+    clone_heidi
+    setup_venv
+    install_dependencies
+    build_heidi
+    install_heidi
+    verify_installation
+    post_install_setup
+    
+    print_success "🎉 Heidi CLI installation completed successfully!"
+    print_info "You can now use: heidi --version"
+}
+
+# Run main function
+main "$@"
diff --git a/install.ps1 b/install.ps1
new file mode 100644
index 0000000..663776b
--- /dev/null
+++ b/install.ps1
@@ -0,0 +1,335 @@
+#
+# Heidi CLI One-Click Installer for Windows
+# 
+# This script will:
+# 1. Clone the latest Heidi CLI from main branch
+# 2. Build and install it
+# 3. Verify installation
+# 4. Clean up temporary files
+#
+# Usage: .\install.ps1
+#
+
+#Requires -RunAsAdministrator
+
+param(
+    [string]$InstallPath = "$env:USERPROFILE\heidi-cli",
+    [switch]$Force
+)
+
+# Colors for output
+$Colors = @{
+    Red = "Red"
+    Green = "Green"
+    Yellow = "Yellow"
+    Blue = "Blue"
+    Cyan = "Cyan"
+    Magenta = "Magenta"
+}
+
+# Print functions
+function Write-Header {
+    Write-Host "================================" -ForegroundColor $Colors.Magenta
+    Write-Host "🚀 Heidi CLI One-Click Installer" -ForegroundColor $Colors.Magenta
+    Write-Host "================================" -ForegroundColor $Colors.Magenta
+    Write-Host ""
+}
+
+function Write-Success {
+    param([string]$Message)
+    Write-Host "✅ $Message" -ForegroundColor $Colors.Green
+}
+
+function Write-Info {
+    param([string]$Message)
+    Write-Host "ℹ️  $Message" -ForegroundColor $Colors.Blue
+}
+
+function Write-Warning {
+    param([string]$Message)
+    Write-Host "⚠️  $Message" -ForegroundColor $Colors.Yellow
+}
+
+function Write-Error {
+    param([string]$Message)
+    Write-Host "❌ $Message" -ForegroundColor $Colors.Red
+}
+
+function Write-Step {
+    param([string]$Message)
+    Write-Host "🔄 $Message" -ForegroundColor $Colors.Cyan
+}
+
+# Check if running as Administrator
+function Test-Administrator {
+    $currentUser = [Security.Principal.WindowsIdentity]::GetCurrent()
+    $principal = New-Object Security.Principal.WindowsPrincipal($currentUser)
+    return $principal.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
+}
+
+# Check system requirements
+function Test-Requirements {
+    Write-Step "Checking system requirements..."
+    
+    # Check Python
+    try {
+        $python = Get-Command python -ErrorAction Stop
+        $pythonVersion = & python --version 2>&1
+        Write-Success "Python found: $pythonVersion"
+    }
+    catch {
+        Write-Error "Python is required but not installed"
+        Write-Info "Please install Python 3.10 or higher from https://python.org"
+        exit 1
+    }
+    
+    # Check pip
+    try {
+        $pip = Get-Command pip -ErrorAction Stop
+        Write-Success "pip found"
+    }
+    catch {
+        Write-Error "pip is required but not installed"
+        Write-Info "Please install pip"
+        exit 1
+    }
+    
+    # Check git
+    try {
+        $git = Get-Command git -ErrorAction Stop
+        Write-Success "git found"
+    }
+    catch {
+        Write-Error "git is required but not installed"
+        Write-Info "Please install git from https://git-scm.com"
+        exit 1
+    }
+    
+    Write-Success "All requirements satisfied"
+}
+
+# Create temporary directory
+function New-TempDirectory {
+    $tempDir = Join-Path $env:TEMP "heidi-cli-install-$(Get-Random)"
+    New-Item -ItemType Directory -Path $tempDir -Force | Out-Null
+    Write-Info "Created temporary directory: $tempDir"
+    return $tempDir
+}
+
+# Clone Heidi CLI
+function Invoke-CloneHeidi {
+    param([string]$TempDir)
+    
+    Write-Step "Cloning Heidi CLI from main branch..."
+    
+    $repoUrl = "https://github.com/heidi-dang/heidi-cli.git"
+    $cloneDir = Join-Path $tempDir "heidi-cli"
+    
+    try {
+        & git clone $repoUrl $cloneDir
+        Set-Location $cloneDir
+        Write-Success "Heidi CLI cloned successfully"
+        
+        # Show current commit info
+        $commitHash = & git rev-parse HEAD
+        $commitDate = & git log -1 --format="%cd" --date=short
+        Write-Info "Installing commit: $commitHash ($commitDate)"
+    }
+    catch {
+        Write-Error "Failed to clone Heidi CLI repository"
+        exit 1
+    }
+}
+
+# Setup virtual environment
+function New-VirtualEnvironment {
+    Write-Step "Setting up virtual environment..."
+    
+    try {
+        & python -m venv venv
+        & venv\Scripts\Activate.ps1
+        Write-Success "Virtual environment created and activated"
+        
+        # Upgrade pip
+        & python -m pip install --upgrade pip
+        Write-Success "pip upgraded"
+    }
+    catch {
+        Write-Error "Failed to create virtual environment"
+        exit 1
+    }
+}
+
+# Install dependencies
+function Install-Dependencies {
+    Write-Step "Installing dependencies..."
+    
+    try {
+        & pip install -e .
+        Write-Success "Dependencies installed"
+    }
+    catch {
+        Write-Error "Failed to install Heidi CLI dependencies"
+        exit 1
+    }
+}
+
+# Build Heidi CLI
+function Build-Heidi {
+    Write-Step "Building Heidi CLI..."
+    
+    try {
+        # Check if build tools are available
+        if (Get-Command "python -m build" -ErrorAction SilentlyContinue) {
+            & python -m build
+            Write-Success "Heidi CLI built with modern build system"
+        } else {
+            # Fallback to setup.py
+            & python setup.py build
+            Write-Success "Heidi CLI built with setup.py"
+        }
+    }
+    catch {
+        Write-Error "Failed to build Heidi CLI"
+        exit 1
+    }
+}
+
+# Install Heidi CLI
+function Install-Heidi {
+    Write-Step "Installing Heidi CLI..."
+    
+    try {
+        & pip install -e . --user
+        Write-Success "Heidi CLI installed successfully"
+    }
+    catch {
+        Write-Error "Failed to install Heidi CLI"
+        exit 1
+    }
+}
+
+# Verify installation
+function Test-Installation {
+    Write-Step "Verifying installation..."
+    
+    # Check if heidi command is available
+    try {
+        $heidi = Get-Command heidi -ErrorAction Stop
+        Write-Success "heidi command is available"
+        
+        # Show version
+        try {
+            $version = & heidi --version 2>$null
+            Write-Success "Heidi CLI version: $version"
+        }
+        catch {
+            Write-Warning "Could not get version - this is normal for first run"
+        }
+        
+        # Test basic functionality
+        try {
+            & heidi --help >$null 2>&1
+            Write-Success "Heidi CLI help command works"
+        }
+        catch {
+            Write-Warning "Heidi CLI help command failed - installation may have issues"
+        }
+    }
+    catch {
+        Write-Error "heidi command not found after installation"
+        Write-Info "You may need to restart your terminal or add Python Scripts to PATH"
+        exit 1
+    }
+}
+
+# Run post-install setup
+function Invoke-PostInstallSetup {
+    Write-Step "Running post-install setup..."
+    
+    # Initialize Heidi CLI
+    try {
+        & heidi doctor >$null 2>&1
+        Write-Success "Heidi CLI initialization successful"
+    }
+    catch {
+        Write-Warning "Heidi CLI initialization had issues - this is normal for first run"
+    }
+    
+    # Show next steps
+    Write-Host ""
+    Write-Info "🎉 Installation completed successfully!"
+    Write-Host ""
+    Write-Info "Next steps:"
+    Write-Host "  1. Run: heidi setup"
+    Write-Host "  2. Generate API key: heidi api generate -name 'My Key'"
+    Write-Host "  3. Start model server: heidi model serve"
+    Write-Host "  4. View help: heidi --help"
+    Write-Host ""
+    Write-Info "Documentation: https://github.com/heidi-dang/heidi-cli/blob/main/docs/how-to-use.md"
+}
+
+# Cleanup temporary files
+function Remove-TempFiles {
+    param([string]$TempDir)
+    
+    Write-Step "Cleaning up temporary files..."
+    
+    try {
+        Set-Location $env:USERPROFILE
+        Remove-Item -Path $TempDir -Recurse -Force
+        Write-Success "Temporary files cleaned up"
+    }
+    catch {
+        Write-Warning "Failed to clean up temporary files: $TempDir"
+    }
+}
+
+# Error handling
+function Handle-Error {
+    Write-Error "Installation failed!"
+    Write-Info "Check the error messages above for details"
+    Write-Info "You can try running the installer again"
+    if ($TempDir) {
+        Remove-TempFiles -TempDir $TempDir
+    }
+    exit 1
+}
+
+# Main installation function
+function Main {
+    Write-Header
+    
+    # Set up error handling
+    trap { Handle-Error } ERR
+    
+    # Check if running as Administrator
+    if (-not (Test-Administrator)) {
+        Write-Warning "Not running as Administrator - installing for current user only"
+    }
+    
+    # Run installation steps
+    Test-Requirements
+    $tempDir = New-TempDirectory
+    Invoke-CloneHeidi -TempDir $tempDir
+    New-VirtualEnvironment
+    Install-Dependencies
+    Build-Heidi
+    Install-Heidi
+    Test-Installation
+    Invoke-PostInstallSetup
+    
+    # Cleanup
+    Remove-TempFiles -TempDir $tempDir
+    
+    Write-Success "🎉 Heidi CLI installation completed successfully!"
+    Write-Info "You can now use: heidi --version"
+}
+
+# Run main function
+try {
+    Main
+}
+catch {
+    Handle-Error
+}