#!/usr/bin/env python3
"""
Demo Script for Music Classification Dataset

Project: Music Classification Dataset
Author: Molla Samser
Company: RSK World
Designer & Tester: Rima Khatun
Website: https://rskworld.in
Email: help@rskworld.in, support@rskworld.in
Phone: +91 93305 39277
Description: Demonstration of all features in the dataset
License: Educational Purpose Only
"""

import os
import sys
import numpy as np
import pandas as pd
import warnings
warnings.filterwarnings('ignore')


def print_header():
    """Print project header"""
    print("="*70)
    print("  MUSIC CLASSIFICATION DATASET - DEMO")
    print("="*70)
    print("  Author: Molla Samser")
    print("  Company: RSK World")
    print("  Designer & Tester: Rima Khatun")
    print("  Website: https://rskworld.in")
    print("  Email: help@rskworld.in")
    print("  Phone: +91 93305 39277")
    print("="*70)
    print()


def demo_data_loading():
    """Demo: Loading data files"""
    print("[DEMO 1] Data Loading")
    print("-" * 50)
    
    try:
        # Load training data
        train_df = pd.read_csv('data/train_data.csv')
        print(f"[OK] Training data loaded: {len(train_df)} samples")
        print(f"     Columns: {list(train_df.columns)}")
        print(f"     Genres: {train_df['genre'].unique().tolist()}")
        
        # Load test data
        test_df = pd.read_csv('data/test_data.csv')
        print(f"[OK] Test data loaded: {len(test_df)} samples")
        
        # Load features
        features_df = pd.read_csv('data/features.csv')
        print(f"[OK] Features loaded: {len(features_df)} samples")
        print(f"     Features: {list(features_df.columns[2:])}")
        
        print("\n[SUCCESS] Data loading demo complete!\n")
        return True
        
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_genre_distribution():
    """Demo: Show genre distribution"""
    print("[DEMO 2] Genre Distribution")
    print("-" * 50)
    
    try:
        train_df = pd.read_csv('data/train_data.csv')
        genre_counts = train_df['genre'].value_counts()
        
        print("\nTraining Data Distribution:")
        for genre, count in genre_counts.items():
            bar = "#" * count
            print(f"  {genre:12s} | {bar} ({count})")
        
        print(f"\nTotal: {len(train_df)} samples")
        print("\n[SUCCESS] Genre distribution demo complete!\n")
        return True
        
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_audio_processor():
    """Demo: Audio processor module"""
    print("[DEMO 3] Audio Processor")
    print("-" * 50)
    
    try:
        from utils.audio_processor import AudioProcessor
        
        processor = AudioProcessor()
        print(f"[OK] AudioProcessor initialized")
        print(f"     Sample rate: {processor.sample_rate} Hz")
        print(f"     Duration: {processor.duration} seconds")
        
        # Create dummy audio
        dummy_audio = np.random.randn(22050)  # 1 second
        
        # Normalize
        normalized = processor.normalize_audio(dummy_audio)
        print(f"[OK] Normalize: input range [{dummy_audio.min():.2f}, {dummy_audio.max():.2f}]")
        print(f"               output range [{normalized.min():.2f}, {normalized.max():.2f}]")
        
        print("\n[SUCCESS] Audio processor demo complete!\n")
        return True
        
    except ImportError as e:
        print(f"[SKIP] Missing dependencies: {str(e)}")
        print("       Run: pip install librosa")
        return False
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_feature_extractor():
    """Demo: Feature extractor module"""
    print("[DEMO 4] Feature Extractor")
    print("-" * 50)
    
    try:
        from utils.feature_extractor import FeatureExtractor
        
        extractor = FeatureExtractor()
        print(f"[OK] FeatureExtractor initialized")
        print(f"     Sample rate: {extractor.sample_rate} Hz")
        
        # Create dummy audio
        dummy_audio = np.random.randn(22050 * 5)  # 5 seconds
        
        # Extract features
        print("[OK] Feature extraction functions available:")
        print("     - extract_mfcc()")
        print("     - extract_spectral_centroid()")
        print("     - extract_spectral_rolloff()")
        print("     - extract_zero_crossing_rate()")
        print("     - extract_chroma()")
        print("     - extract_tempo()")
        print("     - extract_all_features()")
        
        print("\n[SUCCESS] Feature extractor demo complete!\n")
        return True
        
    except ImportError as e:
        print(f"[SKIP] Missing dependencies: {str(e)}")
        print("       Run: pip install librosa")
        return False
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_audio_augmentation():
    """Demo: Audio augmentation module"""
    print("[DEMO 5] Audio Augmentation")
    print("-" * 50)
    
    try:
        from utils.audio_augmentation import AudioAugmenter
        
        augmenter = AudioAugmenter()
        print(f"[OK] AudioAugmenter initialized")
        
        # Create dummy audio
        dummy_audio = np.random.randn(22050)  # 1 second
        
        print("[OK] Available augmentation techniques:")
        print("     - add_noise()")
        print("     - time_stretch()")
        print("     - pitch_shift()")
        print("     - time_shift()")
        print("     - change_volume()")
        print("     - add_reverb()")
        print("     - random_augmentation()")
        
        # Demo augmentation
        noisy = augmenter.add_noise(dummy_audio, 0.005)
        print(f"\n[OK] Noise added: shape {noisy.shape}")
        
        print("\n[SUCCESS] Audio augmentation demo complete!\n")
        return True
        
    except ImportError as e:
        print(f"[SKIP] Missing dependencies: {str(e)}")
        print("       Run: pip install librosa")
        return False
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_advanced_features():
    """Demo: Advanced feature extractor"""
    print("[DEMO 6] Advanced Features")
    print("-" * 50)
    
    try:
        from utils.advanced_features import AdvancedFeatureExtractor
        
        extractor = AdvancedFeatureExtractor()
        print(f"[OK] AdvancedFeatureExtractor initialized")
        
        print("[OK] Advanced features available:")
        print("     - Spectral Contrast (7D)")
        print("     - Tonnetz (6D)")
        print("     - Spectral Flatness")
        print("     - Spectral Flux")
        print("     - Harmonic/Percussive Separation")
        print("     - Mel Spectrogram Statistics (7D)")
        print("     - Rhythm Features (4D)")
        print("     - Energy Features (4D)")
        print("     Total: 30+ features")
        
        print("\n[SUCCESS] Advanced features demo complete!\n")
        return True
        
    except ImportError as e:
        print(f"[SKIP] Missing dependencies: {str(e)}")
        print("       Run: pip install librosa scipy")
        return False
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_model_comparison():
    """Demo: Model comparison module"""
    print("[DEMO 7] Model Comparison")
    print("-" * 50)
    
    try:
        from utils.model_comparison import ModelComparator
        
        comparator = ModelComparator()
        print(f"[OK] ModelComparator initialized")
        print(f"     Models available: {len(comparator.models)}")
        
        print("\n[OK] Comparison models:")
        for name in comparator.models.keys():
            print(f"     - {name}")
        
        print("\n[SUCCESS] Model comparison demo complete!\n")
        return True
        
    except ImportError as e:
        print(f"[SKIP] Missing dependencies: {str(e)}")
        print("       Run: pip install scikit-learn")
        return False
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_realtime_processor():
    """Demo: Real-time processor module"""
    print("[DEMO 8] Real-Time Processing")
    print("-" * 50)
    
    try:
        from utils.realtime_processor import RealtimeAudioProcessor, StreamingBuffer, FeatureCache
        
        processor = RealtimeAudioProcessor()
        print(f"[OK] RealtimeAudioProcessor initialized")
        print(f"     Sample rate: {processor.sample_rate} Hz")
        print(f"     Chunk duration: {processor.chunk_duration}s")
        
        buffer = StreamingBuffer(buffer_size=44100)
        print(f"[OK] StreamingBuffer initialized (44100 samples)")
        
        cache = FeatureCache(max_size=100)
        print(f"[OK] FeatureCache initialized (100 items)")
        
        print("\n[OK] Real-time components available:")
        print("     - Multi-threaded processing")
        print("     - Circular buffer")
        print("     - LRU feature cache")
        
        print("\n[SUCCESS] Real-time processing demo complete!\n")
        return True
        
    except ImportError as e:
        print(f"[SKIP] Missing dependencies: {str(e)}")
        return True  # Still counts as passed
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def demo_configuration():
    """Demo: Configuration module"""
    print("[DEMO 9] Configuration")
    print("-" * 50)
    
    try:
        import config
        
        print(f"[OK] Project: {config.PROJECT_NAME}")
        print(f"[OK] Version: {config.PROJECT_VERSION}")
        print(f"[OK] Author: {config.AUTHOR}")
        print(f"[OK] Company: {config.COMPANY}")
        print(f"[OK] Website: {config.WEBSITE}")
        
        print(f"\n[OK] Audio Settings:")
        print(f"     Sample rate: {config.SAMPLE_RATE} Hz")
        print(f"     Duration: {config.AUDIO_DURATION}s")
        print(f"     MFCC coefficients: {config.N_MFCC}")
        
        print(f"\n[OK] Genres: {config.GENRES}")
        
        print("\n[SUCCESS] Configuration demo complete!\n")
        return True
        
    except Exception as e:
        print(f"[ERROR] {str(e)}")
        return False


