censor_lambda_function.py

import json
import boto3

def lambda_handler(event, context):
    bucket = "censorimagesbucket"
    filename = event["filename"]
    moderation_labels = ["Explicit Nudity", "Suggestive", "Violence", "Visually Disturbing", "Rude Gestures", "Drugs", "Tobacco", "Alcohol", "Gambling", "Hate Symbols"]

    # Call Rekognition
    rekognition_client = boto3.client('rekognition')

    response = rekognition_client.detect_moderation_labels(Image={
            "S3Object": { 
                "Bucket": bucket,
                "Name": filename,
            }
        })
        
    # Check if moderation is needed    
    labels = response['ModerationLabels']
    moderation = any(((label["ParentName"] or label['Name'])in moderation_labels) for label in labels)
    
    # Return if moderation is needed
    return {
        "moderation": moderation
    }