Spaces:
Build error
Build error
| 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"]) | |
| 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") | |