import networkx as nx
from db import get_tables, get_relationships

def build_graph():
    G = nx.Graph()
    G.add_nodes_from(get_tables())
    G.add_edges_from(get_relationships())
    return G