using-leiapix-storage.py

import sys
import os
import uuid
import requests

# print("\n".join(sys.path))
# Reading configuration from environment
if sys.prefix == sys.base_prefix:
    print('Error. You should run application in virtualenv to not pollute global dependencies. Please refer to '
          'README.md for instructions on how to set it up', file=sys.stderr)
    print(f'sys.prefix: {sys.prefix}')
    print(f'sys.base_prefix: {sys.base_prefix}')
    exit(1)

# Get the environment variables
CLIENT_ID = os.getenv('BACKEND_CLIENT_ID')
CLIENT_SECRET = os.getenv('BACKEND_CLIENT_SECRET')

# Check that they are all set
if not CLIENT_ID or not CLIENT_SECRET:
    print('Error. In order to authenticate against Leia Media Cloud API, you need to provide BACKEND_CLIENT_ID and '
          'BACKEND_CLIENT_SECRET env vars', file=sys.stderr)
    exit(1)

MEDIA_CLOUD_REST_API_BASE_URL = 'https://api.leiapix.com'
LEIA_LOGIN_OPENID_TOKEN_URL = 'https://auth.leialoft.com/auth/realms/leialoft/protocol/openid-connect/token'


DEFAULT_ORIGINAL_IMAGE_URL = 'https://images.pexels.com/photos/38771/pexels-photo-38771.jpeg?auto=compress&cs' \
                             '=tinysrgb&w=1260&h=750&dpr=1'
ORIGINAL_IMAGE_URL = os.getenv('ORIGINAL_IMAGE_URL', DEFAULT_ORIGINAL_IMAGE_URL)
ORIGINAL_IMAGE_URL = DEFAULT_ORIGINAL_IMAGE_URL if ORIGINAL_IMAGE_URL == '' else ORIGINAL_IMAGE_URL


TWENTY_FOUR_HRS_IN_S = 24 * 60 * 60
THREE_MIN_IN_S = 3 * 60


try:
    """
    * First, we need to authenticate against Leia Login with Client
    * credentials and acquire a temporary access token.
    *
    * You can generate ClientID and Client Secret in Leia Login API Section.
    """
    print('Acquiring access token from LeiaLogin...')
    token_response = requests.post(LEIA_LOGIN_OPENID_TOKEN_URL, data={
        'client_id': CLIENT_ID,
        'client_secret': CLIENT_SECRET,
        'grant_type': 'client_credentials'
    }).json()
    access_token = token_response['access_token']
    print(f'\nLeiaLogin AccessToken acquired: {access_token}')

    """
    * Now that we have an oidc access token we can call the API. First, let's
    * generate a disparity map for our image.
    """

    # OPTIONAL. We start with preparing a correlationId. This might be an
    # internal ID which you use in your system for the entity represented
    # by the image/etc, or, as we do now, we can just generate new UUIDv4.
    # If not provided, we will generate one for you automatically.
    correlation_id = str(uuid.uuid4())
    print(f'\nGenerating Disparity with correlationId: {correlation_id}...')

    # Before you'll to be able to create an animation, you want to generate a
    # disparity map for your image and store it somewhere.

    # Here we will provide only required parameters: a correlationId and the
    # URL of the image for which we want to generate the disparity map.
    # You can find all available parameters in the documentation
    # on https://cloud.leiapix.com
    response = requests.post(
        f'{MEDIA_CLOUD_REST_API_BASE_URL}/api/v1/disparity',
        headers={
            'Authorization': f'Bearer {access_token}'
        },
        json={
            'inputImageUrl': ORIGINAL_IMAGE_URL,
            # OPTIONALLY:
            'correlationId': correlation_id,
        },
        timeout=THREE_MIN_IN_S
    )
    if not response.status_code == 201:
        raise Exception(f"Request returned with an error {response.status_code}. "
                        f"The full response is: {response.content}")

    # We omit the error handling in this example for simplicity, but
    # you should always check for a returned status & errors from the API
    # in real code.

    # The result of the call contains a short-lived  GET pre-signed URL
    # to download the resulting disparity image:
    get_disparity_presigned_url = response.json()['resultPresignedUrl']

    print(f'\nDisparity has been uploaded to the temporary storage. '
          f'To view it, use this GET URL: {get_disparity_presigned_url}')

    # If you're interested not only in a disparity map, but you also want
    # to generate an animation, you would need to make another request to
    # the service. The steps are very similar to how we called a disparity
    # map endpoint: first we (OPTIONALLY) acquire correlationId...
    correlation_id = str(uuid.uuid4())
    print(f'\nGenerating mp4 animation with correlationId: {correlation_id}...')

    # Then we make a request. This time we need two required inputs:
    # original image we want to animate (which was used for
    # disparity map generation); and an uploadable url for the result animation.
    # OPTIONALLY, you can provide the URL of the disparity map obtained from
    # the previous step. Otherwise, a new disparity map will be generated
    # automatically.
    # You can find all available parameters in the
    # documentation on https://cloud.leiapix.com
    response = requests.post(
        f'{MEDIA_CLOUD_REST_API_BASE_URL}/api/v1/animation',
        headers={
            'Authorization': f'Bearer {access_token}'
        },
        json={
            'inputImageUrl': ORIGINAL_IMAGE_URL,
            'animationLength': 5,
            # OPTIONALLY:
            'correlationId': correlation_id,
            'resultPresignedUrl': get_disparity_presigned_url,
        },
        timeout=THREE_MIN_IN_S
    )
    if not response.status_code == 201:
        raise Exception(f"Request returned with an error {response.status_code}. "
                        f"The full response is: {response.content}")

    # The resulting file is accessible via the pre-signed GET URL, that you
    # can find included in the response to the animation request:
    get_mp4_presigned_url = response.json()['resultPresignedUrl']
    # This step is optional. We generate a presigned url to download the
    # results of the animation call for convenience.

    print(f'\nMP4 Animation has been uploaded to the temporary storage. '
          f'To download, please use this GET URL: {get_mp4_presigned_url}')

except Exception as e:
    print('Error. Unhandled exception: ' + str(e), file=sys.stderr)