#!/usr/bin/env python3
"""
Quick Start 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: Quick start script to test the dataset and tools
License: Educational Purpose Only
"""

import os
import sys

def print_header():
    """Print project header"""
    print("="*70)
    print("  Music Classification Dataset - Quick Start")
    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 check_dependencies():
    """Check if required packages are installed"""
    print("[1/5] Checking dependencies...")
    
    required_packages = [
        'numpy',
        'pandas',
        'librosa',
        'scikit-learn',
        'matplotlib',
        'seaborn'
    ]
    
    missing_packages = []
    
    for package in required_packages:
        try:
            __import__(package)
            print(f"  [OK] {package}")
        except ImportError:
            print(f"  [MISSING] {package}")
            missing_packages.append(package)
    
    if missing_packages:
        print(f"\n[WARNING] Missing packages: {', '.join(missing_packages)}")
        print("Please run: pip install -r requirements.txt")
        return False
    else:
        print("\n[SUCCESS] All dependencies installed!\n")
        return True

def check_directory_structure():
    """Check if directory structure is correct"""
    print("[2/5] Checking directory structure...")
    
    required_dirs = [
        'data',
        'data/audio',
        'models',
        'models/saved_models',
        'utils',
        'notebooks'
    ]
    
    all_exist = True
    for dir_path in required_dirs:
        if os.path.exists(dir_path):
            print(f"  [OK] {dir_path}/")
        else:
            print(f"  [MISSING] {dir_path}/")
            all_exist = False
    
    if all_exist:
        print("\n[SUCCESS] Directory structure is correct!\n")
    else:
        print("\n[WARNING] Some directories are missing.\n")
    
    return all_exist

def check_data_files():
    """Check if data files exist"""
    print("[3/5] Checking data files...")
    
    data_files = [
        'data/train_data.csv',
        'data/test_data.csv',
        'data/features.csv'
    ]
    
    all_exist = True
    for file_path in data_files:
        if os.path.exists(file_path):
            print(f"  [OK] {file_path}")
        else:
            print(f"  [MISSING] {file_path}")
            all_exist = False
    
    if all_exist:
        print("\n[SUCCESS] All data files present!\n")
    else:
        print("\n[WARNING] Some data files are missing.\n")
    
    return all_exist

def test_basic_functionality():
    """Test basic functionality"""
    print("[4/5] Testing basic functionality...")
    
    try:
        import pandas as pd
        from utils.audio_processor import AudioProcessor
        from utils.feature_extractor import FeatureExtractor
        
        # Test loading data
        train_df = pd.read_csv('data/train_data.csv')
        print(f"  [OK] Loaded training data: {len(train_df)} samples")
        
        # Test initializing processors
        audio_proc = AudioProcessor()
        feature_ext = FeatureExtractor()
        print("  [OK] Initialized audio processor")
        print("  [OK] Initialized feature extractor")
        
        print("\n[SUCCESS] Basic functionality test passed!\n")
        return True
        
    except Exception as e:
        print(f"\n[ERROR] Error during testing: {str(e)}\n")
        return False

def display_next_steps():
    """Display next steps for the user"""
    print("[5/5] Next Steps")
    print("="*70)
    
    print("\n[GETTING STARTED]\n")
    
    print("1. Explore the dataset:")
    print("   jupyter notebook notebooks/exploratory_analysis.ipynb")
    
    print("\n2. Train a model:")
    print("   python models/train_model.py --model random_forest")
    
    print("\n3. Make predictions:")
    print("   python models/predict.py --audio path/to/your/music.wav")
    
    print("\n4. Read the documentation:")
    print("   - README.md - Project overview")
    print("   - USAGE_GUIDE.md - Detailed usage instructions")
    print("   - DATASET_INFO.md - Dataset information")
    
    print("\n[NEED HELP?]")
    print("   Email: help@rskworld.in")
    print("   Phone: +91 93305 39277")
    print("   Website: https://rskworld.in")
    
    print("\n" + "="*70)
    print("© 2026 RSK World - Founded by Molla Samser")
    print("Designer & Tester: Rima Khatun")
    print("="*70)

def main():
    """Main function"""
    print_header()
    
    # Run checks
    deps_ok = check_dependencies()
    dirs_ok = check_directory_structure()
    data_ok = check_data_files()
    
    if deps_ok and dirs_ok and data_ok:
        test_ok = test_basic_functionality()
        
        if test_ok:
            print("[SUCCESS] Setup complete! You're ready to start using the dataset.\n")
        else:
            print("[WARNING] Some tests failed. Please check the error messages above.\n")
    else:
        print("[WARNING] Setup incomplete. Please install dependencies and ensure")
        print("          all required files and directories are present.\n")
    
    display_next_steps()

if __name__ == "__main__":
    main()