Update app.py

app.py

@@ -1,88 +1,101 @@
 import streamlit as st
 
-# Function to calculate pump power, efficiency, and other performance parameters
-def calculate_pump_performance(flow_rate, head, density, efficiency, npsh, speed, impeller_diameter, fluid_viscosity):
+# Function to calculate pump parameters
+def calculate_parameters(option, flow_rate, head, density, efficiency, speed, impeller_diameter, height):
     try:
-        # Convert input to float
-        flow_rate = float(flow_rate)  # m3/h
+        flow_rate = float(flow_rate)  # m³/h
         head = float(head)  # meters
-        density = float(density)  # kg/m3
+        density = float(density)  # kg/m³
         efficiency = float(efficiency) / 100  # Percentage to decimal
-        npsh = float(npsh)  # meters
         speed = float(speed)  # RPM
         impeller_diameter = float(impeller_diameter)  # meters
-        fluid_viscosity = float(fluid_viscosity)  # cP
-
-        # Constants
-        g = 9.81  # Gravitational acceleration (m/s²)
-        
-        # Calculate Hydraulic Power (kW)
-        hydraulic_power = (flow_rate / 3600) * density * g * head / 1000  # kW
-        
-        # Calculate Brake Power (Input Power)
-        if efficiency > 0:
-            input_power = hydraulic_power / efficiency  # kW
-        else:
-            input_power = 0
+        height = float(height)  # meters
         
-        # NPSH Check
-        cavitation_risk = "Low" if npsh > (head * 0.1) else "High"
-        
-        # Reynolds Number for Flow
-        reynolds_number = (impeller_diameter * speed * fluid_viscosity) / 1000
+        g = 9.81  # Gravitational acceleration (m/s²)
         
-        # Display results
-        return {
-            "Hydraulic Power (kW)": round(hydraulic_power, 2),
-            "Input Power (kW)": round(input_power, 2),
-            "Efficiency (%)": round(efficiency * 100, 2),
-            "Cavitation Risk": cavitation_risk,
-            "Reynolds Number": round(reynolds_number, 2)
-        }
+        # Initialize results
+        result = {}
+
+        if option == "Hydraulic Power":
+            hydraulic_power = (flow_rate / 3600) * density * g * head / 1000  # kW
+            result["Hydraulic Power (kW)"] = round(hydraulic_power, 2)
+
+        elif option == "Input Power":
+            hydraulic_power = (flow_rate / 3600) * density * g * head / 1000  # kW
+            input_power = hydraulic_power / efficiency  # Input Power
+            result["Input Power (kW)"] = round(input_power, 2)
+
+        elif option == "Efficiency":
+            hydraulic_power = (flow_rate / 3600) * density * g * head / 1000
+            input_power = (hydraulic_power / efficiency) if efficiency > 0 else 0
+            actual_efficiency = (hydraulic_power / input_power) * 100 if input_power > 0 else 0
+            result["Efficiency (%)"] = round(actual_efficiency, 2)
+
+        elif option == "Impeller Speed":
+            impeller_speed = (head / (impeller_diameter ** 2)) * speed
+            result["Impeller Speed (RPM)"] = round(impeller_speed, 2)
+
+        elif option == "Pump Specifications for Height":
+            flow_needed = (density * g * height) / 1000  # Simplified flow for the given height
+            head_needed = height
+            result["Required Head (m)"] = round(head_needed, 2)
+            result["Flow Needed (m³/h)"] = round(flow_needed, 2)
+            result["Hydraulic Power (kW)"] = round((flow_needed / 3600) * density * g * height / 1000, 2)
+            result["Suggested Efficiency (%)"] = 70
+
+        return result
 
     except ValueError:
         return "Invalid input! Please ensure all inputs are numeric."
 
-# Streamlit GUI
+
+# Streamlit App GUI
 st.title("Pump Design and Performance Calculator")
 st.subheader("Developed by Ahmad Hassan")
 st.caption("Supervised by Dr. Hidayatullah Mahar")
 
-# Input fields
-st.header("Input Parameters:")
+# Input Section
+st.header("Input Parameters")
 flow_rate = st.text_input("Flow Rate (m³/h):", "100")
 head = st.text_input("Head (meters):", "50")
 density = st.text_input("Fluid Density (kg/m³):", "1000")
 efficiency = st.text_input("Pump Efficiency (%):", "75")
-npsh = st.text_input("Net Positive Suction Head (NPSH, meters):", "5")
 speed = st.text_input("Pump Speed (RPM):", "2900")
 impeller_diameter = st.text_input("Impeller Diameter (meters):", "0.3")
-fluid_viscosity = st.text_input("Fluid Viscosity (cP):", "1")
+height = st.text_input("Transport Height (meters):", "10")
 
-# Pump Type Selection
-st.header("Select Pump Type:")
-pump_type = st.radio("Pump Type:", ["Centrifugal", "Positive Displacement", "Axial Flow", "Mixed Flow"])
+# Dropdown Menu for Quantity to Calculate
+st.header("Select Quantity to Calculate")
+calculation_option = st.selectbox(
+    "Choose a Parameter to Calculate:",
+    [
+        "Hydraulic Power",
+        "Input Power",
+        "Efficiency",
+        "Impeller Speed",
+        "Pump Specifications for Height",
+    ]
+)
 
-# Calculate button
-if st.button("Calculate Performance"):
-    result = calculate_pump_performance(
-        flow_rate, head, density, efficiency, npsh, speed, impeller_diameter, fluid_viscosity
+# Button to Calculate
+if st.button("Calculate"):
+    result = calculate_parameters(
+        calculation_option, flow_rate, head, density, efficiency, speed, impeller_diameter, height
     )
     if isinstance(result, dict):
-        st.success("Pump Performance Results:")
+        st.success("Calculation Results:")
         for key, value in result.items():
             st.write(f"**{key}**: {value}")
     else:
         st.error(result)
 
-# Note Section
+# Notes Section
 st.info(
     """
     ### Notes:
     - **Flow Rate**: Volume of fluid pumped per hour.
     - **Head**: Pressure increase produced by the pump.
     - **Efficiency**: Mechanical efficiency of the pump.
-    - **NPSH**: Ensures no cavitation occurs.
-    - **Cavitation Risk**: Indicates the likelihood of cavitation (High/Low).
+    - **Pump Specifications for Height**: Estimates specifications required to transport fluid to a given height.
     """
 )