Update app.py

app.py

@@ -76,7 +76,9 @@ def calculate_pump_requirements(selected_distance, distance_value, fluid_propert
         total_head = static_head + (friction_loss if selected_distance == "Horizontal Distance" else 0)
 
         hydraulic_power = (flow_rate / 3600) * density * GRAVITY * total_head / 1000  # kW
-        pump_type = "Centrifugal Pump" if total_head > 10 else "Positive Displacement Pump"  # Simplified decision logic
+
+        # Select pump type based on fluid density and type (liquid or gas)
+        pump_type = "Centrifugal Pump" if total_head > 10 and density > 1 else "Positive Displacement Pump"
 
         cavitation_risk = "Possible" if total_head < 10 else "Low Risk"
 
@@ -86,15 +88,12 @@ def calculate_pump_requirements(selected_distance, distance_value, fluid_propert
         # Pump material selection (based on fluid)
         pump_material = "Stainless Steel" if fluid_properties["density"] < 1000 else "Cast Iron"
 
-        # Required power (hydraulic power already gives this, no need for input)
-        required_power = hydraulic_power  # kW
-
+        # Return pump requirements
         return {
             "Flow Rate (m³/h)": flow_rate,
             "Flow Velocity (m/s)": flow_velocity,
             "Total Head (m)": total_head,
             "Hydraulic Power (kW)": hydraulic_power,
-            "Required Power (kW)": required_power,  # Display required power
             "Pump Type": pump_type,
             "Cavitation Risk": cavitation_risk,
             "Impeller Size (m)": impeller_size,
@@ -132,12 +131,11 @@ if st.button("Calculate Pump Requirements"):
             st.markdown(f"**Flow Velocity**: {results['Flow Velocity (m/s)']} m/s")
             st.markdown(f"**Total Head**: {results['Total Head (m)']} meters")
             st.markdown(f"**Hydraulic Power**: {results['Hydraulic Power (kW)']} kW")
-            st.markdown(f"**Required Power**: {results['Required Power (kW)']} kW")  # Showing required power
             st.markdown(f"**Pump Type**: {results['Pump Type']}")
             st.markdown(f"**Cavitation Risk**: {results['Cavitation Risk']}")
             st.markdown(f"**Impeller Size**: {results['Impeller Size (m)']} m")
             st.markdown(f"**Pump Material**: {results['Pump Material']}")
         else:
-            st.error(results)
+            st.error(results)  # Display error if the input was invalid
     else:
-        st.error("Please enter all required fields (distance and pipe diameter).")
+        st.error("Please fill in all required fields.")