Update app.py

app.py

@@ -2,64 +2,77 @@ import streamlit as st
 
 # Constants
 GRAVITY = 9.81  # Acceleration due to gravity in m/s²
-NPSH_REQUIRED = 5  # Assumed NPSH required in meters (simplified)
+ATM_PRESSURE = 101.325  # Standard atmospheric pressure in kPa
 EFFICIENCY = 0.7  # Default pump efficiency (70%)
 
 # Function to calculate pump requirements
-def calculate_pump_requirements(selected_distance, distance_value, fluid_density):
+def calculate_pump_requirements(selected_distance, distance_value, fluid_density, viscosity, pipe_diameter, altitude, fluid_temp):
     try:
         # Convert inputs to numeric
         distance_value = float(distance_value)  # meters
         fluid_density = float(fluid_density)  # kg/m³
+        viscosity = float(viscosity)  # cP (centipoise)
+        pipe_diameter = float(pipe_diameter)  # mm
+        altitude = float(altitude)  # meters
+        fluid_temp = float(fluid_temp)  # °C
 
-        # Flow rate estimation (simplified formula for transport demands)
-        flow_rate = distance_value * 0.05  # m³/h (arbitrary scaling)
+        # Convert pipe diameter from mm to meters
+        pipe_diameter_m = pipe_diameter / 1000
+
+        # Adjust fluid properties based on temperature (simplified model)
+        density_temp_factor = 1 - ((fluid_temp - 25) * 0.0001)  # Example adjustment
+        adjusted_density = fluid_density * density_temp_factor
+
+        # Calculate flow velocity (assumed flow rate based on transport demand)
+        flow_rate = distance_value * 0.05  # m³/h (simplified)
+        flow_velocity = (flow_rate / 3600) / (3.14 * (pipe_diameter_m / 2) ** 2)  # m/s
 
         # Head Calculation
-        if selected_distance == "Vertical Distance (Height)":
-            static_head = distance_value
-        else:  # Horizontal distance with friction loss
-            static_head = 0
-            friction_loss = distance_value * 0.02  # Simplified friction loss in meters
-            total_head = friction_loss
+        static_head = distance_value if selected_distance == "Vertical Distance (Height)" else 0
+        friction_loss = (distance_value * viscosity * flow_velocity) / (1000 * pipe_diameter_m)  # Simplified
         total_head = static_head + (friction_loss if selected_distance == "Horizontal Distance" else 0)
 
         # Hydraulic power
-        hydraulic_power = (flow_rate / 3600) * fluid_density * GRAVITY * total_head / 1000  # kW
+        hydraulic_power = (flow_rate / 3600) * adjusted_density * GRAVITY * total_head / 1000  # kW
 
         # Input power
         input_power = hydraulic_power / EFFICIENCY  # Adjusted for efficiency
 
         # Cavitation Check
-        npsha = static_head - NPSH_REQUIRED  # Simplified for water at ambient conditions
-        cavitation_risk = "Yes" if npsha < 0 else "No"
+        npsha = static_head - (ATM_PRESSURE - (altitude * 0.012)) / 9.81  # Simplified
+        cavitation_risk = "Yes" if npsha < 5 else "No"
 
-        # Suggested Pump Type
+        # Suggested Pump Type and Material
         if total_head > 50:
             pump_type = "Centrifugal Pump"
+            material = "Stainless Steel"
         elif flow_rate > 100:
             pump_type = "Axial Flow Pump"
+            material = "Plastic or Composite Material"
         else:
             pump_type = "Mixed Flow Pump"
+            material = "Cast Iron"
 
         # Results
         results = {
             "Selected Distance": selected_distance,
             "Flow Rate (m³/h)": round(flow_rate, 2),
+            "Flow Velocity (m/s)": round(flow_velocity, 2),
             "Total Head (m)": round(total_head, 2),
             "Hydraulic Power (kW)": round(hydraulic_power, 2),
             "Input Power (kW)": round(input_power, 2),
             "Net Positive Suction Head Available (NPSHa)": round(npsha, 2),
             "Cavitation Risk": cavitation_risk,
             "Recommended Pump Type": pump_type,
+            "Recommended Material": material,
         }
         return results
     except ValueError:
-        return {"Error": "Invalid input. Please enter numeric values for distance and density."}
+        return {"Error": "Invalid input. Please enter numeric values for all fields."}
 
 
 # Streamlit App GUI
-st.title("Pump Design and Selection Tool")
+st.title("Enhanced Pump Design and Selection Tool")
 st.subheader("Developed by Ahmad Hassan")
 st.caption("Supervised by Dr. Hidayatullah Mahar")
 
@@ -81,6 +94,12 @@ elif fluid_type == "Diesel":
 else:
     density = st.text_input("Enter Custom Fluid Density (kg/m³):", "1000")
 
+# Input Parameters
+viscosity = st.text_input("Enter Fluid Viscosity (cP):", "1")  # Centipoise
+pipe_diameter = st.text_input("Enter Pipe Diameter (mm):", "50")  # mm
+fluid_temp = st.text_input("Enter Fluid Temperature (°C):", "25")
+altitude = st.text_input("Enter Altitude Above Sea Level (meters):", "0")
+
 # Distance Type Selection
 selected_distance = st.selectbox(
     "Select Distance Type:",
@@ -92,7 +111,7 @@ distance_value = st.text_input(f"Enter {selected_distance} (meters):", "50")
 
 # Calculate Button
 if st.button("Calculate Pump Requirements"):
-    results = calculate_pump_requirements(selected_distance, distance_value, density)
+    results = calculate_pump_requirements(selected_distance, distance_value, density, viscosity, pipe_diameter, altitude, fluid_temp)
     if "Error" in results:
         st.error(results["Error"])
     else:
@@ -104,8 +123,9 @@ if st.button("Calculate Pump Requirements"):
 st.info(
     """
     ### How It Works:
-    - **Fluid Type**: Select the fluid or provide a custom density.
+    - **Fluid Type**: Select or provide custom density.
     - **Distance Type**: Choose between vertical height or horizontal pipe length.
-    - **Pump Parameters**: Calculates head, flow rate, power, cavitation risk, and recommends pump type.
+    - **Additional Parameters**: Adjust for fluid viscosity, pipe diameter, and altitude.
+    - **Pump Parameters**: Calculates flow velocity, head, power, cavitation risk, and recommends pump type/material.
     """
 )