Update app.py

app.py

@@ -5,21 +5,44 @@ import json
 from streamlit.components.v1 import html
 
 # Streamlit app layout
-st.title("Interactive Graph Visualization")
-st.write("Upload a JSON file containing nodes and edges to display the graph.")
+st.title("Interactive Graph Visualization and Editor")
+st.write("Edit the graph's JSON data and visualize updates dynamically.")
 
-# File upload
-uploaded_file = st.file_uploader("Upload JSON file", type=["json"])
+# Read the test.json file initially
+default_file = "test.json"
 
-if uploaded_file is not None:
+# Function to load JSON from file
+def load_json_file(file_path):
+    try:
+        with open(file_path, "r") as file:
+            return json.load(file)
+    except Exception as e:
+        st.error(f"Error loading the file: {e}")
+        return {"nodes": [], "edges": []}
+
+# Load the graph data from the default JSON file
+graph_data = load_json_file(default_file)
+
+# Sidebar editor for JSON input
+st.sidebar.header("Edit Graph JSON")
+json_editor = st.sidebar.text_area("Graph JSON", value=json.dumps(graph_data, indent=4), height=300)
+
+if st.sidebar.button("Update Graph"):
+    try:
+        # Parse the updated JSON
+        graph_data = json.loads(json_editor)
+        # Save back to the file
+        with open(default_file, "w") as file:
+            json.dump(graph_data, file, indent=4)
+        st.success("Graph JSON updated successfully!")
+    except Exception as e:
+        st.sidebar.error(f"Invalid JSON format: {e}")
+
+# Validate JSON structure
+if "nodes" not in graph_data or "edges" not in graph_data:
+    st.error("The JSON file must contain 'nodes' and 'edges' keys.")
+else:
     try:
-        # Load JSON data
-        graph_data = json.load(uploaded_file)
-        
-        # Validate JSON structure
-        if "nodes" not in graph_data or "edges" not in graph_data:
-            raise ValueError("The JSON file must contain 'nodes' and 'edges' keys.")
-        
         # Function to create a NetworkX graph from data
         def create_graph(data):
             G = nx.DiGraph()
@@ -28,16 +51,10 @@ if uploaded_file is not None:
             for edge in data["edges"]:
                 G.add_edge(edge["source"], edge["target"], label=edge.get("label", ""))
             return G
-        
+
         # Generate the graph
         graph = create_graph(graph_data)
-        
-        # Check if the graph has any nodes or edges
-        if graph.number_of_nodes() == 0:
-            raise ValueError("The graph has no nodes.")
-        if graph.number_of_edges() == 0:
-            raise ValueError("The graph has no edges.")
-        
+
         # Function to create a pyvis network from NetworkX graph
         def create_pyvis_graph(G):
             net = Network(height="600px", width="100%", directed=True)
@@ -49,7 +66,7 @@ if uploaded_file is not None:
 
         # Create the Pyvis graph
         pyvis_graph = create_pyvis_graph(graph)
-        
+
         # Generate the HTML representation of the graph
         pyvis_graph_html = pyvis_graph.generate_html()
 
@@ -57,6 +74,4 @@ if uploaded_file is not None:
         html(pyvis_graph_html, height=600)
 
     except Exception as e:
-        st.error(f"Error processing the file: {e}")
-else:
-    st.info("Please upload a JSON file to display the graph.")
+        st.error(f"Error processing the graph: {e}")