setup_alpha_v2.py

"""
Alpha Evolve v2.0 - Quick Setup Script
Automated setup and verification for all components
"""

import os
import sys
import subprocess
from pathlib import Path


def print_header(text):
    """Print formatted header."""
    print("\n" + "=" * 70)
    print(f"  {text}")
    print("=" * 70)


def print_step(step, description):
    """Print step description."""
    print(f"\n[{step}] {description}")


def run_command(cmd, description):
    """Run shell command with output."""
    print(f"  Running: {description}")
    try:
        result = subprocess.run(
            cmd, shell=True, check=True,
            capture_output=True, text=True
        )
        print(f"  ✅ Success")
        return True
    except subprocess.CalledProcessError as e:
        print(f"  ❌ Failed: {e.stderr}")
        return False


def check_python_version():
    """Check Python version."""
    version = sys.version_info
    if version.major >= 3 and version.minor >= 9:
        print(f"  ✅ Python {version.major}.{version.minor}.{version.micro}")
        return True
    else:
        print(f"  ❌ Python {version.major}.{version.minor}.{version.micro} (need 3.9+)")
        return False


def check_command_exists(cmd):
    """Check if command exists."""
    try:
        subprocess.run(
            f"{cmd} --version",
            shell=True, check=True,
            capture_output=True
        )
        return True
    except:
        return False


def main():
    """Main setup function."""
    print_header("🚀 Alpha Evolve v2.0 - Quick Setup")
    
    # Step 1: Check Python version
    print_step(1, "Checking Python version")
    if not check_python_version():
        print("\n⚠️  Python 3.9+ required. Please upgrade Python.")
        return False
    
    # Step 2: Check dependencies
    print_step(2, "Checking system dependencies")
    
    dependencies = {
        "docker": "Docker (optional for deployment)",
        "redis-cli": "Redis (optional for caching)",
        "ollama": "Ollama (optional for LLM features)"
    }
    
    for cmd, desc in dependencies.items():
        if check_command_exists(cmd):
            print(f"  ✅ {desc}")
        else:
            print(f"  ⚠️  {desc} not found (optional)")
    
    # Step 3: Create directories
    print_step(3, "Creating directories")
    
    directories = [
        "data",
        "data/chroma_db",
        "data/validation_test",
        "logs"
    ]
    
    for dir_path in directories:
        Path(dir_path).mkdir(parents=True, exist_ok=True)
        print(f"  ✅ Created: {dir_path}")
    
    # Step 4: Install Python dependencies
    print_step(4, "Installing Python dependencies")
    
    if not run_command(
        f"{sys.executable} -m pip install --upgrade pip",
        "Upgrading pip"
    ):
        print("  ⚠️  Pip upgrade failed, continuing...")
    
    if run_command(
        f"{sys.executable} -m pip install -r requirements_alpha_v2.txt",
        "Installing dependencies from requirements_alpha_v2.txt"
    ):
        print("  ✅ All Python dependencies installed")
    else:
        print("  ❌ Failed to install dependencies")
        return False
    
    # Step 5: Verify installation
    print_step(5, "Verifying installation")
    
    packages = [
        "fastapi",
        "streamlit",
        "optuna",
        "networkx",
        "redis",
        "geopy",
        "phonenumbers"
    ]
    
    all_installed = True
    for package in packages:
        try:
            __import__(package)
            print(f"  ✅ {package}")
        except ImportError:
            print(f"  ❌ {package} not found")
            all_installed = False
    
    if not all_installed:
        print("\n⚠️  Some packages missing. Run: pip install -r requirements_alpha_v2.txt")
    
    # Step 6: Run validation
    print_step(6, "Running system validation")
    
    if run_command(
        f"{sys.executable} validate_system.py",
        "Validating all components"
    ):
        print("  ✅ System validation passed")
    else:
        print("  ⚠️  Validation had warnings (check output above)")
    
    # Final summary
    print_header("📊 Setup Complete")
    
    print("""
✅ Alpha Evolve v2.0 is ready!

Next steps:

1. Start services:
   
   Option A: Docker (recommended)
   $ docker-compose up -d
   
   Option B: Local Python
   Terminal 1: $ python -m uvicorn src.api.fastapi_gateway:app --reload
   Terminal 2: $ streamlit run src/dashboard/app.py

2. Access:
   - API: http://localhost:8000
   - API Docs: http://localhost:8000/docs
   - Dashboard: http://localhost:8501

3. Optional: Install Ollama for LLM features
   $ curl https://ollama.ai/install.sh | sh
   $ ollama pull llama3

4. Run tests:
   $ pytest tests/test_alpha_evolve_v2.py -v

For more information, see README_ALPHA_EVOLVE_V2.md
""")
    
    return True


if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)