in static/code/coffee-counter/face_function.py [0:0]
def lambda_handler(event, context):
utime = str(int(time.time())) #Current Unix Time
coffee_cup_detected = False
bucket = event['Records'][0]['s3']['bucket']['name']
key = event['Records'][0]['s3']['object']['key']
image = {
'S3Object': {
'Bucket': bucket,
'Name': key,
}
}
response = rekognition.detect_labels(Image=image, MaxLabels=123, MinConfidence=50)
for object in response["Labels"]:
if object["Name"] == "Coffee Cup" or object["Name"] == "Cup":
coffee_cup_detected = True
break
if coffee_cup_detected:
dynamodb.put_item(
TableName=logging_table,
Item={
'unixtime':{'S':utime},
'mymess':{'S': "Coffee Cup Detected"}
})
message = detect_faces(image, bucket, key)
else:
dynamodb.put_item(
TableName=logging_table,
Item={
'unixtime':{'S':utime},
'mymess':{'S': "Coffee Cup Not Found in the image"}
})
return 'Coffee Cup Not Found in the image'
return message