Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import time
+import matplotlib.pyplot as plt
 
 st.set_page_config(page_title="Structural Designer (Imperial Units)", layout="centered")
 st.title("🏗️ Structural Component Designer (Layman-Friendly)")
@@ -36,28 +37,60 @@ else:
     sbc = sbc_defaults[soil_type]
     st.markdown(f"**Estimated SBC:** {sbc} psf based on soil type")
 
-# 🔄 Mocked Grok API response
+# 🔄 Mocked Grok API logic
 def call_mock_grok_api(inputs):
-    time.sleep(1.5)  # simulate delay
+    time.sleep(1.5)
     return {
         "slab": {
             "type": "Two-way" if inputs["slab_length"] / inputs["slab_width"] < 2 else "One-way",
-            "thickness": 6 + 0.1 * inputs["floors"]
+            "thickness": 6 + 0.1 * inputs["floors"],
+            "bars_top": 6,
+            "bars_bottom": 6
         },
         "beam": {
             "depth": 12 + (5 if inputs["carries_heavy_items"] == "Yes" else 0),
-            "reinforcement": "#4 bars @ 6\" c/c top and bottom"
+            "reinforcement": "#4 bars @ 6\" c/c top and bottom",
+            "top_bars": 2,
+            "bottom_bars": 2,
+            "stirrups": 5
         },
         "column": {
             "size": "12x12",
-            "reinforcement": "4 #5 bars with ties @ 8\" c/c"
+            "reinforcement": "4 #5 bars with ties @ 8\" c/c",
+            "bars": 4
         },
         "footing": {
             "size": f"{round(1.5 + 0.1 * inputs['floors'], 2)} x {round(1.5 + 0.1 * inputs['floors'], 2)} ft",
-            "reinforcement": "#4 bars @ 6\" both ways"
+            "reinforcement": "#4 bars @ 6\" both ways",
+            "bars_x": 5,
+            "bars_y": 5
         }
     }
 
+# Drawing functions
+def draw_rebars(title, x_bars, y_bars=None):
+    fig, ax = plt.subplots()
+    ax.set_aspect('equal')
+    ax.set_xlim(0, 10)
+    ax.set_ylim(0, 10)
+    ax.axis('off')
+
+    if y_bars is None:  # 1D bars
+        spacing = 8 / (x_bars + 1)
+        for i in range(x_bars):
+            ax.plot([1 + spacing * (i + 1)]*2, [2, 8], 'r', linewidth=2)
+    else:  # Grid (2D) bars
+        spacing_x = 8 / (x_bars + 1)
+        spacing_y = 8 / (y_bars + 1)
+        for i in range(x_bars):
+            ax.plot([1 + spacing_x * (i + 1)]*2, [2, 8], 'r', linewidth=1.5)
+        for j in range(y_bars):
+            ax.plot([2, 8], [1 + spacing_y * (j + 1)]*2, 'b', linewidth=1.5)
+
+    ax.set_title(title)
+    st.pyplot(fig)
+
+# Run design
 if st.button("🧠 Run Design"):
     inputs = {
         "building_type": building_type,
@@ -79,18 +112,21 @@ if st.button("🧠 Run Design"):
     st.subheader("🔹 Slab Design")
     st.write(f"**Type:** {grok_response['slab']['type']}")
     st.write(f"**Thickness:** {grok_response['slab']['thickness']} inches")
+    draw_rebars("Slab Reinforcement", grok_response['slab']['bars_top'], grok_response['slab']['bars_bottom'])
 
     st.subheader("🔹 Beam Design")
     st.write(f"**Recommended Depth:** {grok_response['beam']['depth']} inches")
     st.write(f"**Reinforcement:** {grok_response['beam']['reinforcement']}")
+    draw_rebars("Beam Reinforcement", grok_response['beam']['bottom_bars'])
 
     st.subheader("🔹 Column Design")
     st.write(f"**Column Size:** {grok_response['column']['size']} inches")
     st.write(f"**Reinforcement:** {grok_response['column']['reinforcement']}")
+    draw_rebars("Column Vertical Bars", grok_response['column']['bars'])
 
     st.subheader("🔹 Footing Design")
     st.write(f"**Footing Size:** {grok_response['footing']['size']}")
     st.write(f"**Reinforcement:** {grok_response['footing']['reinforcement']}")
+    draw_rebars("Footing Reinforcement", grok_response['footing']['bars_x'], grok_response['footing']['bars_y'])
 
-    st.success("✅ Design completed using mock Grok engine.")
-
+    st.success("✅ Design completed with visual detailing.")