Update app.py

app.py

@@ -7,12 +7,23 @@ def create_graph_from_json_text(json_text):
     json_data = json.loads(json_text)
     
     graph = Digraph()
-    for key, value in json_data.items():
-        if isinstance(value, list):
-            for item in value:
-                graph.edge(key, item)
+    def add_nodes_and_edges(data, parent_key=None):
+        if isinstance(data, dict):
+            for key, value in data.items():
+                if parent_key is not None:
+                    graph.edge(str(parent_key), str(key))
+                add_nodes_and_edges(value, key)
+        elif isinstance(data, list):
+            for i, item in enumerate(data):
+                if parent_key is not None:
+                    graph.edge(str(parent_key), f"{parent_key}_{i}")
+                add_nodes_and_edges(item, f"{parent_key}_{i}")
         else:
-            graph.edge(key, value)
+            if parent_key is not None:
+                graph.edge(str(parent_key), str(data))
+    
+    add_nodes_and_edges(json_data)
+    
     return graph
 
 def main():