#!/usr/bin/env python3
"""
Setup Script for Voice Cloning Dataset
Developer: Molla Samser
Email: help@rskworld.in
Phone: +91 93305 39277
Website: https://rskworld.in
Year: 2026
"""

import os
import sys
import subprocess
from pathlib import Path


def check_python_version():
    """Check if Python version is 3.8 or higher."""
    if sys.version_info < (3, 8):
        print("Error: Python 3.8 or higher is required")
        print(f"Current version: {sys.version}")
        return False
    print(f"✓ Python version: {sys.version.split()[0]}")
    return True


def install_requirements():
    """Install required Python packages."""
    print("\nInstalling required packages...")
    try:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"])
        print("✓ Requirements installed successfully")
        return True
    except subprocess.CalledProcessError:
        print("✗ Error installing requirements")
        return False


def create_directories():
    """Create necessary directories if they don't exist."""
    directories = [
        "audio",
        "processed",
        "features",
        "prepared",
        "config",
        "scripts"
    ]
    
    print("\nCreating directories...")
    for directory in directories:
        Path(directory).mkdir(parents=True, exist_ok=True)
        print(f"✓ {directory}/")
    
    return True


def verify_structure():
    """Verify project structure is correct."""
    print("\nVerifying project structure...")
    
    required_files = [
        "README.md",
        "requirements.txt",
        "config/dataset_config.json",
        "scripts/process_audio.py",
        "scripts/extract_features.py",
        "scripts/prepare_dataset.py"
    ]
    
    all_present = True
    for file_path in required_files:
        if Path(file_path).exists():
            print(f"✓ {file_path}")
        else:
            print(f"✗ {file_path} (missing)")
            all_present = False
    
    return all_present


def main():
    """Main setup function."""
    print("=" * 60)
    print("Voice Cloning Dataset - Setup")
    print("Developer: Molla Samser")
    print("Website: https://rskworld.in")
    print("=" * 60)
    
    # Check Python version
    if not check_python_version():
        sys.exit(1)
    
    # Create directories
    if not create_directories():
        sys.exit(1)
    
    # Verify structure
    if not verify_structure():
        print("\nWarning: Some files are missing. Please check the project structure.")
    
    # Install requirements
    install_requirements()
    
    print("\n" + "=" * 60)
    print("Setup complete!")
    print("=" * 60)
    print("\nNext steps:")
    print("1. Add your audio files to the audio/ directory")
    print("2. Run: python scripts/process_audio.py --input audio/ --output processed/")
    print("3. Run: python scripts/extract_features.py --audio audio/ --output features/")
    print("4. Run: python scripts/prepare_dataset.py --config config/dataset_config.json")
    print("\nFor more information, see README.md")


if __name__ == '__main__':
    main()