[python] Error after upgrading pip: cannot import name 'main'

I used the following code to load a module that might need install, thus avoiding this error (which I also got) - using the latest Python and latest pip with no problem

try
  from colorama import Fore, Back, Style
except:
  !pip install colorama
  from colorama import Fore, Back, Style