Update app.py

app.py

@@ -1,36 +1,54 @@
 import streamlit as st
+import matplotlib.pyplot as plt
+import numpy as np
 
 def calculate_steel_required(total_area, unit, steel_length, box_width, box_length):
-    # Convert everything to meters for consistency
     conversion = {'m': 1, 'cm': 0.01, 'mm': 0.001}
-    total_area_m2 = total_area * (conversion[unit] ** 2)  # Convert area to square meters
-    steel_length_m = steel_length * conversion[unit]  # Convert steel length to meters
-    box_width_m = box_width * conversion[unit]  # Convert box width to meters
-    box_length_m = box_length * conversion[unit]  # Convert box length to meters
+    total_area_m2 = total_area * (conversion[unit] ** 2)
+    steel_length_m = steel_length * conversion[unit]
+    box_width_m = box_width * conversion[unit]
+    box_length_m = box_length * conversion[unit]
     
-    # Number of steel rods required
     num_steel_width = int((total_area_m2 ** 0.5) / box_width_m) + 1
     num_steel_length = int((total_area_m2 ** 0.5) / box_length_m) + 1
     total_steel_rods = num_steel_width + num_steel_length
     
     return total_steel_rods
 
-# Streamlit UI
+def subtract_steel_rods(total_rods, rods_to_subtract):
+    return max(0, total_rods - rods_to_subtract)
+
+def plot_steel_grid(total_area, unit, box_width, box_length):
+    conversion = {'m': 1, 'cm': 0.01, 'mm': 0.001}
+    area_side = (total_area ** 0.5) * conversion[unit]
+    box_width_m = box_width * conversion[unit]
+    box_length_m = box_length * conversion[unit]
+    
+    fig, ax = plt.subplots(figsize=(6,6))
+    ax.set_xticks(np.arange(0, area_side, box_width_m))
+    ax.set_yticks(np.arange(0, area_side, box_length_m))
+    ax.grid(True, linestyle='--', linewidth=0.5)
+    ax.set_xlim(0, area_side)
+    ax.set_ylim(0, area_side)
+    ax.set_title("Graphical Representation of Steel Rods")
+    st.pyplot(fig)
+
 st.title("Steel Rods Calculator")
 st.markdown("### Calculate the number of steel rods required based on the given area and box size.")
 
-# Select measurement unit
 unit = st.selectbox("Select measurement unit", ["m", "cm", "mm"], index=1)
 
-# Input fields
 total_area = st.number_input(f"Enter total area in {unit}²", min_value=0.0, format="%.2f")
 steel_length = st.number_input(f"Enter length of one steel rod in {unit}", min_value=0.0, format="%.2f")
 box_width = st.number_input(f"Enter width of the box in {unit}", min_value=0.0, format="%.2f")
 box_length = st.number_input(f"Enter length of the box in {unit}", min_value=0.0, format="%.2f")
+rods_to_subtract = st.number_input("Enter number of rods to subtract", min_value=0, format="%d")
 
 if st.button("Calculate Steel Required"):
     if total_area > 0 and steel_length > 0 and box_width > 0 and box_length > 0:
-        result = calculate_steel_required(total_area, unit, steel_length, box_width, box_length)
-        st.success(f"Total steel rods required: {result}")
+        total_rods = calculate_steel_required(total_area, unit, steel_length, box_width, box_length)
+        adjusted_rods = subtract_steel_rods(total_rods, rods_to_subtract)
+        st.success(f"Total steel rods required: {adjusted_rods}")
+        plot_steel_grid(total_area, unit, box_width, box_length)
     else:
         st.error("Please enter valid positive numbers for all inputs.")