Keras Custom Generator looping through Multiple Folders

Question

I am trying to build a custom Keras generator that would loop through folders and read a large numpy array. So i have folders like this

data
  --- train