#!/usr/bin/env python3
"""
Simple Enhanced Training Test

Quick test to verify enhanced training system can be enabled and controlled.
"""

import sys
import os
import logging

# Add project root to path
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from core.orchestrator import TradingOrchestrator
from core.data_provider import DataProvider

# Configure logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def test_enhanced_training():
    """Test enhanced training system"""
    try:
        logger.info("Testing Enhanced Training System...")
        
        # 1. Create data provider
        data_provider = DataProvider()
        
        # 2. Create orchestrator with enhanced training ENABLED
        logger.info("Creating orchestrator with enhanced_rl_training=True...")
        orchestrator = TradingOrchestrator(
            data_provider=data_provider,
            enhanced_rl_training=True  # 🔥 THIS ENABLES IT
        )
        
        # 3. Check if training system is available
        logger.info(f"Training system available: {orchestrator.enhanced_training_system is not None}")
        logger.info(f"Training enabled: {orchestrator.training_enabled}")
        
        # 4. Get training stats
        stats = orchestrator.get_enhanced_training_stats()
        logger.info(f"Training stats: {stats}")
        
        # 5. Test start/stop
        if orchestrator.enhanced_training_system:
            logger.info("Testing start/stop functionality...")
            
            # Start training
            start_result = orchestrator.start_enhanced_training()
            logger.info(f"Start result: {start_result}")
            
            # Get updated stats
            updated_stats = orchestrator.get_enhanced_training_stats()
            logger.info(f"Updated stats: {updated_stats}")
            
            # Stop training
            stop_result = orchestrator.stop_enhanced_training()
            logger.info(f"Stop result: {stop_result}")
            
            logger.info("✅ Enhanced training system is working!")
            return True
        else:
            logger.warning("❌ Enhanced training system not available")
            return False
            
    except Exception as e:
        logger.error(f"Error testing enhanced training: {e}")
        return False

if __name__ == "__main__":
    success = test_enhanced_training()
    if success:
        print("\n🎉 Enhanced training system is ready to use!")
        print("To enable it in your main system, use:")
        print("   enhanced_rl_training=True when creating TradingOrchestrator")
    else:
        print("\n⚠️  Enhanced training system has issues. Check the logs above.")