#!/usr/bin/env python3
"""
Setup script for Improved Unified Multi-Model PT v2.0.0
"""

import subprocess
import sys
import os

def install_requirements():
    """Install required packages"""
    print("📦 Installing required packages...")
    
    try:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
        print("✅ Requirements installed successfully!")
    except subprocess.CalledProcessError as e:
        print(f"❌ Error installing requirements: {e}")
        return False
    
    return True

def test_installation():
    """Test the installation"""
    print("🧪 Testing installation...")
    
    try:
        from improved_unified_model_pt import ImprovedUnifiedMultiModelPT, ImprovedUnifiedModelConfig
        print("✅ Model classes imported successfully!")
        
        # Test basic functionality
        config = ImprovedUnifiedModelConfig()
        print("✅ Configuration created successfully!")
        
        return True
    except ImportError as e:
        print(f"❌ Import error: {e}")
        return False

def main():
    """Main setup function"""
    print("🚀 Setting up Improved Unified Multi-Model PT v2.0.0")
    print("=" * 60)
    
    # Install requirements
    if not install_requirements():
        print("❌ Setup failed during requirements installation")
        return False
    
    # Test installation
    if not test_installation():
        print("❌ Setup failed during testing")
        return False
    
    print("🎉 Setup completed successfully!")
    print("📖 Check README.md for usage examples")
    
    return True

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