"""
Medical Imaging Dataset - Directory Creation Script
====================================================

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 creates the necessary directory structure for the dataset.
"""

import os
from pathlib import Path


def create_directory_structure():
    """
    Create the directory structure for the medical imaging dataset.
    """
    directories = [
        'data/xray/images',
        'data/xray/labels',
        'data/ct_scan/images',
        'data/ct_scan/labels',
        'data/mri/images',
        'data/mri/labels',
    ]
    
    print("Creating directory structure...")
    for directory in directories:
        Path(directory).mkdir(parents=True, exist_ok=True)
        print(f"  [OK] Created: {directory}")
    
    print("\nDirectory structure created successfully!")
    print("\nYou can now place your medical images in:")
    print("  - X-ray images: data/xray/images/")
    print("  - CT scan images: data/ct_scan/images/")
    print("  - MRI images: data/mri/images/")
    print("\nAnd corresponding labels in:")
    print("  - X-ray labels: data/xray/labels/")
    print("  - CT scan labels: data/ct_scan/labels/")
    print("  - MRI labels: data/mri/labels/")


if __name__ == '__main__':
    create_directory_structure()