Обучение Python — работа в Москве

Для себя Есть готовый проект - по сути готовый код для координации небольшой группы роботов с помощью нейронной сети, все алгоритмы прописаны, нейронная сеть и ее обучение тоже, также есть видео с объяснением того, как работает код и видео-пример результата. Мне нужна сама визуализация этого кода - то есть по сути получить то же, что есть на видео в результате, но в любом удобном симуляторе или в среде программирования визуализировать результат работы