Update app.py

app.py

@@ -2,29 +2,33 @@ import streamlit as st
 
 # Constants
 GRAVITY = 9.81  # Acceleration due to gravity in m/s²
-ATM_PRESSURE = 101.325  # Standard atmospheric pressure in kPa
 EFFICIENCY = 0.7  # Default pump efficiency (70%)
 
+# Viscosity and Density for Different Fluids (Predefined)
+FLUID_PROPERTIES = {
+    "Water": {"density": 1000, "viscosity": 1},  # kg/m³, cP
+    "Oil": {"density": 850, "viscosity": 100},
+    "Gasoline": {"density": 750, "viscosity": 0.6},
+    "Diesel": {"density": 830, "viscosity": 3},
+    "Custom": {"density": None, "viscosity": None},
+}
+
 # Function to calculate pump requirements
-def calculate_pump_requirements(selected_distance, distance_value, fluid_density, viscosity, pipe_diameter, altitude, fluid_temp):
+def calculate_pump_requirements(selected_distance, distance_value, fluid_properties, pipe_diameter):
     try:
+        # Extract fluid properties
+        density = fluid_properties["density"]
+        viscosity = fluid_properties["viscosity"]
+
         # 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
 
         # 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)
+        # Calculate flow rate based on transport demand (simplified)
+        flow_rate = distance_value * 0.05  # m³/h
         flow_velocity = (flow_rate / 3600) / (3.14 * (pipe_diameter_m / 2) ** 2)  # m/s
 
         # Head Calculation
@@ -33,15 +37,11 @@ def calculate_pump_requirements(selected_distance, distance_value, fluid_density
         total_head = static_head + (friction_loss if selected_distance == "Horizontal Distance" else 0)
 
         # Hydraulic power
-        hydraulic_power = (flow_rate / 3600) * adjusted_density * GRAVITY * total_head / 1000  # kW
+        hydraulic_power = (flow_rate / 3600) * density * GRAVITY * total_head / 1000  # kW
 
         # Input power
         input_power = hydraulic_power / EFFICIENCY  # Adjusted for efficiency
 
-        # Cavitation Check
-        npsha = static_head - (ATM_PRESSURE - (altitude * 0.012)) / 9.81  # Simplified
-        cavitation_risk = "Yes" if npsha < 5 else "No"
-
         # Suggested Pump Type and Material
         if total_head > 50:
             pump_type = "Centrifugal Pump"
@@ -61,8 +61,6 @@ def calculate_pump_requirements(selected_distance, distance_value, fluid_density
             "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,
         }
@@ -82,23 +80,16 @@ fluid_type = st.selectbox(
     ["Water", "Oil", "Gasoline", "Diesel", "Custom"]
 )
 
-# Assign Density Based on Fluid Type
-if fluid_type == "Water":
-    density = 1000  # kg/m³
-elif fluid_type == "Oil":
-    density = 850  # kg/m³
-elif fluid_type == "Gasoline":
-    density = 750  # kg/m³
-elif fluid_type == "Diesel":
-    density = 830  # kg/m³
-else:
+# Get fluid properties
+if fluid_type == "Custom":
     density = st.text_input("Enter Custom Fluid Density (kg/m³):", "1000")
+    viscosity = st.text_input("Enter Custom Fluid Viscosity (cP):", "1")
+    fluid_properties = {"density": float(density), "viscosity": float(viscosity)}
+else:
+    fluid_properties = FLUID_PROPERTIES[fluid_type]
 
 # 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(
@@ -111,7 +102,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, viscosity, pipe_diameter, altitude, fluid_temp)
+    results = calculate_pump_requirements(selected_distance, distance_value, fluid_properties, pipe_diameter)
     if "Error" in results:
         st.error(results["Error"])
     else:
@@ -123,9 +114,9 @@ if st.button("Calculate Pump Requirements"):
 st.info(
     """
     ### How It Works:
-    - **Fluid Type**: Select or provide custom density.
+    - **Fluid Type**: Select or provide custom density and viscosity.
     - **Distance Type**: Choose between vertical height or horizontal pipe length.
-    - **Additional Parameters**: Adjust for fluid viscosity, pipe diameter, and altitude.
-    - **Pump Parameters**: Calculates flow velocity, head, power, cavitation risk, and recommends pump type/material.
+    - **Additional Parameters**: Adjust for pipe diameter and distance.
+    - **Pump Parameters**: Calculates flow velocity, head, power, and recommends pump type/material.
     """
 )