Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import matplotlib.pyplot as plt
 
 # Constants
 VOLTAGE_LEVELS = [220, 380, 440]
@@ -61,7 +62,21 @@ with col3:
 
 with col4:
     st.subheader("Load Distribution (kVA)")
-    st.pie_chart({"Used Load": total_load_kva, "Remaining Capacity": max(transformer_size - total_load_kva, 0) if transformer_size else 0})
+    if transformer_size:
+        # Pie chart with Matplotlib
+        labels = ["Used Load", "Remaining Capacity"]
+        sizes = [total_load_kva, max(transformer_size - total_load_kva, 0)]
+        colors = ["#FF9999", "#66B3FF"]
+        explode = (0.1, 0)  # Highlight the "Used Load" slice
+
+        fig, ax = plt.subplots()
+        ax.pie(
+            sizes, explode=explode, labels=labels, colors=colors, autopct="%1.1f%%", startangle=140
+        )
+        ax.axis("equal")  # Equal aspect ratio ensures the pie chart is circular.
+        st.pyplot(fig)
+    else:
+        st.warning("No transformer selected. Unable to display load distribution.")
 
 # Footer
 st.sidebar.markdown("---")