How to implement Layout Manager for fixed size layout with different number of images?

Question

Android

I am developing an application where the user can create notes. The note contains a title, images, and text. A note has multiple images, but can also contain a single image or white space. The user adds pictures one by one. Pictures have a container layout with a fixed height of 200dp. I have to add an unknown number of images to this container, where it will be beautifully combined from them without empty space, like in the picture below. How can I implement this?
Creating notes

I need to implement something like this

Answer 1

To solve the problem, I used Flexbox from Google.

Answer 2

I think you can get inspired by something like this