Update app.py

app.py

@@ -1,75 +1,130 @@
 import streamlit as st
 import pandas as pd
 import numpy as np
-import io
-import matplotlib.pyplot as plt
 import os
-from openai import OpenAI
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
+import matplotlib.pyplot as plt
+import io
+from openai import OpenAI
 
-# --- Page config ---
+# Set page config FIRST
 st.set_page_config(page_title="Construction Estimator", layout="centered")
 
-# --- Groq setup ---
+# Setup OpenAI/Groq client
 client = OpenAI(
     api_key=os.getenv("GROQ_API_KEY"),
     base_url="https://api.groq.com/openai/v1"
 )
 GROQ_MODEL = "llama3-8b-8192"
 
-# --- File loaders ---
-@st.cache_data
-def load_excel(file):
-    return pd.read_excel(file)
+# Embed chunks - keep for similarity search if needed in future
+@st.cache_resource
+def embed_chunks(chunks):
+    model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
+    embeddings = model.encode(chunks)
+    return embeddings, model
+
+# Generate estimate with Groq
+def generate_estimate(context, user_input):
+    prompt = f"""You are a construction estimator working in Pakistan. Using the following schedule of rates and quantities:
+
+{context}
 
-# --- BOQ Generator ---
-def generate_boq(quant_df, rate_df):
-    boq = quant_df.merge(rate_df, on='Item', how='left')
-    boq['Amount'] = boq['Quantity'] * boq['Rate']
-    return boq[['Item', 'Description', 'Quantity', 'Unit', 'Rate', 'Amount']]
+Generate a detailed BOQ estimate including item numbers, full descriptions, unit rates, quantities, and total amount in Rs for:
+{user_input}
 
-# --- Sketch Generator (placeholder or Replicate integration) ---
-def generate_sketch_prompt(rooms, baths, living, car_porch):
-    return f"Simple sketch showing {rooms} rooms, {baths} baths, {living} living, {car_porch} car porch"
+Present the result in a markdown table with columns: Item No, Description, Qty, Unit, Rate, Amount."""
+    
+    response = client.chat.completions.create(
+        model=GROQ_MODEL,
+        messages=[{"role": "user", "content": prompt}]
+    )
+    return response.choices[0].message.content
+
+# Sketch generator
+def draw_floor_plan(rooms, baths, living, car_porch, area):
+    total_spaces = rooms + baths + living + car_porch
+    cols = int(np.ceil(np.sqrt(total_spaces)))
+    rows = int(np.ceil(total_spaces / cols))
+
+    fig, ax = plt.subplots(figsize=(10, 8))
+    scale = np.sqrt(area) / 10
+    width, height = scale, scale * 0.75
+    labels = (["Room"] * rooms + ["Bath"] * baths + ["Living"] * living + ["Car Porch"] * car_porch)
+
+    for i, label in enumerate(labels):
+        row = i // cols
+        col = i % cols
+        x = col * width
+        y = (rows - 1 - row) * height
+        ax.add_patch(plt.Rectangle((x, y), width, height, edgecolor='black', facecolor='lightblue'))
+        ax.text(x + width / 2, y + height / 2, label, ha='center', va='center', fontsize=8)
+
+    ax.set_xlim(0, cols * width)
+    ax.set_ylim(0, rows * height)
+    ax.set_aspect('equal')
+    ax.set_title(f"Tentative Floor Plan (Scale: 1 unit = {int(scale)} sqft)")
+    ax.axis('off')
+
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png')
+    buf.seek(0)
+    return buf
 
 # -------------------- MAIN APP --------------------
 def main():
-    st.title("🧱 Construction Estimator")
+    st.title("🧱 Construction Estimator (RAG + LLaMA 3 + Sketch)")
+
+    st.subheader("Upload Material Data")
+
+    material_qty_file = st.file_uploader("Upload Material Quantities Excel", type=["xlsx", "xlsm"])
+    material_cost_file = st.file_uploader("Upload Material Costs Excel", type=["xlsx", "xlsm"])
+
+    if material_qty_file and material_cost_file:
+        qty_df = pd.read_excel(material_qty_file)
+        cost_df = pd.read_excel(material_cost_file)
 
-    st.subheader("🔢 Upload Quantity Excel")
-    quant_file = st.file_uploader("Material Quantities (Excel)", type=["xlsx"])
+        st.write("### Material Quantities")
+        st.dataframe(qty_df)
 
-    st.subheader("💰 Upload Rates Excel")
-    rate_file = st.file_uploader("Material Rates (Excel)", type=["xlsx"])
+        st.write("### Material Costs")
+        st.dataframe(cost_df)
 
-    if quant_file and rate_file:
-        quant_df = load_excel(quant_file)
-        rate_df = load_excel(rate_file)
+        # Attempt to get covered area from quantities file if exists
+        area = None
+        if "Covered Area (sqft)" in qty_df.columns:
+            try:
+                area = float(qty_df["Covered Area (sqft)"].iloc[0])
+            except Exception:
+                area = None
 
-        st.success("Files loaded. Preview:")
-        st.dataframe(quant_df)
-        st.dataframe(rate_df)
+        context_qty = qty_df.to_string(index=False)
+        context_cost = cost_df.to_string(index=False)
+        context = f"Material Quantities:\n{context_qty}\n\nMaterial Costs:\n{context_cost}"
 
-        st.subheader("🏗️ Input Additional Project Info")
-        rooms = st.number_input("Rooms", 1)
-        baths = st.number_input("Bathrooms", 1)
-        living = st.number_input("Living Rooms", 0)
-        car_porch = st.number_input("Car Porches", 0)
+        st.subheader("🏗️ Enter Other Project Details")
+        rooms = st.number_input("Number of Rooms", min_value=1, value=3)
+        baths = st.number_input("Number of Washrooms", min_value=1, value=2)
+        living = st.number_input("Number of Living Rooms", min_value=0, value=1)
+        car_porch = st.number_input("Number of Car Porches", min_value=0, value=1)
 
-        if st.button("Generate BOQ"):
-            boq_df = generate_boq(quant_df, rate_df)
-            st.subheader("📋 BOQ")
-            st.dataframe(boq_df)
+        if st.button("Generate Estimate"):
+            user_query = f"Estimate cost for {rooms} rooms, {baths} bathrooms, {living} living rooms, and {car_porch} car porch(es)."
+            response = generate_estimate(context, user_query)
 
-            total = boq_df["Amount"].sum()
-            st.markdown(f"### 💸 Total Estimate: Rs {total:,.0f}")
+            st.subheader("💸 Estimated Construction Cost (BOQ Style)")
+            st.markdown("**This BOQ is a simplified estimate. Use it for planning, not execution.**")
+            st.markdown(response)
 
-            # Optional: Sketch
-            sketch_text = generate_sketch_prompt(rooms, baths, living, car_porch)
-            st.subheader("🏠 Sketch Prompt (for Replicate):")
-            st.text(sketch_text)
-            st.info("Integrate Replicate API here if available.")
+            # Use area if available, else default 1200 sqft for sketch scaling
+            sketch_area = area if area else 1200
+            buf = draw_floor_plan(rooms, baths, living, car_porch, sketch_area)
+            st.subheader("🏠 Tentative Floor Plan Sketch")
+            st.image(buf, caption="Auto-generated Line Plan", use_column_width=True)
+            st.download_button("📥 Download Sketch", buf, file_name="floor_plan.png", mime="image/png")
+    else:
+        st.info("Please upload both Material Quantities and Material Costs Excel files.")
 
 if __name__ == "__main__":
     main()