import boto3

def add_cors_headers_to_gateway_responses(api_id, region):
    client = boto3.client('apigateway', region_name=region)

    # Define the CORS headers
    cors_headers = {
        'Access-Control-Allow-Origin': "'*'",
        'Access-Control-Allow-Headers': "'Content-Type,X-Amz-Date,Authorization,X-Api-Key,X-Amz-Security-Token'",
        'Access-Control-Allow-Methods': "'GET,OPTIONS,POST,PUT'"
    }

    # Update default 4xx response
    client.update_gateway_response(
        restApiId=api_id,
        responseType='DEFAULT_4XX',
        patchOperations=[
            {
                'op': 'add',
                'path': '/responseParameters/gatewayresponse.header.Access-Control-Allow-Origin',
                'value': "'*'"
            },
            {
                'op': 'add',
                'path': '/responseParameters/gatewayresponse.header.Access-Control-Allow-Headers',
                'value': "'Content-Type,X-Amz-Date,Authorization,X-Api-Key,X-Amz-Security-Token'"
            },
            {
                'op': 'add',
                'path': '/responseParameters/gatewayresponse.header.Access-Control-Allow-Methods',
                'value': "'GET,OPTIONS,POST,PUT'"
            }
        ]
    )

    # Update default 5xx response
    client.update_gateway_response(
        restApiId=api_id,
        responseType='DEFAULT_5XX',
        patchOperations=[
            {
                'op': 'add',
                'path': '/responseParameters/gatewayresponse.header.Access-Control-Allow-Origin',
                'value': "'*'"
            },
            {
                'op': 'add',
                'path': '/responseParameters/gatewayresponse.header.Access-Control-Allow-Headers',
                'value': "'Content-Type,X-Amz-Date,Authorization,X-Api-Key,X-Amz-Security-Token'"
            },
            {
                'op': 'add',
                'path': '/responseParameters/gatewayresponse.header.Access-Control-Allow-Methods',
                'value': "'GET,OPTIONS,POST,PUT'"
            }
        ]
    )

    print("CORS headers added to DEFAULT_4XX and DEFAULT_5XX gateway responses.")

# Usage
api_id = 'your-api-id'
region = 'your-region'
add_cors_headers_to_gateway_responses(api_id, region)