Stevia | :leaves: Concise Autolayout code | iOS library

This is about body mass index classification and each bmi grade have a suggest diet.

How can I create a function whether the div element is same with constant's object, while the constant have multiple object. If the element inside div is same as specific constant's object, then toggleClass(). I tried to use a if else and is() function but it doesn't work. The example is below:

My goal is to create a layout presented below:

I know how to create these custom UICollectionViewCells, but I trouble with the layout. All of the shown cells differ in width, so there can be, for instance: four in the first row, two in the second, and the last one remaining - in the third. That's just one possible configuration, and there are many more (including one shown on an image) depending on the label's width.

I'm building everything programmatically. Also I feel like using UICollectionView is the best choice, but I'm open to any suggestions.

Thanks in advance!

What I've already tried:

I am able to select the image from the directory and store in the specified cell, but I can't work out how to get it to save to the Database. I hit enter or tab and the image just goes away.

Here is the delegate I am currently using