from load_data import items # Example of populating unique_items unique_items = set(item for sublist in items for item in sublist) # Flatten the list of lists flattened_list = [item for sublist in items for item in sublist] # Get unique items by converting the list to a set unique_items_set = set(flattened_list) # Optional: Convert the set back to a list if you need it in list form unique_items_list = list(unique_items_set) # Define the categories and products mapping product_to_category = { 'salmon': {'category': 'seafood', 'image': 'resized_images/salmon.png'}, 'tea': {'category': 'beverages', 'image': 'resized_images/tea.png'}, 'cottage cheese': {'category': 'dairy', 'image': 'resized_images/cottage cheese.png'}, 'nuggets': {'category': 'meat', 'image': 'resized_images/nuggets.png'}, 'gums': {'category': 'snacks', 'image': 'resized_images/gums.png'}, 'shampoo': {'category': 'personal care', 'image': 'resized_images/shampoo.png'}, 'water spray': {'category': 'personal care', 'image': 'resized_images/water spray.png'}, 'ketchup': {'category': 'condiments', 'image': 'resized_images/ketchup.png'}, 'protein bar': {'category': 'snacks', 'image': 'resized_images/protein bar.png'}, 'butter': {'category': 'dairy', 'image': 'resized_images/butter.png'}, 'green grapes': {'category': 'fruits', 'image': 'resized_images/green grapes.png'}, 'pickles': {'category': 'condiments', 'image': 'resized_images/pickles.png'}, 'meatballs': {'category': 'meat', 'image': 'resized_images/meatballs.png'}, 'toothpaste': {'category': 'personal care', 'image': 'resized_images/toothpaste.png'}, 'body spray': {'category': 'personal care', 'image': 'resized_images/body spray.png'}, 'shrimp': {'category': 'seafood', 'image': 'resized_images/shrimp.png'}, 'oil': {'category': 'cooking essentials', 'image': 'resized_images/oil.png'}, 'flax seed': {'category': 'cooking essentials', 'image': 'resized_images/flax seed.png'}, 'chutney': {'category': 'condiments', 'image': 'resized_images/chutney.png'}, 'asparagus': {'category': 'vegetables', 'image': 'resized_images/asparagus.png'}, 'avocado': {'category': 'fruits', 'image': 'resized_images/avocado.png'}, 'light mayo': {'category': 'condiments', 'image': 'resized_images/light mayo.png'}, 'salad': {'category': 'vegetables', 'image': 'resized_images/salad.png'}, 'eggplant': {'category': 'vegetables', 'image': 'resized_images/eggplant.png'}, 'cooking oil': {'category': 'cooking essentials', 'image': 'resized_images/cooking oil.png'}, 'salt': {'category': 'cooking essentials', 'image': 'resized_images/salt.png'}, 'extra dark chocolate': {'category': 'snacks', 'image': 'resized_images/extra dark chocolate.png'}, 'magazines': {'category': 'miscellaneous', 'image': 'resized_images/magazines.png'}, 'antioxydant juice': {'category': 'beverages', 'image': 'resized_images/antioxydant juice.png'}, 'apple juice': {'category': 'beverages', 'image': 'resized_images/apple juice.png'}, 'honey': {'category': 'cooking essentials', 'image': 'resized_images/honey.png'}, 'chili': {'category': 'vegetables', 'image': 'resized_images/chili.png'}, 'babies food': {'category': 'baby care', 'image': 'resized_images/babies food.png'}, 'cider': {'category': 'beverages', 'image': 'resized_images/cider.png'}, 'vegetables mix': {'category': 'vegetables', 'image': 'resized_images/vegetables mix.png'}, 'redbull': {'category': 'beverages', 'image': 'resized_images/redbull.png'}, 'cheese cake': {'category': 'desserts', 'image': 'resized_images/cheese cake.png'}, 'herb & pepper': {'category': 'spices', 'image': 'resized_images/herb & pepper.png'}, 'cereals': {'category': 'breakfast', 'image': 'resized_images/cereals.png'}, 'corn': {'category': 'vegetables', 'image': 'resized_images/corn.png'}, 'mayonnaise': {'category': 'condiments', 'image': 'resized_images/mayonnaise.png'}, 'carrots': {'category': 'vegetables', 'image': 'resized_images/carrots.png'}, 'napkins': {'category': 'household', 'image': 'resized_images/napkins.png'}, 'chocolate': {'category': 'snacks', 'image': 'resized_images/chocolate.png'}, 'soup': {'category': 'canned goods', 'image': 'resized_images/soup.png'}, 'yams': {'category': 'vegetables', 'image': 'resized_images/yams.png'}, 'cake': {'category': 'desserts', 'image': 'resized_images/cake.png'}, 'mint': {'category': 'spices', 'image': 'resized_images/mint.png'}, 'spaghetti': {'category': 'pasta', 'image': 'resized_images/spaghetti.png'}, 'pasta': {'category': 'pasta', 'image': 'resized_images/pasta.png'}, 'fromage blanc': {'category': 'dairy', 'image': 'resized_images/fromage blanc.png'}, 'fresh bread': {'category': 'bakery', 'image': 'resized_images/fresh bread.png'}, 'cream': {'category': 'dairy', 'image': 'resized_images/cream.png'}, 'ground beef': {'category': 'meat', 'image': 'resized_images/ground beef.png'}, 'eggs': {'category': 'breakfast', 'image': 'resized_images/eggs.png'}, 'chicken': {'category': 'meat', 'image': 'resized_images/chicken.png'}, 'hand protein bar': {'category': 'snacks', 'image': 'resized_images/hand protein bar.png'}, 'energy drink': {'category': 'beverages', 'image': 'resized_images/energy drink.png'}, 'oatmeal': {'category': 'breakfast', 'image': 'resized_images/oatmeal.png'}, 'cookies': {'category': 'snacks', 'image': 'resized_images/cookies.png'}, 'muffins': {'category': 'bakery', 'image': 'resized_images/muffins.png'}, 'frozen vegetables': {'category': 'frozen foods', 'image': 'resized_images/frozen vegetables.png'}, 'almonds': {'category': 'nuts', 'image': 'resized_images/almonds.png'}, 'beef': {'category': 'meat', 'image': 'resized_images/beef.png'}, 'bug spray': {'category': 'household', 'image': 'resized_images/bug spray.png'}, 'tomato sauce': {'category': 'condiments', 'image': 'resized_images/tomato sauce.png'}, 'spinach': {'category': 'vegetables', 'image': 'resized_images/spinach.png'}, 'sparkling water': {'category': 'beverages', 'image': 'resized_images/sparkling water.png'}, 'cauliflower': {'category': 'vegetables', 'image': 'resized_images/cauliflower.png'}, 'whole wheat rice': {'category': 'grains', 'image': 'resized_images/whole wheat rice.png'}, 'chocolate bread': {'category': 'bakery', 'image': 'resized_images/chocolate bread.png'}, 'burger sauce': {'category': 'condiments', 'image': 'resized_images/burger sauce.png'}, 'shallot': {'category': 'vegetables', 'image': 'resized_images/shallot.png'}, 'whole weat flour': {'category': 'cooking essentials', 'image': 'resized_images/whole weat flour.png'}, 'white wine': {'category': 'beverages', 'image': 'resized_images/white wine.png'}, 'barbecue sauce': {'category': 'condiments', 'image': 'resized_images/barbecue sauce.png'}, 'fresh tuna': {'category': 'seafood', 'image': 'resized_images/fresh tuna.png'}, 'blueberries': {'category': 'fruits', 'image': 'resized_images/blueberries.png'}, 'energy bar': {'category': 'snacks', 'image': 'resized_images/energy bar.png'}, 'sandwich': {'category': 'snacks', 'image': 'resized_images/sandwich.png'}, 'light cream': {'category': 'dairy', 'image': 'resized_images/light cream.png'}, 'olive oil': {'category': 'cooking essentials', 'image': 'resized_images/olive oil.png'}, 'clothes accessories': {'category': 'miscellaneous', 'image': 'resized_images/clothes accessories.png'}, 'mushroom cream sauce': {'category': 'sauces', 'image': 'resized_images/mushroom cream sauce.png'}, 'escalope': {'category': 'meat', 'image': 'resized_images/escalope.png'}, 'pancakes': {'category': 'breakfast', 'image': 'resized_images/pancakes.png'}, 'black tea': {'category': 'beverages', 'image': 'resized_images/black tea.png'}, 'grated cheese': {'category': 'dairy', 'image': 'resized_images/grated cheese.png'}, 'whole wheat pasta': {'category': 'pasta', 'image': 'resized_images/whole wheat pasta.png'}, 'mineral water': {'category': 'beverages', 'image': 'resized_images/mineral water.png'}, 'tomato juice': {'category': 'beverages', 'image': 'resized_images/tomato juice.png'}, 'candy bars': {'category': 'snacks', 'image': 'resized_images/candy bars.png'}, 'pepper': {'category': 'spices', 'image': 'resized_images/pepper.png'}, 'pet food': {'category': 'pet care', 'image': 'resized_images/pet food.png'}, 'french fries': {'category': 'snacks', 'image': 'resized_images/french fries.png'}, 'green beans': {'category': 'vegetables', 'image': 'resized_images/green beans.png'}, 'strawberries': {'category': 'fruits', 'image': 'resized_images/strawberries.png'}, 'melons': {'category': 'fruits', 'image': 'resized_images/melons.png'}, 'zucchini': {'category': 'vegetables', 'image': 'resized_images/zucchini.png'}, 'brownies': {'category': 'desserts', 'image': 'resized_images/brownies.png'}, 'milk': {'category': 'dairy', 'image': 'resized_images/milk.png'}, 'bramble': {'category': 'fruits', 'image': 'resized_images/bramble.png'}, 'low fat yogurt': {'category': 'dairy', 'image': 'resized_images/low fat yogurt.png'}, 'yogurt cake': {'category': 'desserts', 'image': 'resized_images/yogurt cake.png'}, 'frozen smoothie': {'category': 'frozen foods', 'image': 'resized_images/frozen smoothie.png'}, 'mint green tea': {'category': 'beverages', 'image': 'resized_images/mint green tea.png'}, 'tomatoes': {'category': 'vegetables', 'image': 'resized_images/tomatoes.png'}, 'strong cheese': {'category': 'dairy', 'image': 'resized_images/strong cheese.png'}, 'nonfat milk': {'category': 'dairy', 'image': 'resized_images/nonfat milk.png'}, 'green tea': {'category': 'beverages', 'image': 'resized_images/green tea.png'}, 'rice': {'category': 'grains', 'image': 'resized_images/rice.png'}, 'hot dogs': {'category': 'meat', 'image': 'resized_images/hot dogs.png'}, 'soda': {'category': 'beverages', 'image': 'resized_images/soda.png'}, 'mashed potato': {'category': 'vegetables', 'image': 'resized_images/mashed potato.png'}, 'turkey': {'category': 'meat', 'image': 'resized_images/turkey.png'}, 'french wine': {'category': 'beverages', 'image': 'resized_images/french wine.png'}, 'parmesan cheese': {'category': 'dairy', 'image': 'resized_images/parmesan cheese.png'}, 'gluten free bar': {'category': 'snacks', 'image': 'resized_images/gluten free bar.png'}, 'burgers': {'category': 'meat', 'image': 'resized_images/burgers.png'} } # Initialize the goal format dictionary goal_format = { "beverages": [], "dairy": [], "snacks": [], "seafood": [], "personal care": [], "vegetables": [], "fruits": [], "condiments": [], "cooking essentials": [], "meat": [], "baby care": [], "frozen foods": [], "nuts": [], "household": [], "breakfast": [], "bakery": [], "spices": [], "canned goods": [], "desserts": [], "pasta": [], "grains": [], "pet care": [], "sauces": [], "miscellaneous": [], } # Populate the goal_format dictionary with products from the unique_items_list for product in unique_items_list: category_dict = product_to_category.get(product) # Get the category dictionary if category_dict: category_name = category_dict['category'] # Extract the category name (string) goal_format[category_name].append(product) # Use category name as key # Optional: Remove duplicates (if any) in the product lists for category in goal_format: goal_format[category] = list(set(goal_format[category]))