@souscheflabs/ml-vision 0.1.0

package/dist/hooks/classLabels.js

@@ -0,0 +1,439 @@
+"use strict";
+/**
+ * Class label mappings for the trained SousChef YOLOv8 model
+ *
+ * These labels correspond to the 233 classes in the souschef_coco_finetuned_v1 model.
+ * The model was fine-tuned on a dataset of food and kitchen items.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CATEGORY_CLASSES = exports.NUM_CLASSES = exports.CLASS_LABELS = void 0;
+exports.getClassInfo = getClassInfo;
+exports.getLabelsArray = getLabelsArray;
+exports.categorizeLabel = categorizeLabel;
+/**
+ * Categorize a food item label
+ */
+function categorizeLabel(label) {
+    // Produce
+    const produce = [
+        'acorn_squash', 'apple', 'asparagus', 'baby_carrots', 'baby_spinach',
+        'banana', 'basil', 'bell_pepper', 'blackberries', 'blueberries',
+        'broccoli', 'brussels_sprouts', 'butternut_squash', 'button_mushrooms',
+        'carrot', 'cauliflower', 'celery', 'cherries', 'cherry_tomatoes',
+        'chives', 'clementine', 'coconut', 'corn', 'cucumber', 'dill',
+        'eggplant', 'english_cucumber', 'fuji_apple', 'garlic', 'ginger',
+        'grapes', 'green_bell_pepper', 'green_onion', 'honeydew_melon',
+        'jalapeno', 'kiwi', 'lemon', 'lettuce', 'lime', 'mango', 'mixed_greens',
+        'mushrooms', 'napa_cabbage', 'onion', 'orange', 'pomegranate', 'potato',
+        'raspberries', 'red_cabbage', 'red_onion', 'roma_tomato', 'romaine_lettuce',
+        'rosemary', 'sage', 'shallot', 'spinach', 'spaghetti_squash',
+        'strawberries', 'sweet_potato', 'tomatillo', 'tomato', 'white_onion',
+        'yellow_onion', 'yellow_squash', 'yukon_gold_potato', 'zucchini',
+        'd_anjou_p_r', 'd_anjou_p_rs_2_pound', 'p_es', 'p_rs', 'pe_s', 'chillies'
+    ];
+    // Meat & Seafood
+    const meat = [
+        'ahi_tuna', 'bacon', 'beef_brisket', 'beef_broth', 'beef_short_ribs',
+        'beef_tenderloin', 'catfish_fillets', 'catfish_fillets_fr_en_big_d_l',
+        'catfish_st_k_sustainably_sourced', 'chicken_drumsticks', 'chicken_legs',
+        'chicken_tenders', 'chicken_thighs', 'chicken_wings', 'chuck_roast',
+        'cod', 'cod_fillet', 'filet_mignon', 'flank_steak', 'ground_beef',
+        'ground_chicken', 'ground_lamb', 'ham', 'lobster', 'pork_belly',
+        'pork_chops', 'pork_loin', 'pork_shoulder', 'pork_tenderloin',
+        'rotisserie_chicken', 'ry_catfish_fillets', 'shrimp', 'sirloin_steak',
+        'smoked_salmon', 'tilapia', 'tilapia_fillet'
+    ];
+    // Dairy
+    const dairy = [
+        'blue_cheese', 'butter', 'cacique_queso_fresco', 'cheddar_cheese',
+        'cr_m', 'egg_whites', 'eggs', 'goat_cheese', 'gouda_cheese',
+        'greek_yogurt', 'h_vy_cr_m', 'half_and_half', 'large_eggs', 'milk',
+        'mozzarella_cheese', 'parmesan_cheese', 'provolone_cheese',
+        'salted_butter', 'shredded_cheese', 'skim_milk', 'sliced_cheese',
+        'soy_milk', 'unsalted_butter', 'vegan_cr_m', 'whole_milk', 'yogurt'
+    ];
+    // Bakery
+    const bakery = [
+        'baguette', 'br_d', 'br_dcrumbs', 'crusty_br_d', 'hot_dog_buns',
+        'ladyfingers_bought_or', 'panko_br_d_crumbs', 'shortcrust_pastry',
+        'white_bread'
+    ];
+    // Packaged/Pantry
+    const packaged = [
+        'all_purpose_flour', 'balsamic_vinegar', 'bbq_sauce', 'blackberry_preserves_seedless',
+        'blackeye_p_s', 'bonne_maman_blackberry_preserves', 'brown_rice', 'brown_sugar',
+        'cardamom_pods', 'chickpeas', 'chilli_flakes', 'citric_acid', 'crushed_tomatoes',
+        'diced_tomatoes', 'double_strength_liquid_rennet', 'extra_virgin_olive_oil',
+        'flour', 'garam_masala', 'garbanzo_b_ns', 'herbs_de_province', 'honey',
+        'hot_sauce', 'jam', 'marinara_sauce', 'mayonnaise', 'mustard',
+        'non_dairy_nutritional_y_st', 'olive_oil', 'on_the_border_monterey_jack_queso',
+        'oregano', 'paprika', 'parchment_pa_to_line_the_baking_pan', 'pasta',
+        'pinto_beans', 'polaner_with_fiber_seedless_blackberry_preserves_g',
+        'powdered_sugar', 'pre_d_pepitas', 'pumpkin_pie_spice_shaker', 'quick_p_rled_barley',
+        'rice', 's_soning', 'salsa', 'shirataki', 'spaghetti', 'stock', 'sugar',
+        'tomato_paste', 'tomato_sauce', 'unsweetened_chocolate_baking_squares',
+        'unsweetened_coconut_flakes', 'vanilla', 'vegetable_oil', 'xanthan_gum',
+        'rosemary_l_ves', 'bay_l_f'
+    ];
+    // Snacks
+    const snack = [
+        'cashews', 'chifles_plantain_strips', 'granola_bar', 'kind_pumpkin_seeds',
+        'mixed_nuts', 'peanuts', 'philadelphia_snack_de_s_chocolate', 'plantain_single',
+        'pretzels', 'protein_bar', 'tortilla_chips', 'trail_mix', 'walnuts',
+        'vegan_marshmallows'
+    ];
+    // Beverages
+    const beverage = [
+        'bottled_water', 'cr_m_soda', 'cranberry_juice', 'orange_juice',
+        'sparkling_water', 'bigelow_benefits_cinnamon_blackberry_herbal_t_s',
+        'ly_pulled_espresso_to', 'mamma_chia_chia_squeeze_blackberry_bliss_vitality',
+        'twinings_londin_immue_support_blackberry_flavored'
+    ];
+    // Frozen
+    const frozen = [
+        'bakery_pumpkin_pie', 'frozen_fruit', 'marie_callender_s_pumpkin_pie_fr_en_dessert',
+        'microwave_the_noodles'
+    ];
+    // Supplements/Other
+    const other = [
+        'melatonin_gummies_blackberry_5mg', 'olly_men_s_multivitamin_blackberry_flavored_gummy',
+        'olly_sleep_blackberry_flavor_gummy_supplement_mela',
+        'olly_strength_sleep_blackberry_flavor_gummy_supple',
+        'purina_friskies_fish_a_licious_wet_cat_in_gravy_va',
+        'spice_islands_pumpkin_pie_spice_kosher', 'grown_fairytale_pumpkin',
+        'grown_pie_pumpkins', 'grown_pumpkins_up_to_13_s', 'grown_pumpkins_up_to_19_s',
+        'la_costena_tomatillos', 'portobello_mushrooms'
+    ];
+    if (produce.includes(label))
+        return 'produce';
+    if (meat.includes(label))
+        return 'meat';
+    if (dairy.includes(label))
+        return 'dairy';
+    if (bakery.includes(label))
+        return 'bakery';
+    if (packaged.includes(label))
+        return 'packaged';
+    if (snack.includes(label))
+        return 'snack';
+    if (beverage.includes(label))
+        return 'beverage';
+    if (frozen.includes(label))
+        return 'frozen';
+    if (other.includes(label))
+        return 'unknown';
+    // Default categorization based on common patterns
+    if (label.includes('cheese'))
+        return 'dairy';
+    if (label.includes('milk') || label.includes('cream') || label.includes('butter'))
+        return 'dairy';
+    if (label.includes('beef') || label.includes('chicken') || label.includes('pork') ||
+        label.includes('fish') || label.includes('shrimp') || label.includes('salmon'))
+        return 'meat';
+    if (label.includes('bread') || label.includes('bun'))
+        return 'bakery';
+    if (label.includes('juice') || label.includes('water') || label.includes('soda'))
+        return 'beverage';
+    return 'unknown';
+}
+/**
+ * Map class index to label and category (233 classes)
+ */
+exports.CLASS_LABELS = {
+    0: { label: 'acorn_squash', category: 'produce' },
+    1: { label: 'ahi_tuna', category: 'meat' },
+    2: { label: 'all_purpose_flour', category: 'packaged' },
+    3: { label: 'apple', category: 'produce' },
+    4: { label: 'asparagus', category: 'produce' },
+    5: { label: 'baby_carrots', category: 'produce' },
+    6: { label: 'baby_spinach', category: 'produce' },
+    7: { label: 'bacon', category: 'meat' },
+    8: { label: 'baguette', category: 'bakery' },
+    9: { label: 'bakery_pumpkin_pie', category: 'frozen' },
+    10: { label: 'balsamic_vinegar', category: 'packaged' },
+    11: { label: 'banana', category: 'produce' },
+    12: { label: 'basil', category: 'produce' },
+    13: { label: 'bay_l_f', category: 'spice' },
+    14: { label: 'bbq_sauce', category: 'condiment' },
+    15: { label: 'beef_brisket', category: 'meat' },
+    16: { label: 'beef_broth', category: 'packaged' },
+    17: { label: 'beef_short_ribs', category: 'meat' },
+    18: { label: 'beef_tenderloin', category: 'meat' },
+    19: { label: 'bell_pepper', category: 'produce' },
+    20: { label: 'bigelow_benefits_cinnamon_blackberry_herbal_t_s', category: 'beverage' },
+    21: { label: 'blackberries', category: 'produce' },
+    22: { label: 'blackberry_preserves_seedless', category: 'packaged' },
+    23: { label: 'blackeye_p_s', category: 'packaged' },
+    24: { label: 'blue_cheese', category: 'dairy' },
+    25: { label: 'blueberries', category: 'produce' },
+    26: { label: 'bonne_maman_blackberry_preserves', category: 'packaged' },
+    27: { label: 'bottled_water', category: 'beverage' },
+    28: { label: 'br_d', category: 'bakery' },
+    29: { label: 'br_dcrumbs', category: 'packaged' },
+    30: { label: 'broccoli', category: 'produce' },
+    31: { label: 'brown_rice', category: 'packaged' },
+    32: { label: 'brown_sugar', category: 'packaged' },
+    33: { label: 'brussels_sprouts', category: 'produce' },
+    34: { label: 'butter', category: 'dairy' },
+    35: { label: 'butternut_squash', category: 'produce' },
+    36: { label: 'button_mushrooms', category: 'produce' },
+    37: { label: 'cacique_queso_fresco', category: 'dairy' },
+    38: { label: 'cardamom_pods', category: 'spice' },
+    39: { label: 'carrot', category: 'produce' },
+    40: { label: 'cashews', category: 'snack' },
+    41: { label: 'catfish_fillets', category: 'meat' },
+    42: { label: 'catfish_fillets_fr_en_big_d_l', category: 'meat' },
+    43: { label: 'catfish_st_k_sustainably_sourced', category: 'meat' },
+    44: { label: 'cauliflower', category: 'produce' },
+    45: { label: 'celery', category: 'produce' },
+    46: { label: 'cheddar_cheese', category: 'dairy' },
+    47: { label: 'cherries', category: 'produce' },
+    48: { label: 'cherry_tomatoes', category: 'produce' },
+    49: { label: 'chicken_drumsticks', category: 'meat' },
+    50: { label: 'chicken_legs', category: 'meat' },
+    51: { label: 'chicken_tenders', category: 'meat' },
+    52: { label: 'chicken_thighs', category: 'meat' },
+    53: { label: 'chicken_wings', category: 'meat' },
+    54: { label: 'chickpeas', category: 'packaged' },
+    55: { label: 'chifles_plantain_strips', category: 'snack' },
+    56: { label: 'chilli_flakes', category: 'spice' },
+    57: { label: 'chillies', category: 'produce' },
+    58: { label: 'chives', category: 'produce' },
+    59: { label: 'chuck_roast', category: 'meat' },
+    60: { label: 'citric_acid', category: 'packaged' },
+    61: { label: 'clementine', category: 'produce' },
+    62: { label: 'coconut', category: 'produce' },
+    63: { label: 'cod', category: 'meat' },
+    64: { label: 'cod_fillet', category: 'meat' },
+    65: { label: 'corn', category: 'produce' },
+    66: { label: 'cr_m', category: 'dairy' },
+    67: { label: 'cr_m_soda', category: 'beverage' },
+    68: { label: 'cranberry_juice', category: 'beverage' },
+    69: { label: 'crushed_tomatoes', category: 'packaged' },
+    70: { label: 'crusty_br_d', category: 'bakery' },
+    71: { label: 'cucumber', category: 'produce' },
+    72: { label: 'd_anjou_p_r', category: 'produce' },
+    73: { label: 'd_anjou_p_rs_2_pound', category: 'produce' },
+    74: { label: 'diced_tomatoes', category: 'packaged' },
+    75: { label: 'dill', category: 'produce' },
+    76: { label: 'double_strength_liquid_rennet', category: 'packaged' },
+    77: { label: 'egg_whites', category: 'dairy' },
+    78: { label: 'eggplant', category: 'produce' },
+    79: { label: 'eggs', category: 'dairy' },
+    80: { label: 'english_cucumber', category: 'produce' },
+    81: { label: 'extra_virgin_olive_oil', category: 'packaged' },
+    82: { label: 'filet_mignon', category: 'meat' },
+    83: { label: 'flank_steak', category: 'meat' },
+    84: { label: 'flour', category: 'packaged' },
+    85: { label: 'frozen_fruit', category: 'frozen' },
+    86: { label: 'fuji_apple', category: 'produce' },
+    87: { label: 'garam_masala', category: 'spice' },
+    88: { label: 'garbanzo_b_ns', category: 'packaged' },
+    89: { label: 'garlic', category: 'produce' },
+    90: { label: 'ginger', category: 'produce' },
+    91: { label: 'goat_cheese', category: 'dairy' },
+    92: { label: 'gouda_cheese', category: 'dairy' },
+    93: { label: 'granola_bar', category: 'snack' },
+    94: { label: 'grapes', category: 'produce' },
+    95: { label: 'greek_yogurt', category: 'dairy' },
+    96: { label: 'green_bell_pepper', category: 'produce' },
+    97: { label: 'green_onion', category: 'produce' },
+    98: { label: 'ground_beef', category: 'meat' },
+    99: { label: 'ground_chicken', category: 'meat' },
+    100: { label: 'ground_lamb', category: 'meat' },
+    101: { label: 'grown_fairytale_pumpkin', category: 'produce' },
+    102: { label: 'grown_pie_pumpkins', category: 'produce' },
+    103: { label: 'grown_pumpkins_up_to_13_s', category: 'produce' },
+    104: { label: 'grown_pumpkins_up_to_19_s', category: 'produce' },
+    105: { label: 'h_vy_cr_m', category: 'dairy' },
+    106: { label: 'half_and_half', category: 'dairy' },
+    107: { label: 'herbs_de_province', category: 'spice' },
+    108: { label: 'honey', category: 'packaged' },
+    109: { label: 'honeydew_melon', category: 'produce' },
+    110: { label: 'hot_dog_buns', category: 'bakery' },
+    111: { label: 'hot_sauce', category: 'condiment' },
+    112: { label: 'jalapeno', category: 'produce' },
+    113: { label: 'jam', category: 'packaged' },
+    114: { label: 'kind_pumpkin_seeds', category: 'snack' },
+    115: { label: 'kiwi', category: 'produce' },
+    116: { label: 'la_costena_tomatillos', category: 'packaged' },
+    117: { label: 'ladyfingers_bought_or', category: 'bakery' },
+    118: { label: 'large_eggs', category: 'dairy' },
+    119: { label: 'lemon', category: 'produce' },
+    120: { label: 'lettuce', category: 'produce' },
+    121: { label: 'lime', category: 'produce' },
+    122: { label: 'lobster', category: 'meat' },
+    123: { label: 'ly_pulled_espresso_to', category: 'beverage' },
+    124: { label: 'mamma_chia_chia_squeeze_blackberry_bliss_vitality', category: 'beverage' },
+    125: { label: 'mango', category: 'produce' },
+    126: { label: 'marie_callender_s_pumpkin_pie_fr_en_dessert', category: 'frozen' },
+    127: { label: 'marinara_sauce', category: 'packaged' },
+    128: { label: 'mayonnaise', category: 'condiment' },
+    129: { label: 'melatonin_gummies_blackberry_5mg', category: 'unknown' },
+    130: { label: 'microwave_the_noodles', category: 'frozen' },
+    131: { label: 'mixed_greens', category: 'produce' },
+    132: { label: 'mixed_nuts', category: 'snack' },
+    133: { label: 'mozzarella_cheese', category: 'dairy' },
+    134: { label: 'mushrooms', category: 'produce' },
+    135: { label: 'mustard', category: 'condiment' },
+    136: { label: 'napa_cabbage', category: 'produce' },
+    137: { label: 'non_dairy_nutritional_y_st', category: 'packaged' },
+    138: { label: 'olive_oil', category: 'packaged' },
+    139: { label: 'olly_men_s_multivitamin_blackberry_flavored_gummy', category: 'unknown' },
+    140: { label: 'olly_sleep_blackberry_flavor_gummy_supplement_mela', category: 'unknown' },
+    141: { label: 'olly_strength_sleep_blackberry_flavor_gummy_supple', category: 'unknown' },
+    142: { label: 'on_the_border_monterey_jack_queso', category: 'packaged' },
+    143: { label: 'onion', category: 'produce' },
+    144: { label: 'orange', category: 'produce' },
+    145: { label: 'orange_juice', category: 'beverage' },
+    146: { label: 'oregano', category: 'spice' },
+    147: { label: 'p_es', category: 'produce' },
+    148: { label: 'p_rs', category: 'produce' },
+    149: { label: 'panko_br_d_crumbs', category: 'packaged' },
+    150: { label: 'paprika', category: 'spice' },
+    151: { label: 'parchment_pa_to_line_the_baking_pan', category: 'packaged' },
+    152: { label: 'parmesan_cheese', category: 'dairy' },
+    153: { label: 'pasta', category: 'packaged' },
+    154: { label: 'pe_s', category: 'produce' },
+    155: { label: 'peanuts', category: 'snack' },
+    156: { label: 'philadelphia_snack_de_s_chocolate', category: 'snack' },
+    157: { label: 'pinto_beans', category: 'packaged' },
+    158: { label: 'plantain_single', category: 'produce' },
+    159: { label: 'polaner_with_fiber_seedless_blackberry_preserves_g', category: 'packaged' },
+    160: { label: 'pomegranate', category: 'produce' },
+    161: { label: 'pork_belly', category: 'meat' },
+    162: { label: 'pork_chops', category: 'meat' },
+    163: { label: 'pork_loin', category: 'meat' },
+    164: { label: 'pork_shoulder', category: 'meat' },
+    165: { label: 'pork_tenderloin', category: 'meat' },
+    166: { label: 'portobello_mushrooms', category: 'produce' },
+    167: { label: 'potato', category: 'produce' },
+    168: { label: 'powdered_sugar', category: 'packaged' },
+    169: { label: 'pre_d_pepitas', category: 'snack' },
+    170: { label: 'pretzels', category: 'snack' },
+    171: { label: 'protein_bar', category: 'snack' },
+    172: { label: 'provolone_cheese', category: 'dairy' },
+    173: { label: 'pumpkin_pie_spice_shaker', category: 'spice' },
+    174: { label: 'purina_friskies_fish_a_licious_wet_cat_in_gravy_va', category: 'unknown' },
+    175: { label: 'quick_p_rled_barley', category: 'packaged' },
+    176: { label: 'raspberries', category: 'produce' },
+    177: { label: 'red_cabbage', category: 'produce' },
+    178: { label: 'red_onion', category: 'produce' },
+    179: { label: 'rice', category: 'packaged' },
+    180: { label: 'roma_tomato', category: 'produce' },
+    181: { label: 'romaine_lettuce', category: 'produce' },
+    182: { label: 'rosemary', category: 'produce' },
+    183: { label: 'rosemary_l_ves', category: 'spice' },
+    184: { label: 'rotisserie_chicken', category: 'meat' },
+    185: { label: 'ry_catfish_fillets', category: 'meat' },
+    186: { label: 's_soning', category: 'spice' },
+    187: { label: 'sage', category: 'produce' },
+    188: { label: 'salsa', category: 'condiment' },
+    189: { label: 'salted_butter', category: 'dairy' },
+    190: { label: 'shallot', category: 'produce' },
+    191: { label: 'shirataki', category: 'packaged' },
+    192: { label: 'shortcrust_pastry', category: 'bakery' },
+    193: { label: 'shredded_cheese', category: 'dairy' },
+    194: { label: 'shrimp', category: 'meat' },
+    195: { label: 'sirloin_steak', category: 'meat' },
+    196: { label: 'skim_milk', category: 'dairy' },
+    197: { label: 'sliced_cheese', category: 'dairy' },
+    198: { label: 'smoked_salmon', category: 'meat' },
+    199: { label: 'soy_milk', category: 'dairy' },
+    200: { label: 'spaghetti', category: 'packaged' },
+    201: { label: 'spaghetti_squash', category: 'produce' },
+    202: { label: 'sparkling_water', category: 'beverage' },
+    203: { label: 'spice_islands_pumpkin_pie_spice_kosher', category: 'spice' },
+    204: { label: 'spinach', category: 'produce' },
+    205: { label: 'stock', category: 'packaged' },
+    206: { label: 'sugar', category: 'packaged' },
+    207: { label: 'tilapia', category: 'meat' },
+    208: { label: 'tilapia_fillet', category: 'meat' },
+    209: { label: 'tomatillo', category: 'produce' },
+    210: { label: 'tomato', category: 'produce' },
+    211: { label: 'tomato_paste', category: 'packaged' },
+    212: { label: 'tomato_sauce', category: 'packaged' },
+    213: { label: 'tortilla_chips', category: 'snack' },
+    214: { label: 'trail_mix', category: 'snack' },
+    215: { label: 'twinings_londin_immue_support_blackberry_flavored', category: 'beverage' },
+    216: { label: 'unsalted_butter', category: 'dairy' },
+    217: { label: 'unsweetened_chocolate_baking_squares', category: 'packaged' },
+    218: { label: 'unsweetened_coconut_flakes', category: 'packaged' },
+    219: { label: 'vanilla', category: 'packaged' },
+    220: { label: 'vegan_cr_m', category: 'dairy' },
+    221: { label: 'vegan_marshmallows', category: 'snack' },
+    222: { label: 'vegetable_oil', category: 'packaged' },
+    223: { label: 'walnuts', category: 'snack' },
+    224: { label: 'white_bread', category: 'bakery' },
+    225: { label: 'white_onion', category: 'produce' },
+    226: { label: 'whole_milk', category: 'dairy' },
+    227: { label: 'xanthan_gum', category: 'packaged' },
+    228: { label: 'yellow_onion', category: 'produce' },
+    229: { label: 'yellow_squash', category: 'produce' },
+    230: { label: 'yogurt', category: 'dairy' },
+    231: { label: 'yukon_gold_potato', category: 'produce' },
+    232: { label: 'zucchini', category: 'produce' },
+};
+/**
+ * Number of classes in the model
+ */
+exports.NUM_CLASSES = 233;
+/**
+ * Get label info for a class index
+ */
+function getClassInfo(classIndex) {
+    const info = exports.CLASS_LABELS[classIndex];
+    if (info) {
+        return info;
+    }
+    // Fallback for unknown indices
+    return { label: `class_${classIndex}`, category: 'unknown' };
+}
+/**
+ * Get all labels as an array (ordered by class index)
+ */
+function getLabelsArray() {
+    const labels = [];
+    for (let i = 0; i < exports.NUM_CLASSES; i++) {
+        labels.push(exports.CLASS_LABELS[i]?.label ?? `class_${i}`);
+    }
+    return labels;
+}
+/**
+ * Category to class indices mapping
+ */
+exports.CATEGORY_CLASSES = {
+    produce: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'produce')
+        .map(([k]) => parseInt(k, 10)),
+    meat: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'meat')
+        .map(([k]) => parseInt(k, 10)),
+    dairy: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'dairy')
+        .map(([k]) => parseInt(k, 10)),
+    packaged: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'packaged')
+        .map(([k]) => parseInt(k, 10)),
+    bakery: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'bakery')
+        .map(([k]) => parseInt(k, 10)),
+    snack: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'snack')
+        .map(([k]) => parseInt(k, 10)),
+    beverage: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'beverage')
+        .map(([k]) => parseInt(k, 10)),
+    condiment: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'condiment')
+        .map(([k]) => parseInt(k, 10)),
+    spice: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'spice')
+        .map(([k]) => parseInt(k, 10)),
+    frozen: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'frozen')
+        .map(([k]) => parseInt(k, 10)),
+    unknown: Object.entries(exports.CLASS_LABELS)
+        .filter(([, v]) => v.category === 'unknown')
+        .map(([k]) => parseInt(k, 10)),
+};

