import streamlit as st import ezdxf import io import pandas as pd import matplotlib.pyplot as plt from fpdf import FPDF from tempfile import NamedTemporaryFile import matplotlib.patches as patches import tempfile # Constants BRICK_VOLUME_CFT = (9/12) * (4.5/12) * (3/12) CEMENT_SAND_RATIO = 1 / 6 SAND_RATIO = 5 / 6 CEMENT_DENSITY_KG_PER_CFT = 1440 / 35.3147 CEMENT_BAG_WEIGHT_KG = 50 DEFAULT_WALL_THICKNESS = 0.75 st.set_page_config(page_title="Building Estimator from CAD", layout="wide") st.title("🏗️ Auto Estimation from AutoCAD (.dxf) Drawing") uploaded_file = st.file_uploader("Upload your DXF file", type=["dxf"]) @st.cache_data def extract_geometry(file_bytes): try: text_buffer = io.TextIOWrapper(file_bytes, encoding='utf-8', errors='ignore') doc = ezdxf.read(text_buffer) except Exception as e: st.error(f"Error reading DXF file: {e}") return [], 0.75, [], 1, [] msp = doc.modelspace() rooms = [] room_shapes = [] openings = [] floors = 1 wall_thickness = DEFAULT_WALL_THICKNESS for entity in msp: if entity.dxftype() == "TEXT": content = entity.dxf.text.lower() if "wall thickness" in content: try: wall_thickness = float(content.split(":")[1].strip()) except: continue elif "floor" in content: try: floors = int(content.split(":")[1].strip()) except: continue elif entity.dxftype() == "LWPOLYLINE": if entity.closed and len(entity) == 4: points = entity.get_points() x_vals = [p[0] for p in points] y_vals = [p[1] for p in points] length = abs(max(x_vals) - min(x_vals)) / 12 width = abs(max(y_vals) - min(y_vals)) / 12 height = 10 if length > 2 and width > 2: rooms.append((length, width, height)) room_shapes.append((min(x_vals), min(y_vals), max(x_vals), max(y_vals))) else: openings.append(("opening", length, height)) return rooms, wall_thickness, openings, floors, room_shapes def estimate(rooms, wall_thickness, openings, floors): wall_volume = sum(2 * (l + w) * h * wall_thickness for l, w, h in rooms) opening_volume = sum(l * h * wall_thickness for _, l, h in openings) beam_volume = sum((l + 1) * 0.75 * 0.75 for _, l, _ in openings) net_volume = (wall_volume - opening_volume - beam_volume) * floors number_of_bricks = round((net_volume / BRICK_VOLUME_CFT) * 1.05) mortar_volume = net_volume * 0.25 cement_volume = mortar_volume * CEMENT_SAND_RATIO sand_volume = mortar_volume * SAND_RATIO cement_bag_volume_cft = CEMENT_BAG_WEIGHT_KG / CEMENT_DENSITY_KG_PER_CFT cement_bags = round(cement_volume / cement_bag_volume_cft) return number_of_bricks, sand_volume, cement_bags, { "Wall Volume (cft)": wall_volume, "Opening Volume (cft)": opening_volume, "Beam Volume (cft)": beam_volume, "Net Volume (cft)": net_volume, "Mortar Volume (cft)": mortar_volume, "Cement Volume (cft)": cement_volume, "Sand Volume (cft)": sand_volume, "Cement Bag Volume (cft)": cement_bag_volume_cft, "Brick Volume (cft)": BRICK_VOLUME_CFT } def draw_plan_image(room_shapes): fig, ax = plt.subplots() for x0, y0, x1, y1 in room_shapes: width = x1 - x0 height = y1 - y0 rect = patches.Rectangle((x0, y0), width, height, linewidth=1, edgecolor='black', facecolor='none') ax.add_patch(rect) ax.text(x0 + width / 2, y0 - 2, f"{round(width / 12, 1)} ft", ha='center', fontsize=8) ax.text(x1 + 2, y0 + height / 2, f"{round(height / 12, 1)} ft", va='center', fontsize=8, rotation=90) ax.set_aspect('equal') ax.axis('off') temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png") fig.savefig(temp_file.name, bbox_inches='tight') plt.close(fig) return temp_file.name def generate_pdf(data_dict, calc_details, room_shapes): image_path = draw_plan_image(room_shapes) pdf = FPDF() pdf.add_page() pdf.set_font("Arial", 'B', size=14) pdf.cell(200, 10, "Estimation Report", ln=True, align='C') pdf.ln(5) pdf.set_font("Arial", 'B', size=12) pdf.cell(200, 10, "Summary of Quantities", ln=True) pdf.set_font("Arial", '', size=11) for key, value in data_dict.items(): pdf.cell(200, 10, f"{key}: {value}", ln=True) pdf.ln(8) pdf.set_font("Arial", 'B', size=12) pdf.cell(200, 10, "Step-by-Step Calculations with Formulas", ln=True) pdf.set_font("Arial", '', size=10) pdf.multi_cell(0, 8, f""" 1. Wall Volume = 2 × (L + W) × H × t = {round(calc_details['Wall Volume (cft)'], 2)} cft 2. Opening Volume = L × H × t = {round(calc_details['Opening Volume (cft)'], 2)} cft 3. Beam Volume = (L+1) × 0.75 × 0.75 = {round(calc_details['Beam Volume (cft)'], 2)} cft 4. Net Volume = (Wall - Opening - Beam) × Floors = {round(calc_details['Net Volume (cft)'], 2)} cft 5. Brick Volume = 9\" × 4.5\" × 3\" = {round(calc_details['Brick Volume (cft)'], 4)} cft 6. Bricks = Net Volume / Brick Vol × 1.05 = {data_dict['Bricks Required']} 7. Mortar = Net Volume × 0.25 = {round(calc_details['Mortar Volume (cft)'], 2)} cft 8. Cement = Mortar × 1/6 = {round(calc_details['Cement Volume (cft)'], 2)} cft 9. Sand = Mortar × 5/6 = {round(calc_details['Sand Volume (cft)'], 2)} cft 10. Cement Bags = Cement / Bag Volume = {data_dict['Cement Bags']} """) pdf.ln(5) pdf.set_font("Arial", 'B', size=12) pdf.cell(200, 10, "2D Plan with Dimensions (in ft)", ln=True) pdf.image(image_path, x=10, y=None, w=180) tmp = NamedTemporaryFile(delete=False, suffix=".pdf") pdf.output(tmp.name) return tmp.name def plot_rooms(shapes): fig, ax = plt.subplots() for x0, y0, x1, y1 in shapes: width = (x1 - x0) height = (y1 - y0) rect = plt.Rectangle((x0, y0), width, height, fill=False, edgecolor='blue', linewidth=2) ax.add_patch(rect) ax.set_title("🗏️ 2D Floor Plan") ax.set_aspect("equal") ax.axis("off") st.pyplot(fig) if uploaded_file: file_bytes = io.BytesIO(uploaded_file.read()) rooms, wall_thickness, openings, floors, room_shapes = extract_geometry(file_bytes) st.success(f"✔️ Parsed {len(rooms)} rooms and {len(openings)} openings across {floors} floor(s).") st.write(f"📏 Wall Thickness: {wall_thickness} ft") bricks, sand, cement, calc_details = estimate(rooms, wall_thickness, openings, floors) st.subheader("📊 Room Dimensions") df_rooms = pd.DataFrame(rooms, columns=["Length (ft)", "Width (ft)", "Height (ft)"]) st.dataframe(df_rooms) st.subheader("🧱 Estimation Result") df_result = pd.DataFrame({ "Item": ["Bricks", "Sand (cft)", "Cement (bags)"], "Quantity": [bricks, round(sand, 2), cement] }) st.dataframe(df_result) st.subheader("🖼️ 2D Floor Plan") plot_rooms(room_shapes) st.subheader("📄 Export") col1, col2 = st.columns(2) with col1: st.download_button("⬇️ Download Excel", df_result.to_csv(index=False).encode(), "estimates.csv", "text/csv") with col2: pdf_file = generate_pdf({ "Bricks Required": f"{bricks:,}", "Sand Volume": f"{sand:.2f} cft", "Cement Bags": f"{cement} bags" }, calc_details, room_shapes) with open(pdf_file, "rb") as f: st.download_button("⬇️ Download PDF", f.read(), "estimates.pdf", "application/pdf")