Spaces:
Sleeping
Sleeping
| 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])) |