package/dist/hooks/classLabelsCoco.d.ts

@@ -0,0 +1,43 @@
+/**
+ * Class label mappings for YOLOv8 COCO model
+ *
+ * These labels correspond to the 80 classes in the COCO dataset.
+ * Food-related classes: banana (46), apple (47), sandwich (48), orange (49),
+ * broccoli (50), carrot (51), hot dog (52), pizza (53), donut (54), cake (55)
+ */
+import type { ProductCategory, ClassLabelMap } from '../types';
+/**
+ * COCO class names in order
+ */
+export declare const COCO_CLASSES: readonly ["person", "bicycle", "car", "motorcycle", "airplane", "bus", "train", "truck", "boat", "traffic light", "fire hydrant", "stop sign", "parking meter", "bench", "bird", "cat", "dog", "horse", "sheep", "cow", "elephant", "bear", "zebra", "giraffe", "backpack", "umbrella", "handbag", "tie", "suitcase", "frisbee", "skis", "snowboard", "sports ball", "kite", "baseball bat", "baseball glove", "skateboard", "surfboard", "tennis racket", "bottle", "wine glass", "cup", "fork", "knife", "spoon", "bowl", "banana", "apple", "sandwich", "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair", "couch", "potted plant", "bed", "dining table", "toilet", "tv", "laptop", "mouse", "remote", "keyboard", "cell phone", "microwave", "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase", "scissors", "teddy bear", "hair drier", "toothbrush"];
+/**
+ * Map class index to label and category for COCO
+ */
+export declare const CLASS_LABELS_COCO: ClassLabelMap;
+/**
+ * Number of classes in COCO model
+ */
+export declare const NUM_CLASSES_COCO = 80;
+/**
+ * Food class indices in COCO (useful for filtering)
+ */
+export declare const FOOD_CLASS_INDICES: number[];
+/**
+ * Kitchen/dining class indices (context)
+ */
+export declare const KITCHEN_CLASS_INDICES: number[];
+/**
+ * Get label info for a COCO class index
+ */
+export declare function getCocoClassInfo(classIndex: number): {
+    label: string;
+    category: ProductCategory;
+};
+/**
+ * Check if a class index is a food item
+ */
+export declare function isFoodClass(classIndex: number): boolean;
+/**
+ * Get all COCO labels as an array
+ */
+export declare function getCocoLabelsArray(): string[];

