https://jayeshkrt.github.io/posts/take-input-of-graph-python/