"""
Medical Imaging Dataset - CT Scan and MRI Downloader
=====================================================

Project: Medical Imaging Dataset
Website: https://rskworld.in
Contact: help@rskworld.in, support@rskworld.in
Phone: +91 93305 39277
Founder: Molla Samser
Designer & Tester: Rima Khatun

This script downloads real CT scan and MRI datasets from public sources.
"""

import os
import requests
import zipfile
import shutil
import json
from pathlib import Path


def download_brain_mri_kaggle():
    """Instructions for downloading Brain MRI dataset from Kaggle."""
    print("=" * 60)
    print("Brain MRI Images for Brain Tumor Detection")
    print("=" * 60)
    print()
    print("Dataset: Brain MRI Images for Brain Tumor Detection")
    print("Source: Kaggle")
    print("URL: https://www.kaggle.com/datasets/navoneel/brain-mri-images-for-brain-tumor-detection")
    print("Size: ~300 MB")
    print("Images: ~3,000+ brain MRI images")
    print("License: CC BY 4.0")
    print()
    print("To download:")
    print("1. Install Kaggle API: pip install kaggle")
    print("2. Setup Kaggle credentials (see DOWNLOAD_INSTRUCTIONS.md)")
    print("3. Run: kaggle datasets download -d navoneel/brain-mri-images-for-brain-tumor-detection")
    print("4. Extract and organize to data/mri/images/")
    print()


def download_medical_mnist():
    """Instructions for downloading Medical MNIST (includes CT and MRI)."""
    print("=" * 60)
    print("Medical MNIST Dataset")
    print("=" * 60)
    print()
    print("Dataset: Medical MNIST")
    print("Source: Kaggle / Official Website")
    print("URL: https://www.kaggle.com/datasets/andrewmvd/medical-mnist")
    print("Official: https://medmnist.com/")
    print("Size: ~200 MB")
    print("Images: ~66,000 images across multiple modalities")
    print("Includes: CT, MRI, X-ray, Ultrasound, etc.")
    print("License: CC BY 4.0")
    print()
    print("To download:")
    print("Option 1 - Kaggle:")
    print("  kaggle datasets download -d andrewmvd/medical-mnist")
    print()
    print("Option 2 - Official Website:")
    print("  1. Visit: https://medmnist.com/")
    print("  2. Click 'Download' on desired dataset")
    print("  3. Extract and organize by modality")
    print()


def download_from_tcia():
    """Instructions for The Cancer Imaging Archive (TCIA)."""
    print("=" * 60)
    print("The Cancer Imaging Archive (TCIA)")
    print("=" * 60)
    print()
    print("Source: TCIA")
    print("URL: https://www.cancerimagingarchive.net/")
    print("Description: Open-access database of medical images for cancer research")
    print("Modalities: CT, MRI, PET scans")
    print("Access: Free registration required")
    print()
    print("To download:")
    print("1. Register at: https://www.cancerimagingarchive.net/")
    print("2. Browse collections by cancer type")
    print("3. Download via TCIA Data Portal or NBIA")
    print("4. Organize CT scans to data/ct_scan/images/")
    print("5. Organize MRI to data/mri/images/")
    print()


def download_from_radiopaedia():
    """Instructions for Radiopaedia."""
    print("=" * 60)
    print("Radiopaedia")
    print("=" * 60)
    print()
    print("Source: Radiopaedia")
    print("URL: https://radiopaedia.org/")
    print("Description: Educational radiology resource with case studies")
    print("Modalities: X-ray, CT, MRI, Ultrasound")
    print("Access: Free, educational use")
    print("License: CC BY-NC-SA")
    print()
    print("To download:")
    print("1. Visit: https://radiopaedia.org/")
    print("2. Browse cases by modality (CT, MRI)")
    print("3. Download images for educational purposes")
    print("4. Organize to appropriate directories")
    print()


def download_from_opennuro():
    """Instructions for OpenNeuro (MRI datasets)."""
    print("=" * 60)
    print("OpenNeuro")
    print("=" * 60)
    print()
    print("Source: OpenNeuro")
    print("URL: https://openneuro.org/")
    print("Description: Open-science database for brain imaging")
    print("Modalities: MRI, fMRI, EEG")
    print("Access: Free, open access")
    print()
    print("To download:")
    print("1. Visit: https://openneuro.org/")
    print("2. Browse datasets")
    print("3. Download via web interface or API")
    print("4. Data in BIDS format")
    print("5. Extract MRI images to data/mri/images/")
    print()


