I want to do a different layout for different screen sizes. Can you please tell me how it should be done? (e.g. getting screen size in composable function?)

You could use a

ConfigurationAmbient

to access screen width and height, ex.

ConfigurationAmbient.current.screenWidthDp