screenSize function
Gets the screen size category of the screen.
Implementation
ScreenSizes screenSize(BuildContext context) {
final double width = MediaQuery.of(context).size.width;
if (width < 400.0) {
return ScreenSizes.small;
} else {
return ScreenSizes.medium;
}
}