def create_kaggle_download_script():
    """Create a script to download CT and MRI datasets from Kaggle."""
    script_content = """@echo off
REM Medical Imaging Dataset - CT and MRI Kaggle Downloader
REM ========================================================
REM 
REM Project: Medical Imaging Dataset
REM Website: https://rskworld.in
REM Contact: help@rskworld.in, support@rskworld.in
REM Phone: +91 93305 39277
REM Founder: Molla Samser
REM Designer & Tester: Rima Khatun

echo ========================================
echo CT Scan and MRI Dataset Downloader
echo ========================================
echo.

REM Check if kaggle is installed
python -c "import kaggle" 2>nul
if errorlevel 1 (
    echo Installing Kaggle API...
    pip install kaggle
)

echo.
echo Available datasets:
echo 1. Brain MRI Images for Brain Tumor Detection
echo 2. Medical MNIST (includes CT and MRI)
echo.

set /p choice="Enter dataset number (1-2): "

if "%choice%"=="1" (
    echo Downloading Brain MRI Images...
    kaggle datasets download -d navoneel/brain-mri-images-for-brain-tumor-detection
    echo Extracting...
    python -m zipfile -e brain-mri-images-for-brain-tumor-detection.zip data/temp/mri/
    echo Organizing MRI images...
    python organize_mri_data.py
    echo Dataset downloaded and organized!
)

if "%choice%"=="2" (
    echo Downloading Medical MNIST...
    kaggle datasets download -d andrewmvd/medical-mnist
    echo Extracting...
    python -m zipfile -e medical-mnist.zip data/temp/medmnist/
    echo Organizing by modality...
    python organize_medmnist.py
    echo Dataset downloaded and organized!
)

echo.
echo Download complete!
pause
"""
    
    with open('download_ct_mri_kaggle.bat', 'w') as f:
        f.write(script_content)
    
    print("Created: download_ct_mri_kaggle.bat")


def create_organize_scripts():
    """Create scripts to organize downloaded CT and MRI data."""
    
    # MRI organizer script
    mri_script = """\"\"\"
Organize Brain MRI Images
\"\"\"
import shutil
from pathlib import Path

source = Path("data/temp/mri")
target = Path("data/mri/images")
target.mkdir(parents=True, exist_ok=True)

# Find all image files
image_extensions = ['.png', '.jpg', '.jpeg', '.dcm']
count = 0

for ext in image_extensions:
    for img_file in source.rglob(f'*{ext}'):
        if img_file.is_file():
            dest = target / img_file.name
            if not dest.exists():
                shutil.copy2(img_file, dest)
                count += 1

print(f"Organized {count} MRI images")
"""
    
    with open('organize_mri_data.py', 'w') as f:
        f.write(mri_script)
    
    # MedMNIST organizer script
    medmnist_script = """\"\"\"
Organize Medical MNIST by Modality
\"\"\"
import shutil
from pathlib import Path

source = Path("data/temp/medmnist")
ct_target = Path("data/ct_scan/images")
mri_target = Path("data/mri/images")

ct_target.mkdir(parents=True, exist_ok=True)
mri_target.mkdir(parents=True, exist_ok=True)

# MedMNIST structure may vary, adjust based on actual structure
# This is a template - adjust based on actual dataset structure

image_extensions = ['.png', '.jpg', '.jpeg']
ct_count = 0
mri_count = 0

for ext in image_extensions:
    for img_file in source.rglob(f'*{ext}'):
        if img_file.is_file():
            name_lower = img_file.name.lower()
            # Adjust these conditions based on actual file naming
            if 'ct' in name_lower or 'ctscan' in name_lower:
                dest = ct_target / img_file.name
                if not dest.exists():
                    shutil.copy2(img_file, dest)
                    ct_count += 1
            elif 'mri' in name_lower or 'brain' in name_lower:
                dest = mri_target / img_file.name
                if not dest.exists():
                    shutil.copy2(img_file, dest)
                    mri_count += 1

print(f"Organized {ct_count} CT scan images")
print(f"Organized {mri_count} MRI images")
"""
    
    with open('organize_medmnist.py', 'w') as f:
        f.write(medmnist_script)
    
    print("Created: organize_mri_data.py")
    print("Created: organize_medmnist.py")


def main():
    """Main function."""
    print("=" * 60)
    print("CT Scan and MRI Dataset Download Guide")
    print("=" * 60)
    print()
    print("Current status:")
    print("  - CT scan images: 5 (sample data)")
    print("  - MRI images: 5 (sample data)")
    print()
    print("To get REAL CT scan and MRI images:")
    print()
    
    # Show all options
    download_brain_mri_kaggle()
    download_medical_mnist()
    download_from_tcia()
    download_from_radiopaedia()
    download_from_opennuro()
    
    # Create helper scripts
    print("=" * 60)
    print("Creating helper scripts...")
    print("=" * 60)
    create_kaggle_download_script()
    create_organize_scripts()
    
    print()
    print("=" * 60)
    print("Summary")
    print("=" * 60)
    print()
    print("EASIEST OPTIONS:")
    print("1. Brain MRI (Kaggle): ~3,000+ images")
    print("   - Requires Kaggle account")
    print("   - Run: kaggle datasets download -d navoneel/brain-mri-images-for-brain-tumor-detection")
    print()
    print("2. Medical MNIST (Kaggle/Website): ~66,000 images")
    print("   - Includes CT, MRI, and other modalities")
    print("   - Visit: https://medmnist.com/")
    print()
    print("3. TCIA (Registration required)")
    print("   - Visit: https://www.cancerimagingarchive.net/")
    print("   - Large collection of CT and MRI scans")
    print()
    print("For detailed instructions, see: DOWNLOAD_INSTRUCTIONS.md")
    print()


if __name__ == '__main__':
    main()