def create_yaml_file(dataset_path):
    """
    Creates a dataset configuration YAML file for YOLO training.

    Parameters:
    dataset_path (str): Path to the dataset.
    """
    datasets_yaml = '''
    path: /Dataset/YOLO

    train: train/images
    val: val/images
    test: test/images

    # number of classes
    nc: 1

    # class names
    names: ['license_plate']
    '''

    # Write the content to the datasets.yaml file
    with open(os.path.join(dataset_path,'datasets.yaml'), 'w') as file:
        file.write(datasets_yaml)
    
    print("datasets.yaml file created.")

if __name__ == "__main__":
    create_yaml_file('Dataset')