Update app.py

app.py

@@ -1,19 +1,19 @@
 import streamlit as st
 
 # Function to calculate pump parameters
-def calculate_parameters(option, flow_rate, head, density, efficiency, speed, impeller_diameter, height):
+def calculate_parameters(option, inputs):
     try:
-        flow_rate = float(flow_rate)  # m³/h
-        head = float(head)  # meters
-        density = float(density)  # kg/m³
-        efficiency = float(efficiency) / 100  # Percentage to decimal
-        speed = float(speed)  # RPM
-        impeller_diameter = float(impeller_diameter)  # meters
-        height = float(height)  # meters
-        
         g = 9.81  # Gravitational acceleration (m/s²)
-        
-        # Initialize results
+
+        # Extract relevant inputs
+        flow_rate = float(inputs.get("Flow Rate", 0))  # m³/h
+        head = float(inputs.get("Head", 0))  # meters
+        density = float(inputs.get("Density", 0))  # kg/m³
+        efficiency = float(inputs.get("Efficiency", 0)) / 100  # % to decimal
+        speed = float(inputs.get("Speed", 0))  # RPM
+        impeller_diameter = float(inputs.get("Impeller Diameter", 0))  # meters
+        height = float(inputs.get("Height", 0))  # meters
+
         result = {}
 
         if option == "Hydraulic Power":
@@ -54,16 +54,6 @@ st.title("Pump Design and Performance Calculator")
 st.subheader("Developed by Ahmad Hassan")
 st.caption("Supervised by Dr. Hidayatullah Mahar")
 
-# 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")
-speed = st.text_input("Pump Speed (RPM):", "2900")
-impeller_diameter = st.text_input("Impeller Diameter (meters):", "0.3")
-height = st.text_input("Transport Height (meters):", "10")
-
 # Dropdown Menu for Quantity to Calculate
 st.header("Select Quantity to Calculate")
 calculation_option = st.selectbox(
@@ -77,11 +67,38 @@ calculation_option = st.selectbox(
     ]
 )
 
+# Dynamic Input Fields Based on Selection
+inputs = {}
+
+if calculation_option == "Hydraulic Power":
+    inputs["Flow Rate"] = st.text_input("Flow Rate (m³/h):", "100")
+    inputs["Head"] = st.text_input("Head (meters):", "50")
+    inputs["Density"] = st.text_input("Fluid Density (kg/m³):", "1000")
+
+elif calculation_option == "Input Power":
+    inputs["Flow Rate"] = st.text_input("Flow Rate (m³/h):", "100")
+    inputs["Head"] = st.text_input("Head (meters):", "50")
+    inputs["Density"] = st.text_input("Fluid Density (kg/m³):", "1000")
+    inputs["Efficiency"] = st.text_input("Pump Efficiency (%):", "75")
+
+elif calculation_option == "Efficiency":
+    inputs["Flow Rate"] = st.text_input("Flow Rate (m³/h):", "100")
+    inputs["Head"] = st.text_input("Head (meters):", "50")
+    inputs["Density"] = st.text_input("Fluid Density (kg/m³):", "1000")
+    inputs["Efficiency"] = st.text_input("Actual Efficiency (%):", "70")
+
+elif calculation_option == "Impeller Speed":
+    inputs["Head"] = st.text_input("Head (meters):", "50")
+    inputs["Impeller Diameter"] = st.text_input("Impeller Diameter (meters):", "0.3")
+    inputs["Speed"] = st.text_input("Pump Speed (RPM):", "2900")
+
+elif calculation_option == "Pump Specifications for Height":
+    inputs["Height"] = st.text_input("Transport Height (meters):", "10")
+    inputs["Density"] = st.text_input("Fluid Density (kg/m³):", "1000")
+
 # Button to Calculate
 if st.button("Calculate"):
-    result = calculate_parameters(
-        calculation_option, flow_rate, head, density, efficiency, speed, impeller_diameter, height
-    )
+    result = calculate_parameters(calculation_option, inputs)
     if isinstance(result, dict):
         st.success("Calculation Results:")
         for key, value in result.items():
@@ -93,9 +110,9 @@ if st.button("Calculate"):
 st.info(
     """
     ### Notes:
-    - **Flow Rate**: Volume of fluid pumped per hour.
-    - **Head**: Pressure increase produced by the pump.
-    - **Efficiency**: Mechanical efficiency of the pump.
-    - **Pump Specifications for Height**: Estimates specifications required to transport fluid to a given height.
+    - Relevant inputs will appear dynamically based on the selected parameter.
+    - For example:
+        - To calculate Hydraulic Power, you need Flow Rate, Head, and Density.
+        - To calculate Impeller Speed, you need Head, Impeller Diameter, and Speed.
     """
 )