Update app.py

app.py

@@ -127,6 +127,26 @@ def lint_template(template):
     
     return lint_results
 
+def generate_system_diagram(resources):
+    diagram = ["graph TD"]
+    resource_nodes = {}
+    
+    # Create nodes for each resource
+    for i, resource in enumerate(resources):
+        resource_type = resource['type'].split('/')[-1]
+        node_id = f"R{i}"
+        resource_nodes[resource['name']] = node_id
+        diagram.append(f"{node_id}[{resource_type}<br/>{resource['name']}]")
+    
+    # Create connections based on dependencies
+    for resource in resources:
+        if 'dependsOn' in resource:
+            for dependency in resource['dependsOn']:
+                if dependency in resource_nodes:
+                    diagram.append(f"{resource_nodes[dependency]} --> {resource_nodes[resource['name']]}")
+    
+    return "\n".join(diagram)
+
 def simulate_deployment(arm_template):
     try:
         template = json.loads(arm_template)
@@ -134,7 +154,7 @@ def simulate_deployment(arm_template):
         # Validate template
         is_valid, validation_message = validate_template(template)
         if not is_valid:
-            return [], [], [], [], 0, [], validation_message
+            return [], [], [], [], 0, None, [], "", validation_message
 
         # Resolve expressions
         resolved_template = resolve_template_expressions(template)
@@ -165,7 +185,8 @@ def simulate_deployment(arm_template):
                 "name": resource_name,
                 "type": resource_type,
                 "location": location,
-                "properties": properties
+                "properties": properties,
+                "dependsOn": resource.get('dependsOn', [])
             })
 
         for resource_name in existing_resources:
@@ -181,9 +202,12 @@ def simulate_deployment(arm_template):
         # Lint template
         lint_results = lint_template(template)
 
-        return resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, "Simulation completed successfully."
+        # Generate system diagram
+        system_diagram = generate_system_diagram(resource_details)
+
+        return resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, system_diagram, "Simulation completed successfully."
     except json.JSONDecodeError:
-        return [], [], [], [], 0, None, [], "Invalid JSON format. Please check your ARM template."
+        return [], [], [], [], 0, None, [], "", "Invalid JSON format. Please check your ARM template."
 
 # Streamlit UI
 st.title("Comprehensive ARM Template Simulator")
@@ -195,7 +219,7 @@ template_input = st.text_area("Paste ARM Template JSON here:", height=300)
 # Button to simulate the evaluation
 if st.button("Simulate Template"):
     if template_input:
-        resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, message = simulate_deployment(template_input)
+        resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, system_diagram, message = simulate_deployment(template_input)
         
         st.subheader("Simulation Results:")
         st.write(message)
@@ -237,6 +261,12 @@ if st.button("Simulate Template"):
             else:
                 st.write("No dependencies found between resources.")
 
+            st.write("### System Diagram:")
+            if system_diagram:
+                st.mermaid(system_diagram)
+            else:
+                st.write("Unable to generate system diagram.")
+
             st.write("### Template Lint Results:")
             if lint_results:
                 for result in lint_results: