graph.py

from rec import make_graph
import networkx as nx
from matplotlib import pyplot as plt

graph_dict = make_graph('fiona apple', 3, 3)
walk_graph = nx.DiGraph(graph_dict)

nx.draw_planar(walk_graph, with_labels=True)
plt.title('Random Walk through Spotify')
plt.draw()
plt.show()