"""
    This Lambda function creates a SageMaker model.
    As input event, it receives the endpoint_name, the image_uri and the execution role. 
"""

import json
import boto3 
     
def lambda_handler(event, context):   
    
    #Amazon SageMaker session
    sm = boto3.client("sagemaker")
    region = boto3.Session().region_name
    
    #Input parameters
    endpoint_name = event['endpoint_name']
    image_uri = event['image_uri']
    role = event['role']
    model_url = event['model_path']
    
    #Create a Model using Amazon SageMaker 
    model = sm.create_model(
        ModelName=endpoint_name,
            Containers=[
                {
                    "Image": image_uri,
                    'Mode': 'SingleModel',
                    'ModelDataUrl': model_url,
                },
            ],
            ExecutionRoleArn=role,
            EnableNetworkIsolation=False,
    )
    
    return {
        "statusCode": 200,
        "body": json.dumps("Created Model!"),
        "model_name": str(endpoint_name),
    }