'''
python uploading_files_to_hugging_face.py --folder_directory /path/to/your/folder --repository MrGiovanni/AbdomenAtlasInternal --repository_type dataset
'''

import huggingface_hub
import argparse

def main(args):
    huggingface_hub.upload_folder(
        folder_path=args.folder_directory,
        repo_id=args.repository,
        repo_type=args.repository_type,
        multi_commits=True,
        multi_commits_verbose=True
        )

if __name__ == "__main__":
    
    parser = argparse.ArgumentParser()
    parser.add_argument('--folder_directory', 
                        dest='folder_directory', 
                        type=str, 
                        default='/Users/zongwei.zhou/Desktop/AbdomenAtlasInternal',
                        help='the directory of the AbdomenAtlas dataset',
                       )
    parser.add_argument('--repository', 
                        dest='repository', 
                        type=str, 
                        default='MrGiovanni/AbdomenAtlasInternal',
                        help='repository directory',
                       )
    parser.add_argument('--repository_type',
                        dest='repository_type',
                        type=str,
                        default='dataset',
                        help='the type of repository',
                       )
    args = parser.parse_args()
    
    main(args)