Create Docker container setup for the project

[Rabbidon]

[rk1a]

https://colab.research.google.com/drive/1KDvWgmhV445JB73WRqGijkegDVlY7e6p?usp=sharing

^ something like this colab converted to Dockerfile is what we need also there should be options like whether to build headless or not