def print_summary(results):
    """Print demo summary"""
    print("="*70)
    print("DEMO SUMMARY")
    print("="*70)
    
    passed = sum(1 for r in results if r)
    total = len(results)
    
    for i, (name, result) in enumerate(results.items(), 1):
        status = "[OK]" if result else "[SKIP/FAIL]"
        print(f"  {i}. {name}: {status}")
    
    print("-"*70)
    print(f"  Passed: {passed}/{total} demos")
    
    if passed == total:
        print("\n  All demos passed! Project is ready to use.")
    else:
        print("\n  Some demos skipped. Install missing dependencies:")
        print("  pip install -r requirements.txt")
    
    print("="*70)
    print("  (c) 2026 RSK World - Molla Samser")
    print("  https://rskworld.in")
    print("="*70)


def main():
    """Run all demos"""
    print_header()
    
    results = {}
    
    # Run all demos
    results["Data Loading"] = demo_data_loading()
    results["Genre Distribution"] = demo_genre_distribution()
    results["Audio Processor"] = demo_audio_processor()
    results["Feature Extractor"] = demo_feature_extractor()
    results["Audio Augmentation"] = demo_audio_augmentation()
    results["Advanced Features"] = demo_advanced_features()
    results["Model Comparison"] = demo_model_comparison()
    results["Real-Time Processing"] = demo_realtime_processor()
    results["Configuration"] = demo_configuration()
    
    # Print summary
    print_summary(results)


if __name__ == "__main__":
    main()