app.py

import gradio as gr

def pump_power(flow_rate, head, efficiency):
    """
    flow_rate: m^3/s
    head: m
    efficiency: decimal (0.7 for 70%)
    """
    rho = 1000  # water density kg/m³
    g = 9.81    # gravity m/s²
    try:
        P = (rho * g * flow_rate * head) / (efficiency if efficiency > 0 else 1)
        return f"Pump Power: {P/1000:.2f} kW"
    except Exception as e:
        return f"Error: {e}"

iface = gr.Interface(
    fn=pump_power,
    inputs=[
        gr.Number(label="Flow Rate (m³/s)"),
        gr.Number(label="Head (m)"),
        gr.Number(label="Efficiency (0-1)")
    ],
    outputs="text",
    title="Pump Power Calculator",
    description="Calculates pump power requirement for water flow."
)

if __name__ == "__main__":
    iface.launch()