Update app.py

app.py

@@ -4,6 +4,7 @@ from jsonschema import validate, ValidationError
 import re
 import networkx as nx
 import matplotlib.pyplot as plt
+import io
 
 # Mock existing resources (customizable)
 existing_resources = {
@@ -211,69 +212,87 @@ def simulate_deployment(arm_template):
 
 # Streamlit UI
 st.title("Comprehensive ARM Template Simulator")
-st.subheader("Paste your ARM template below to simulate its deployment.")
+st.subheader("Upload your ARM template or paste it below to simulate its deployment.")
+
+# File uploader
+uploaded_file = st.file_uploader("Choose an ARM template file", type=["json"])
 
 # Input box for ARM template
-template_input = st.text_area("Paste ARM Template JSON here:", height=300)
+template_input = st.text_area("Or paste ARM Template JSON here:", height=300)
+
+# Function to read file contents
+def read_file_contents(file):
+    return file.getvalue().decode("utf-8")
 
 # 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, system_diagram, message = simulate_deployment(template_input)
-        
-        st.subheader("Simulation Results:")
-        st.write(message)
+    # Check if a file was uploaded
+    if uploaded_file is not None:
+        # Read the contents of the uploaded file
+        file_contents = read_file_contents(uploaded_file)
+        # Use the file contents for simulation
+        template_to_simulate = file_contents
+    elif template_input:
+        # Use the pasted input for simulation
+        template_to_simulate = template_input
+    else:
+        st.error("Please either upload a file or paste an ARM template to simulate.")
+        st.stop()
 
-        if resources_to_create or resources_to_update or resources_to_delete:
-            st.write("### Resources to be Created:")
-            if resources_to_create:
-                st.write(resources_to_create)
-            else:
-                st.write("No new resources will be created.")
+    # Proceed with simulation
+    resources_to_create, resources_to_update, resources_to_delete, resource_details, estimated_cost, dependency_graph, lint_results, system_diagram, message = simulate_deployment(template_to_simulate)
+    
+    st.subheader("Simulation Results:")
+    st.write(message)
 
-            st.write("### Resources to be Updated:")
-            if resources_to_update:
-                st.write(resources_to_update)
-            else:
-                st.write("No resources will be updated.")
+    if resources_to_create or resources_to_update or resources_to_delete:
+        st.write("### Resources to be Created:")
+        if resources_to_create:
+            st.write(resources_to_create)
+        else:
+            st.write("No new resources will be created.")
 
-            st.write("### Resources to be Deleted:")
-            if resources_to_delete:
-                st.write(resources_to_delete)
-            else:
-                st.write("No resources will be deleted.")
-
-            st.write("### Detailed Resource Information:")
-            for resource in resource_details:
-                st.write(f"**Name:** {resource['name']}")
-                st.write(f"**Type:** {resource['type']}")
-                st.write(f"**Location:** {resource['location']}")
-                st.write("**Properties:**")
-                st.json(resource['properties'])
-                st.write("---")
-
-            st.write(f"### Estimated Monthly Cost: ${estimated_cost}")
-
-            st.write("### Resource Dependency Graph:")
-            if dependency_graph:
-                fig = plot_dependency_graph(dependency_graph)
-                st.pyplot(fig)
-            else:
-                st.write("No dependencies found between resources.")
+        st.write("### Resources to be Updated:")
+        if resources_to_update:
+            st.write(resources_to_update)
+        else:
+            st.write("No resources will be updated.")
 
-            st.write("### System Diagram:")
-            if system_diagram:
-                st.mermaid(system_diagram)
-            else:
-                st.write("Unable to generate system diagram.")
+        st.write("### Resources to be Deleted:")
+        if resources_to_delete:
+            st.write(resources_to_delete)
+        else:
+            st.write("No resources will be deleted.")
+
+        st.write("### Detailed Resource Information:")
+        for resource in resource_details:
+            st.write(f"**Name:** {resource['name']}")
+            st.write(f"**Type:** {resource['type']}")
+            st.write(f"**Location:** {resource['location']}")
+            st.write("**Properties:**")
+            st.json(resource['properties'])
+            st.write("---")
+
+        st.write(f"### Estimated Monthly Cost: ${estimated_cost}")
+
+        st.write("### Resource Dependency Graph:")
+        if dependency_graph:
+            fig = plot_dependency_graph(dependency_graph)
+            st.pyplot(fig)
+        else:
+            st.write("No dependencies found between resources.")
 
-            st.write("### Template Lint Results:")
-            if lint_results:
-                for result in lint_results:
-                    st.write(result)
-            else:
-                st.write("No linting issues found.")
+        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:
+                st.write(result)
         else:
-            st.write("No changes detected in the simulation.")
+            st.write("No linting issues found.")
     else:
-        st.error("Please paste an ARM template to simulate.")
+        st.write("No changes detected in the simulation.")