chore: fix public imports (#987)

### Description

- fix public imports in `__init__` files
- Add `rich` to direct dependencies. It is one of `cookiecutter`'s
dependencies, but we use it directly in `statistics._models.py`

---------

Co-authored-by: Vlada Dusek <[email protected]>

Author: Mantisus