package/dist/hooks/classLabelsCoco.js

@@ -0,0 +1,103 @@
+"use strict";
+/**
+ * Class label mappings for YOLOv8 COCO model
+ *
+ * These labels correspond to the 80 classes in the COCO dataset.
+ * Food-related classes: banana (46), apple (47), sandwich (48), orange (49),
+ * broccoli (50), carrot (51), hot dog (52), pizza (53), donut (54), cake (55)
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KITCHEN_CLASS_INDICES = exports.FOOD_CLASS_INDICES = exports.NUM_CLASSES_COCO = exports.CLASS_LABELS_COCO = exports.COCO_CLASSES = void 0;
+exports.getCocoClassInfo = getCocoClassInfo;
+exports.isFoodClass = isFoodClass;
+exports.getCocoLabelsArray = getCocoLabelsArray;
+/**
+ * COCO class names in order
+ */
+exports.COCO_CLASSES = [
+    'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck',
+    'boat', 'traffic light', 'fire hydrant', 'stop sign', 'parking meter', 'bench',
+    'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant', 'bear', 'zebra',
+    'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee',
+    'skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove',
+    'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup',
+    'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange',
+    'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch',
+    'potted plant', 'bed', 'dining table', 'toilet', 'tv', 'laptop', 'mouse',
+    'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink',
+    'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier',
+    'toothbrush'
+];
+/**
+ * Map COCO class to a product category
+ */
+function getCocoCategory(label) {
+    // Food items
+    const produceItems = ['banana', 'apple', 'orange', 'broccoli', 'carrot'];
+    const meatItems = ['hot dog'];
+    const bakeryItems = ['sandwich', 'pizza', 'donut', 'cake'];
+    const dairyItems = [];
+    const beverageItems = ['bottle', 'wine glass', 'cup'];
+    const condimentItems = ['bowl']; // Often contains condiments/sauces
+    if (produceItems.includes(label))
+        return 'produce';
+    if (meatItems.includes(label))
+        return 'meat';
+    if (bakeryItems.includes(label))
+        return 'bakery';
+    if (dairyItems.includes(label))
+        return 'dairy';
+    if (beverageItems.includes(label))
+        return 'beverage';
+    if (condimentItems.includes(label))
+        return 'condiment';
+    // Kitchen items (useful for context)
+    const kitchenItems = [
+        'fork', 'knife', 'spoon', 'microwave', 'oven', 'toaster', 'sink',
+        'refrigerator', 'dining table'
+    ];
+    if (kitchenItems.includes(label))
+        return 'unknown'; // Not a product but kitchen context
+    return 'unknown';
+}
+/**
+ * Map class index to label and category for COCO
+ */
+exports.CLASS_LABELS_COCO = Object.fromEntries(exports.COCO_CLASSES.map((label, index) => [
+    index,
+    { label, category: getCocoCategory(label) }
+]));
+/**
+ * Number of classes in COCO model
+ */
+exports.NUM_CLASSES_COCO = 80;
+/**
+ * Food class indices in COCO (useful for filtering)
+ */
+exports.FOOD_CLASS_INDICES = [46, 47, 48, 49, 50, 51, 52, 53, 54, 55];
+/**
+ * Kitchen/dining class indices (context)
+ */
+exports.KITCHEN_CLASS_INDICES = [39, 40, 41, 42, 43, 44, 45, 60, 68, 69, 70, 71, 72];
+/**
+ * Get label info for a COCO class index
+ */
+function getCocoClassInfo(classIndex) {
+    const info = exports.CLASS_LABELS_COCO[classIndex];
+    if (info) {
+        return info;
+    }
+    return { label: `class_${classIndex}`, category: 'unknown' };
+}
+/**
+ * Check if a class index is a food item
+ */
+function isFoodClass(classIndex) {
+    return exports.FOOD_CLASS_INDICES.includes(classIndex);
+}
+/**
+ * Get all COCO labels as an array
+ */
+function getCocoLabelsArray() {
+    return [...exports.COCO_CLASSES];
+}

