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 configuration
 st.set_page_config(
@@ -17,7 +19,7 @@ st.sidebar.markdown(
     - Flow Rate (\( Q \)) in cubic meters per second.
     - Velocity (\( V \)) in meters per second.
     
-    Developed for engineers and fluid mechanics enthusiasts! 🌊
+    Also visualize pipe diameters over a range of flow rates and velocities. 🌊
     """
 )
 
@@ -64,6 +66,51 @@ with col2:
     else:
         st.info("Click the button to calculate the pipe diameter.")
 
+# Graphical visualization
+st.header("Graphical Visualization 📊")
+st.markdown(
+    """
+    Visualize how pipe diameter varies with different **flow rates** and **velocities**.
+    """
+)
+
+# User input for visualization range
+col3, col4 = st.columns(2)
+with col3:
+    flow_rate_range = st.slider(
+        "Select Flow Rate Range (m³/s):",
+        min_value=0.1, 
+        max_value=5.0, 
+        value=(0.5, 2.5),
+        step=0.1
+    )
+with col4:
+    velocity_range = st.slider(
+        "Select Velocity Range (m/s):",
+        min_value=0.1, 
+        max_value=10.0, 
+        value=(1.0, 5.0),
+        step=0.5
+    )
+
+# Generate and plot graph
+flow_rates = np.linspace(flow_rate_range[0], flow_rate_range[1], 50)
+velocities = np.linspace(velocity_range[0], velocity_range[1], 50)
+
+flow_grid, velocity_grid = np.meshgrid(flow_rates, velocities)
+diameter_grid = np.sqrt((4 * flow_grid) / (np.pi * velocity_grid)) * 100  # Diameter in cm
+
+fig, ax = plt.subplots(figsize=(8, 6))
+contour = ax.contourf(flow_grid, velocity_grid, diameter_grid, cmap="coolwarm")
+cbar = fig.colorbar(contour, ax=ax)
+cbar.set_label("Pipe Diameter (cm)")
+ax.set_title("Pipe Diameter vs. Flow Rate and Velocity")
+ax.set_xlabel("Flow Rate (m³/s)")
+ax.set_ylabel("Velocity (m/s)")
+ax.grid(True)
+
+st.pyplot(fig)
+
 # Footer with additional info
 st.sidebar.markdown("---")
 st.sidebar.markdown(
@@ -73,3 +120,4 @@ st.sidebar.markdown(
     """
 )
 
+