Update app.py

app.py

@@ -2,16 +2,21 @@ import streamlit as st
 import pandas as pd
 import numpy as np
 import torch
-import openai
 import os
 from sentence_transformers import SentenceTransformer
 from sklearn.metrics.pairwise import cosine_similarity
 import matplotlib.pyplot as plt
 import io
+from openai import OpenAI
 
-# Set up Groq API
-openai.api_key = os.getenv("GROQ_API_KEY")
-openai.api_base = "https://api.groq.com/openai/v1"
+# Set page config FIRST
+st.set_page_config(page_title="Construction Estimator", layout="centered")
+
+# 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"
 
 # Load Excel file
@@ -23,11 +28,7 @@ def load_excel(file):
 
 # Chunk data
 def chunk_data(df, chunk_size=5):
-    chunks = []
-    for i in range(0, len(df), chunk_size):
-        chunk = df.iloc[i:i+chunk_size].to_string(index=False)
-        chunks.append(chunk)
-    return chunks
+    return [df.iloc[i:i+chunk_size].to_string(index=False) for i in range(0, len(df), chunk_size)]
 
 # Embed chunks
 @st.cache_resource
@@ -43,7 +44,7 @@ def query_embedding(user_query, chunks, embeddings, model):
     top_idx = np.argmax(similarities)
     return chunks[top_idx]
 
-# Generate estimate
+# 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:
 
@@ -53,13 +54,14 @@ Generate a detailed BOQ estimate including item numbers, full descriptions, unit
 {user_input}
 
 Present the result in a markdown table with columns: Item No, Description, Qty, Unit, Rate, Amount."""
-    response = openai.ChatCompletion.create(
+
+    response = client.chat.completions.create(
         model=GROQ_MODEL,
         messages=[{"role": "user", "content": prompt}]
     )
-    return response['choices'][0]['message']['content']
+    return response.choices[0].message.content
 
-# Calculate quantities
+# Quantity logic
 def calculate_quantities(rooms, area, baths, car_porch, living):
     return {
         "Total Area (sqft)": area,
@@ -69,19 +71,16 @@ def calculate_quantities(rooms, area, baths, car_porch, living):
         "Car Porch Area (est.)": car_porch * 200
     }
 
-# Generate scaled sketch
+# 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  # Simple scale factor
+    scale = np.sqrt(area) / 10
     width, height = scale, scale * 0.75
-
-    labels = (["Room"] * rooms + ["Bath"] * baths +
-              ["Living"] * living + ["Car Porch"] * car_porch)
+    labels = (["Room"] * rooms + ["Bath"] * baths + ["Living"] * living + ["Car Porch"] * car_porch)
 
     for i, label in enumerate(labels):
         row = i // cols
@@ -102,9 +101,8 @@ def draw_floor_plan(rooms, baths, living, car_porch, area):
     buf.seek(0)
     return buf
 
-# Main app
+# -------------------- MAIN APP --------------------
 def main():
-    st.set_page_config(page_title="Construction Estimator", layout="centered")
     st.title("🧱 Construction Estimator (RAG + LLaMA 3 + Sketch)")
 
     excel_file = st.file_uploader("Upload Schedule of Rates (.xlsx or .xlsm)", type=["xlsx", "xlsm"])
@@ -131,14 +129,9 @@ def main():
             st.json(quantities)
 
             st.subheader("💸 Estimated Construction Cost (BOQ Style)")
-            st.markdown(
-                """
-                **The BOQ generated here is a SIMPLIFIED ESTIMATE. A detailed and accurate BOQ requires professional quantity surveying and design specifications.**
-                """
-            )
+            st.markdown("**This BOQ is a simplified estimate. Use it for planning, not execution.**")
             st.markdown(response)
 
-            # Generate Sketch
             buf = draw_floor_plan(rooms, baths, living, car_porch, area)
             st.subheader("🏠 Tentative Floor Plan Sketch")
             st.image(buf, caption="Auto-generated Line Plan", use_column_width=True)