"""
==================================================================================
    Action Recognition Dataset - Video Converter
==================================================================================
    Project: Action Recognition Dataset
    Category: Video Data / Data Science
    
    This script converts videos to browser-compatible H.264 format.
    
==================================================================================
    DEVELOPER INFORMATION
==================================================================================
    Website: RSK World (https://rskworld.in)
    Founded by: Molla Samser
    Designer & Tester: Rima Khatun
    
    Contact Information:
    - Email: help@rskworld.in
    - Email: support@rskworld.in
    - Phone: +91 93305 39277
    
==================================================================================
    COPYRIGHT
==================================================================================
    (C) 2026 RSK World. All Rights Reserved.
    
==================================================================================

Usage:
    python convert_videos.py
"""

import os
import subprocess
import sys
from pathlib import Path

def check_ffmpeg():
    """Check if FFmpeg is available"""
    try:
        subprocess.run(['ffmpeg', '-version'], capture_output=True, check=True)
        return True
    except (subprocess.CalledProcessError, FileNotFoundError):
        return False

def convert_video_ffmpeg(input_path, output_path):
    """Convert video using FFmpeg"""
    cmd = [
        'ffmpeg', '-y',
        '-i', str(input_path),
        '-c:v', 'libx264',
        '-preset', 'fast',
        '-crf', '23',
        '-c:a', 'aac',
        '-movflags', '+faststart',
        str(output_path)
    ]
    subprocess.run(cmd, capture_output=True, check=True)

def convert_all_videos():
    """Convert all videos in sample_data to browser-compatible format"""
    sample_dir = Path('sample_data')
    
    if not sample_dir.exists():
        print("No sample_data directory found!")
        return
    
    videos = list(sample_dir.glob('**/*.mp4'))
    print(f"Found {len(videos)} videos to convert")
    
    for i, video_path in enumerate(videos, 1):
        print(f"[{i}/{len(videos)}] Converting: {video_path}")
        
        # Create temporary output path
        temp_path = video_path.with_suffix('.temp.mp4')
        
        try:
            convert_video_ffmpeg(video_path, temp_path)
            
            # Replace original with converted
            os.remove(video_path)
            os.rename(temp_path, video_path)
            
            print(f"  [OK] Converted successfully")
        except Exception as e:
            print(f"  [ERROR] Failed to convert: {e}")
            if temp_path.exists():
                os.remove(temp_path)
    
    print("\nConversion complete!")

if __name__ == '__main__':
    if not check_ffmpeg():
        print("FFmpeg not found! Installing via pip...")
        print("Alternatively, download FFmpeg from: https://ffmpeg.org/download.html")
        print("\nTrying imageio-ffmpeg as fallback...")
        
        try:
            import imageio_ffmpeg
            print("imageio-ffmpeg is available")
        except ImportError:
            print("Installing imageio-ffmpeg...")
            subprocess.run([sys.executable, '-m', 'pip', 'install', 'imageio-ffmpeg'], check=True)
    
    convert_all_videos()