visualize.py

from pyvis.network import Network
import streamlit as st
import tempfile
import os

def visualize_graph(G):
    net = Network(height="550px", width="100%")
    for node in G.nodes:
        net.add_node(str(node), label=str(node))
    for edge in G.edges:
        net.add_edge(str(edge[0]), str(edge[1]))

    with tempfile.NamedTemporaryFile(delete=False, suffix=".html") as f:
        net.save_graph(f.name)
        st.components.v1.html(open(f.name).read(), height=600)