Is it possible to use

val boxModifier = Modifier
    .fillMaxSize() or .fillMaxHeight()
    .weight(1f)

while making sure that the box remains square, so basically, fill the maximum size/height as long as you are square? At the moment I have rectangles which is not looking great...

There is a modifier called

aspectRatio

. That might be what you are looking for

ah cool thanks!