Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import math
+import numpy as np
+import matplotlib.pyplot as plt
 
 # Set page config with an icon and title
 st.set_page_config(page_title="Pipe Sizing Helper", page_icon="🛠️")
@@ -57,3 +59,26 @@ if st.button("Generate Recommended Pipe Diameter"):
         st.success(f"Recommended Pipe Diameter: {diameter_mm:.2f} mm")
     else:
         st.error("Permissible velocity must be greater than zero.")
+
+# Graphical visualization section
+st.header("Graphical Visualization")
+st.write("Explore how pipe diameter changes with flow rate and velocity.")
+
+# Generate ranges for flow rate and velocity
+flow_rates = np.linspace(0.1, 10, 100)  # Flow rates from 0.1 to 10 m³/s
+velocities = np.linspace(0.1, 5, 50)  # Velocities from 0.1 to 5 m/s
+
+# Create a grid for flow rates and velocities
+flow_rate_mesh, velocity_mesh = np.meshgrid(flow_rates, velocities)
+pipe_diameters = np.sqrt(4 * flow_rate_mesh / (np.pi * velocity_mesh)) * 1000  # Diameter in mm
+
+# Plotting
+fig, ax = plt.subplots(figsize=(8, 6))
+c = ax.contourf(flow_rate_mesh, velocity_mesh, pipe_diameters, cmap="viridis", levels=20)
+plt.colorbar(c, label="Pipe Diameter (mm)")
+ax.set_title("Pipe Diameter Heatmap")
+ax.set_xlabel("Flow Rate (m³/s)")
+ax.set_ylabel("Velocity (m/s)")
+
+# Display plot in Streamlit
+st.pyplot(fig)