package/dist/hooks/index.d.ts

@@ -0,0 +1,8 @@
+/**
+ * Hooks exports
+ */
+export { useProductDetector, default as useProductDetectorDefault } from './useProductDetector';
+export { useMultiBarcodeScanner, default as useMultiBarcodeScannerDefault } from './useMultiBarcodeScanner';
+export { useReceiptScanner, default as useReceiptScannerDefault } from './useReceiptScanner';
+export { useVideoScanner, default as useVideoScannerDefault, type ScanMode, type VideoScanResult, type UseVideoScannerOptions, type UseVideoScannerReturn, } from './useVideoScanner';
+export { CLASS_LABELS, NUM_CLASSES, getClassInfo, getLabelsArray, CATEGORY_CLASSES } from './classLabels';

package/dist/hooks/index.js

@@ -0,0 +1,27 @@
+"use strict";
+/**
+ * Hooks exports
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CATEGORY_CLASSES = exports.getLabelsArray = exports.getClassInfo = exports.NUM_CLASSES = exports.CLASS_LABELS = exports.useVideoScannerDefault = exports.useVideoScanner = exports.useReceiptScannerDefault = exports.useReceiptScanner = exports.useMultiBarcodeScannerDefault = exports.useMultiBarcodeScanner = exports.useProductDetectorDefault = exports.useProductDetector = void 0;
+var useProductDetector_1 = require("./useProductDetector");
+Object.defineProperty(exports, "useProductDetector", { enumerable: true, get: function () { return useProductDetector_1.useProductDetector; } });
+Object.defineProperty(exports, "useProductDetectorDefault", { enumerable: true, get: function () { return __importDefault(useProductDetector_1).default; } });
+var useMultiBarcodeScanner_1 = require("./useMultiBarcodeScanner");
+Object.defineProperty(exports, "useMultiBarcodeScanner", { enumerable: true, get: function () { return useMultiBarcodeScanner_1.useMultiBarcodeScanner; } });
+Object.defineProperty(exports, "useMultiBarcodeScannerDefault", { enumerable: true, get: function () { return __importDefault(useMultiBarcodeScanner_1).default; } });
+var useReceiptScanner_1 = require("./useReceiptScanner");
+Object.defineProperty(exports, "useReceiptScanner", { enumerable: true, get: function () { return useReceiptScanner_1.useReceiptScanner; } });
+Object.defineProperty(exports, "useReceiptScannerDefault", { enumerable: true, get: function () { return __importDefault(useReceiptScanner_1).default; } });
+var useVideoScanner_1 = require("./useVideoScanner");
+Object.defineProperty(exports, "useVideoScanner", { enumerable: true, get: function () { return useVideoScanner_1.useVideoScanner; } });
+Object.defineProperty(exports, "useVideoScannerDefault", { enumerable: true, get: function () { return __importDefault(useVideoScanner_1).default; } });
+var classLabels_1 = require("./classLabels");
+Object.defineProperty(exports, "CLASS_LABELS", { enumerable: true, get: function () { return classLabels_1.CLASS_LABELS; } });
+Object.defineProperty(exports, "NUM_CLASSES", { enumerable: true, get: function () { return classLabels_1.NUM_CLASSES; } });
+Object.defineProperty(exports, "getClassInfo", { enumerable: true, get: function () { return classLabels_1.getClassInfo; } });
+Object.defineProperty(exports, "getLabelsArray", { enumerable: true, get: function () { return classLabels_1.getLabelsArray; } });
+Object.defineProperty(exports, "CATEGORY_CLASSES", { enumerable: true, get: function () { return classLabels_1.CATEGORY_CLASSES; } });