Update app.py

app.py

@@ -1,8 +1,13 @@
 import streamlit as st
+import matplotlib.pyplot as plt
+from PIL import Image
 
 # Set page configuration
 st.set_page_config(page_title="Hydraulic Pump Power Calculator", page_icon="\U0001F6E0", layout="centered")
 
+# Load engineering icon
+icon = Image.open("engineering_icon.png")
+
 # Unit conversion factors
 UNIT_CONVERSIONS = {
     "Displacement": {"cc/rev": 1, "in^3/rev": 16.387},
@@ -32,7 +37,21 @@ def calculate_hydraulic_parameters(displacement, pressure, rpm, torque, displace
 
     return displacement_cc, pressure_bar, rpm, torque_nm, power_displacement, power_torque
 
+# Generate chart
+@st.cache
+def generate_chart(displacement, pressure, rpm, torque):
+    terms = ["Displacement (cc/rev)", "Pressure (bar)", "RPM", "Torque (Nm)"]
+    values = [displacement, pressure, rpm, torque]
+
+    fig, ax = plt.subplots()
+    ax.bar(terms, values, color=['blue', 'green', 'red', 'orange'])
+    ax.set_title("Hydraulic Parameters")
+    ax.set_ylabel("Values")
+    plt.xticks(rotation=45)
+    return fig
+
 # Streamlit App
+st.image(icon, width=100)
 st.title("Hydraulic Pump Power Calculator")
 st.sidebar.header("Input Units")
 
@@ -62,6 +81,10 @@ if st.sidebar.button("Calculate"):
     st.write(f"Hydraulic Power based on Displacement: {power_displacement:.2f} kW" if power_displacement else "Power based on Displacement: Not calculable.")
     st.write(f"Hydraulic Power based on Torque: {power_torque:.2f} kW" if power_torque else "Power based on Torque: Not calculable.")
 
+    # Display chart
+    fig = generate_chart(displacement, pressure, rpm, torque)
+    st.pyplot(fig)
+
 # Hugging Face Space Deployment Notice
 st.sidebar.markdown("---")
 st.sidebar.write("This app is ready for deployment on Hugging Face Spaces using Streamlit.")