# -*- coding:utf-8 -*-
from image_sdk.gettoken import get_token
from image_sdk.utils import encode_to_base64
from image_sdk.celebrity_recognition import celebrity_recognition
from image_sdk.utils import init_global_env

if __name__ == '__main__':
    # Services currently support North China-Beijing(cn-north-4), CN-Hong Kong(ap-southeast-1)
    init_global_env('cn-north-4')

    #
    # access image tagging ， post data by token
    #
    user_name = '******'
    password = '******'
    account_name = '******'  # the same as user_name in commonly use

    # The OBS link should match the region, and the OBS resources of different regions are not shared
    demo_data_url = 'https://sdk-obs-source-save.obs.cn-north-4.myhuaweicloud.com/celebrity-recognition.jpg'
    token = get_token(user_name, password, account_name)

    # call interface use the url
    result = celebrity_recognition(token, '', demo_data_url, 0.48)
    print(result)

    # call interface use the file
    result = celebrity_recognition(token, encode_to_base64('data/celebrity-recognition.jpg'), '', 0.48)
    print(result)