You could try using Flex if you want to code in python. If you prefer to shift to flutter you could try converting your Pygame code to Flutter Flame.