Update app.py

app.py

@@ -1,3 +1,40 @@
+import streamlit as st
+import ezdxf
+import tempfile
+import pandas as pd
+import os
+
+# ---------- Function to extract data from DXF ----------
+def extract_entities(uploaded_file):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".dxf") as tmp:
+        tmp.write(uploaded_file.read())
+        tmp_path = tmp.name
+
+    doc = ezdxf.readfile(tmp_path)
+    msp = doc.modelspace()
+
+    plan_data = {"rooms": [], "doors": [], "windows": []}
+    section_data = {"slabs": [], "beams": [], "columns": []}
+
+    for e in msp:
+        if e.dxftype() == "LWPOLYLINE" and "plan" in e.dxf.layer.lower():
+            points = e.get_points()
+            plan_data["rooms"].append({"points": points})
+        elif e.dxftype() == "LINE" and "door" in e.dxf.layer.lower():
+            plan_data["doors"].append({"start": e.dxf.start, "end": e.dxf.end})
+        elif e.dxftype() == "LINE" and "window" in e.dxf.layer.lower():
+            plan_data["windows"].append({"start": e.dxf.start, "end": e.dxf.end})
+        elif "section" in e.dxf.layer.lower():
+            if "slab" in e.dxf.layer.lower():
+                section_data["slabs"].append(e)
+            elif "beam" in e.dxf.layer.lower():
+                section_data["beams"].append(e)
+            elif "column" in e.dxf.layer.lower():
+                section_data["columns"].append(e)
+
+    return plan_data, section_data
+
+# ---------- Function to estimate quantities ----------
 def estimate_quantities(plan_data, section_data):
     items = []
 
@@ -107,3 +144,27 @@ def estimate_quantities(plan_data, section_data):
 
     df = pd.DataFrame(items)
     return df
+
+# ---------- Streamlit App UI ----------
+st.set_page_config(layout="centered")
+st.title("🏗️ AutoCAD Drawing Estimator (AU Standards)")
+
+uploaded_file = st.file_uploader("Upload AutoCAD DXF File", type=["dxf"])
+
+if uploaded_file:
+    with st.spinner("Reading and analyzing the drawing..."):
+        try:
+            plan_data, section_data = extract_entities(uploaded_file)
+            df = estimate_quantities(plan_data, section_data)
+
+            st.success("✅ Drawing processed and estimate generated.")
+            st.dataframe(df)
+
+            output_path = "Construction_Estimate.xlsx"
+            df.to_excel(output_path, index=False)
+
+            with open(output_path, "rb") as f:
+                st.download_button("📥 Download Estimate (Excel)", f, file_name=output_path)
+
+        except Exception as e:
+            st.error(f"Error processing file: {e}")