How to interpret .py files as jupyter notebooks

What you're looking for is jupytext. You just need to install it into python env from which you're running your jupyter notebooks:

pip install jupytext --upgrade

And you get this: