"""Curated candidate lists + question-style suggestions.

Three categories: animal, fruit, vegetable. The model writes its own kid-friendly
multiple-choice questions (guided by SUGGESTIONS) and guesses from the matching
candidate list.
"""

ANIMALS = [
  "cat", "dog", "lion", "tiger", "elephant", "giraffe", "zebra", "monkey",
  "horse", "cow", "pig", "sheep", "goat", "chicken", "duck", "rabbit",
  "bear", "panda", "kangaroo", "koala", "fox", "wolf", "deer", "camel",
  "crocodile", "snake", "frog", "penguin", "dolphin", "owl",
]

FRUITS = [
  "apple", "banana", "orange", "mango", "grapes", "strawberry", "pineapple",
  "watermelon", "pear", "cherry", "kiwi", "papaya", "guava", "lemon", "lime",
  "pomegranate", "raspberry", "blueberry", "coconut", "avocado", "apricot",
  "peach", "plum", "fig", "grapefruit", "cantaloupe", "lychee", "dragonfruit",
  "jackfruit", "cranberry",
]

VEGETABLES =[
  "carrot",
  "spinach",
  "potato",
  "tomato",
  "broccoli",
  "onion",
  "garlic",
  "cucumber",
  "pumpkin",
  "corn",
  "mushroom",
  "cabbage",
  "eggplant",
  "zucchini",
  "peas",
  "cauliflower",
  "bell pepper",
  "sweet potato",
  "radish",
  "asparagus",
  "celery",
  "beetroot",
  "okra",
  "kale",
  "leek",
  "ginger",
  "chili pepper",
  "lettuce",
  "turnip",
  "green beans"
]


CANDIDATES = {
    "animal": ANIMALS,
    "fruit": FRUITS,
    "vegetable": VEGETABLES,
}
 

# Fixed pool of 20 kid-friendly multiple-choice questions per category.
# The model MUST pick the next question ONLY from the matching list below
# (it may not invent new questions or reword these).
QUESTIONS = {
    "animal": [
        
{"question": "Does it live in water?", "answers": ["Yes", "No", "Partially"]},
{"question": "Can it fly?", "answers": ["Yes", "No"]},
{"question": "Is it a domesticated animal?", "answers": ["Yes", "No"]},
{"question": "Is it larger than a human?", "answers": ["Yes", "No", "About the same size"]},
{"question": "Does it have fur?", "answers": ["Yes", "No"]},
{"question": "Is it primarily active at night?", "answers": ["Yes", "No", "I am not sure"]},
{"question": "Does it lay eggs?", "answers": ["Yes", "No"]},
{"question": "Is it a predator?", "answers": ["Yes", "No", "I am not sure"]},
{"question": "Does it live in the jungle?", "answers": ["Yes", "No"]},
{"question": "Is it found in cold climates?", "answers": ["Yes", "No"]},
{"question": "Does it have stripes?", "answers": ["Yes", "No"]},
{"question": "Does it have horns or antlers?", "answers": ["Yes", "No"]},
{"question": "Is it known for being fast?", "answers": ["Yes", "No"]},
{"question": "Is it commonly kept as a pet?", "answers": ["Yes", "No"]},
{"question": "Does it live underground?", "answers": ["Yes", "No"]},
{"question": "Is it a herbivore?", "answers": ["Yes", "No", "Partially"]},
{"question": "Does it have a long tail?", "answers": ["Yes", "No"]},
{"question": "Is it dangerous to humans?", "answers": ["Yes", "No", "I am not sure"]},
{"question": "Is it native to Africa?", "answers": ["Yes", "No"]},
{"question": "Does it have wings?", "answers": ["Yes", "No"]},
{"question": "Is it known for its intelligence?", "answers": ["Yes", "No"]},
{"question": "Does it live in groups?", "answers": ["Yes", "No", "I am not sure"]},
{"question": "Is it a reptile?", "answers": ["Yes", "No"]},
{"question": "Is it an endangered species?", "answers": ["Yes", "No"]},
{"question": "Does it live on land?", "answers": ["Yes", "No", "Both land and water"]}

    ],
    "fruit": [
        
 
  {
    "question": "Is it sweet?",
    "answers": ["Yes", "No", "A Little"]
  },
  {
    "question": "Can you eat it raw?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it yellow?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Is it red?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Is it green?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Is it orange?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Does it have seeds?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Can you peel it before eating?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Is it juicy?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it soft inside?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it round?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Does it grow on a tree?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Does it grow on a vine?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it a tropical fruit?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Can it fit in one hand?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it bigger than an apple?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it commonly used in juice?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it commonly used in desserts?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it sour?",
    "answers": ["Yes", "No", "A Little"]
  },
  {
    "question": "Is it crunchy when you bite it?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Does it have a hard pit or stone inside?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Can you eat the skin?",
    "answers": ["Yes", "No", "I am not sure"]
  },
  {
    "question": "Is it usually eaten as a snack?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Is it available all year round?",
    "answers": ["Yes", "No"]
  },
  {
    "question": "Would you usually find it in a fruit salad?",
    "answers": ["Yes", "No"]
  }


    ],
    "vegetable": [
        
  {"question": "Is it a root vegetable?", "answers": ["Yes", "No"]},
  {"question": "Is it green in color?", "answers": ["Yes", "No", "I am not sure"]},
  {"question": "Is it commonly eaten raw?", "answers": ["Yes", "No", "I am not sure"]},
  {"question": "Is it used in salads?", "answers": ["Yes", "No"]},
  {"question": "Is it a leafy vegetable?", "answers": ["Yes", "No"]},
  {"question": "Is it spicy or pungent?", "answers": ["Yes", "No"]},
  {"question": "Is it commonly used in soups?", "answers": ["Yes", "No"]},
  {"question": "Is it grown underground?", "answers": ["Yes", "No"]},
  {"question": "Is it typically cooked before eating?", "answers": ["Yes", "No"]},
  {"question": "Is it considered a fruit botanically?", "answers": ["Yes", "No"]},
  {"question": "Is it used as a spice or seasoning?", "answers": ["Yes", "No"]},
  {"question": "Is it bitter in taste?", "answers": ["Yes", "No"]},
  {"question": "Is it commonly used in curries?", "answers": ["Yes", "No"]},
  {"question": "Is it high in water content?", "answers": ["Yes", "No"]},
  {"question": "Is it a type of bean or legume?", "answers": ["Yes", "No"]},
  {"question": "Is it part of the cabbage family?", "answers": ["Yes", "No"]},
  {"question": "Is it long in shape?", "answers": ["Yes", "No"]},
  {"question": "Is it round in shape?", "answers": ["Yes", "No"]},
  {"question": "Is it commonly pickled?", "answers": ["Yes", "No"]},
  {"question": "Is it used to make juice?", "answers": ["Yes", "No"]},
  {"question": "Is it a starchy vegetable?", "answers": ["Yes", "No"]},
  {"question": "Is it commonly grilled?", "answers": ["Yes", "No"]},
  {"question": "Is it used in stir‑fry dishes?", "answers": ["Yes", "No"]},
  {"question": "Is it known for its strong smell?", "answers": ["Yes", "No"]},
  {"question": "Is it commonly used as a garnish?", "answers": ["Yes", "No"]}


    ],
}