Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import networkx as nx
 from pyvis.network import Network
 import json
-import os
+from streamlit.components.v1 import html
 
 # Streamlit app layout
 st.title("Interactive Graph Visualization")
@@ -20,9 +20,6 @@ if uploaded_file is not None:
         if "nodes" not in graph_data or "edges" not in graph_data:
             raise ValueError("The JSON file must contain 'nodes' and 'edges' keys.")
         
-        if not isinstance(graph_data["nodes"], list) or not isinstance(graph_data["edges"], list):
-            raise ValueError("'nodes' and 'edges' must be lists.")
-        
         # Function to create a NetworkX graph from data
         def create_graph(data):
             G = nx.DiGraph()
@@ -53,17 +50,12 @@ if uploaded_file is not None:
         # Create the Pyvis graph
         pyvis_graph = create_pyvis_graph(graph)
         
-        # Save and display the graph
-        output_file = "graph.html"
-        pyvis_graph.show(output_file)
-        
+        # Generate the HTML representation of the graph
+        pyvis_graph_html = pyvis_graph.generate_html()
+
         # Display the graph in Streamlit
-        with open(output_file, "r") as file:
-            st.components.v1.html(file.read(), height=600, width=800)
-        
-        # Clean up the temporary file
-        if os.path.exists(output_file):
-            os.remove(output_file)
+        html(pyvis_graph_html, height=600)
+
     except Exception as e:
         st.error(f"Error processing the file: {e}")
 else: