Update app.py

app.py

@@ -33,7 +33,7 @@ def calculate_relay_curve(tr_kva, tr_volt, ct_pr, ct_sec, fac, tms, i_f_fac, cur
         st.error("Error: Division by zero in fault trip time calculation. Check input values.")
         return
 
-    # Generate data for the relay curve
+    # Generate data for the relay curve (current vs time)
     x_curr = np.linspace(pick * ct_pr * 1.1, pick * ct_pr * 5, 500)
     y_time = []
     for x in x_curr:
@@ -51,14 +51,22 @@ def calculate_relay_curve(tr_kva, tr_volt, ct_pr, ct_sec, fac, tms, i_f_fac, cur
     st.write(f'**Trip Time:** {round(fault_time, 3)} sec')
 
     # Plot the relay curve
-    fig, ax = plt.subplots()
+    fig, ax = plt.subplots(figsize=(10, 6))
     ax.plot(x_curr, y_time, label="Relay Curve", color='blue')
+
+    # Mark the fault current and trip time on the graph
     ax.axvline(x=i_f, color='red', linestyle='--', label=f"Fault Current: {i_f} A")
     ax.axhline(y=fault_time, color='green', linestyle='--', label=f"Trip Time: {round(fault_time, 3)} sec")
+
+    # Adding labels and title
     ax.set_xlabel('Current (Amp)')
     ax.set_ylabel('Time (sec)')
     ax.set_title(f"Relay Curve: {curve}")
+
+    # Adding a legend
     ax.legend()
+
+    # Show the plot
     st.pyplot(fig)
 
 # Streamlit app layout