Update app.py

app.py

@@ -1,5 +1,4 @@
 import numpy as np
-import pandas as pd
 import matplotlib.pyplot as plt
 import streamlit as st
 
@@ -9,6 +8,7 @@ def calculate_relay_curve(tr_kva, tr_volt, ct_pr, ct_sec, fac, tms, i_f_fac, cur
     pick = round(fac * tr_curr / ct_pr, 3)
     i_f = round(i_f_fac * tr_curr * fac, 3)
 
+    # Curve coefficients
     if curve == 'IEC Normal Inverse':
         b_ = 0.14
         a_ = 0.02
@@ -25,13 +25,17 @@ def calculate_relay_curve(tr_kva, tr_volt, ct_pr, ct_sec, fac, tms, i_f_fac, cur
         st.error("Invalid Curve Selected.")
         return
 
+    # Calculate trip time
     trip_t = b_ * tms / (((i_f * (ct_sec / ct_pr) / (pick * ct_sec)) ** a_) - 1)
 
+    # Generate data for the relay curve
     x_curr = np.linspace(pick * ct_pr + 0.1 * pick * ct_pr, pick * ct_pr * 5, 100)
     y_time = b_ * tms / (((x_curr * (ct_sec / ct_pr) / (pick * ct_sec)) ** a_) - 1)
 
+    # Fault time calculation
     fault_time = b_ * tms / (((i_f * (ct_sec / ct_pr) / (pick * ct_sec)) ** a_) - 1)
 
+    # Display results
     st.write('### Results:')
     st.write('**Rated Current:**', tr_curr, 'A')
     st.write('**Pickup:**', pick, 'x In')
@@ -40,18 +44,14 @@ def calculate_relay_curve(tr_kva, tr_volt, ct_pr, ct_sec, fac, tms, i_f_fac, cur
 
     # Plot the relay curve
     fig, ax = plt.subplots()
-    ax.plot(x_curr, y_time)
-    ax.plot(i_f, b_ * tms / (((i_f * (ct_sec / ct_pr) / (pick * ct_sec)) ** a_) - 1), 'o')
-    ax.xlabel('Current (Amp)')
-    ax.ylabel('Time (sec)')
-    ax.title(f"Relay Curve: {curve}")
+    ax.plot(x_curr, y_time, label="Relay Curve")
+    ax.plot(i_f, fault_time, 'ro', label="Fault Point")
     ax.set_xlabel('Current (Amp)')
     ax.set_ylabel('Time (sec)')
     ax.set_title(f"Relay Curve: {curve}")
     ax.legend()
     st.pyplot(fig)
 
-
 # Streamlit app layout
 st.title("Relay Curve Calculator")
 st.sidebar.header("Input Parameters")
@@ -69,4 +69,3 @@ curve = st.sidebar.selectbox("Relay Curve Type", ['IEC Normal Inverse', 'IEC Ver
 # Run the calculation when the button is clicked
 if st.sidebar.button("Calculate"):
     calculate_relay_curve(tr_kva, tr_volt, ct_pr, ct_sec, fac, tms, i_f_fac, curve)
-