// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

const ModeratedBoundingBoxLabels = [
  'Arrow',
  'Axe',
  'Beer',
  'Bow',
  'Bullet',
  'Crossbow',
  'Dagger',
  'Dynamite',
  'Grenade',
  'Gun',
  'Knife',
  'Mace Club',
  'Mortar Shell',
  'Pill',
  'Sword',
  'Tank',
  'Tattoo',
];

const ModeratedImageLabels = [
  'Absinthe',
  'Alcohol',
  'Ambulance',
  'Ammunition',
  'Archer',
  'Archery',
  'Armor',
  'Armored',
  'Armory',
  'Arrow',
  'Arrowhead',
  'Ashtray',
  'Axe',
  'Bar',
  'Bar Counter',
  'Bartender',
  'Beachwear',
  'Beer',
  'Beer Bottle',
  'Beer Glass',
  'Bikini',
  'Blade',
  'Bomb',
  'Bomber',
  'Boxer',
  'Boxing',
  'Bow',
  'Bra',
  'Brewery',
  'Bullet',
  'Casino',
  'Chain Saw',
  'Crime Scene',
  'Crossbow',
  'Cuddling',
  'Dagger',
  'Dice',
  'Drinking Alcohol',
  'Dynamite',
  'Explosion',
  'Fighting',
  'Fire',
  'Firearm',
  'Gambling',
  'Gin',
  'Graffiti',
  'Grenade',
  'Gun',
  'Handgun',
  'Handsaw',
  'Hugging',
  'Injury',
  'Kissing',
  'Knife',
  'Lager',
  'Machine Gun',
  'Medical Procedure',
  'Missile',
  'Mortar',
  'Mortar Shell',
  'Nuclear Explosion',
  'Piercing',
  'Pill',
  'Protest',
  'Pub',
  'Red Wine',
  'Rifle',
  'Riot',
  'Romantic',
  'Roulette',
  'Shooting',
  'Shooting Range',
  'Shotgun',
  'Shouting',
  'Shoving',
  'Shuffling Cards',
  'Slamming',
  'Slapping',
  'Smoke Pipe',
  'Smoking',
  'Sniper',
  'Spandex',
  'Sunbathing',
  'Swimming Trunks',
  'Swimwear',
  'Sword',
  'Swordfight',
  'Tank',
  'Tequila',
  'Tobacco',
  'Undershirt',
  'Underwear',
  'Vandalism',
  'War',
  'Weapon',
  'Whisky',
  'Wine',
  'Wine Bottle',
  'Wine Cellar',
  'Wine Glass',
  'Wrestling',
];

/* cultural labels */
const CulturalLabels = [
  'Butcher Shop',
  'Beef',
  'Pork',
  'Meat',
  'Meatball',
  'Meat Loaf',
  'Pig',
  'Boar',
];

const WhitelistLabels = [
  ...new Set([
    ...ModeratedBoundingBoxLabels,
    ...ModeratedImageLabels,
    ...CulturalLabels,
  ]),
].reduce((a0, c0) => ({
  ...a0,
  [c0]: true,
}), {});

module.exports = WhitelistLabels;
