Update app.py

app.py

@@ -2,71 +2,78 @@ import streamlit as st
 from graphviz import Digraph
 import json
 
-def create_graph_from_json_text(json_text):
-    # Konvertieren des JSON-Texts in ein Python Dictionary
+def create_graph_from_json_text(json_text, graph=None, parent_node=None):
+    # Convert JSON text to a Python dictionary
     json_data = json.loads(json_text)
     
-    graph = Digraph()
-    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:
-            if parent_key is not None:
-                graph.edge(str(parent_key), str(data))
+    # Create a new Digraph if not provided
+    if graph is None:
+        graph = Digraph()
     
-    add_nodes_and_edges(json_data)
+    if isinstance(json_data, dict):
+        for key, value in json_data.items():
+            if isinstance(value, dict) or isinstance(value, list):
+                node_id = f"{parent_node}.{key}" if parent_node else key
+                graph.node(node_id, label=key)
+                if parent_node is not None:
+                    graph.edge(parent_node, node_id)
+                create_graph_from_json_text(json.dumps(value), graph, node_id)
+            else:
+                node_id = f"{parent_node}.{key}" if parent_node else key
+                graph.node(node_id, label=f"{key}: {value}")
+                if parent_node is not None:
+                    graph.edge(parent_node, node_id)
+    elif isinstance(json_data, list):
+        for index, item in enumerate(json_data):
+            node_id = f"{parent_node}[{index}]" if parent_node else f"[{index}]"
+            graph.node(node_id, label=f"[{index}]")
+            if parent_node is not None:
+                graph.edge(parent_node, node_id)
+            create_graph_from_json_text(json.dumps(item), graph, node_id)
     
     return graph
 
 def main():
     st.title("Graph Visualization with Graphviz and Streamlit")
     
-    # JSON-Text in Streamlit
+    # JSON text in Streamlit
     json_text = """
     {
-"id": "urn:example:aas:asset1",
-"administration": {
-"version": "2.1",
-"revision": "A",
-"validUntil": "2050-12-31T23:59:59Z"
-},
-"canonicalModel": {
-"id": "urn:example:model:asset1",
-"version": "1.0"
-},
-"submodels": [
-{
-"id": "urn:example:submodel:1",
-"elements": [
-{
-"id": "property1",
-"propertyType": "string",
-"value": "Value 1"
-},
-{
-"id": "property2",
-"propertyType": "integer",
-"value": 25
-}
-]
-}
-]
-}
+        "id": "urn:example:aas:asset1",
+        "administration": {
+            "version": "2.1",
+            "revision": "A",
+            "validUntil": "2050-12-31T23:59:59Z"
+        },
+        "canonicalModel": {
+            "id": "urn:example:model:asset1",
+            "version": "1.0"
+        },
+        "submodels": [
+            {
+                "id": "urn:example:submodel:1",
+                "elements": [
+                    {
+                        "id": "property1",
+                        "propertyType": "string",
+                        "value": "Value 1"
+                    },
+                    {
+                        "id": "property2",
+                        "propertyType": "integer",
+                        "value": 25
+                    }
+                ]
+            }
+        ]
+    }
     """
 
-    # Erstellen des Diagramms aus dem JSON-Text
+    # Create the diagram from the JSON text
     graph = create_graph_from_json_text(json_text)
 
-    # Anzeige des Diagramms in Streamlit
+    # Display the diagram in Streamlit
     st.graphviz_chart(graph.source)
 
 if __name__ == "__main__":
-    main()
+    main()