Update app.py

app.py

@@ -119,6 +119,71 @@ if "requirements_summary" in st.session_state:
 
             except Exception as e:
                 st.error(f"❌ Replicate API failed: {e}")
+                # === Agent 3: Material Explorer ===
+import re
+
+st.header("Agent 3: Material Explorer")
+
+# Use the previous layout description as input
+your_dalle_prompt = st.session_state.get("requirements_summary", "")
+
+if st.button("Get Suggested Materials"):
+    with st.spinner("Using Gemini to extract materials..."):
+
+        material_prompt = f"""
+        From the following blueprint prompt and layout concept:
+
+        Blueprint Prompt:
+        \"\"\"{your_dalle_prompt}\"\"\"
+
+        Return 5 key building materials implied by the layout, even if not explicitly listed.
+
+        Respond ONLY with a raw JSON list like:
+        ["Concrete slab foundation", "Metal roofing", "Plywood sheathing", "Double-pane windows", "Insulation"]
+        """
+
+        try:
+            model = genai.GenerativeModel("gemini-1.5-pro")
+            response = model.generate_content(material_prompt)
+            raw = response.text.strip()
+            cleaned = re.sub(r"```json|```", "", raw).strip()
+            materials = json.loads(cleaned)
+            st.session_state.materials = materials
+            st.success("✅ Materials extracted from layout!")
+        except Exception as e:
+            st.error("❌ Could not parse Gemini response.")
+            st.code(raw)
+            materials = []
+
+# Predefined Home Depot product links
+material_links = {
+    "Engineered wood flooring": "https://www.homedepot.ca/product/home-decorators-collection-laminate-flooring-castle-oak/1001624267",
+    "Laminate countertops": "https://www.homedepot.ca/product/hampton-bay-premium-laminate-countertop-butchers-block/1001185104",
+    "Tile (bathroom)": "https://www.homedepot.ca/product/enigma-12-inch-x-24-inch-glazed-porcelain-tile-aria-grey/1001066513",
+    "Painted drywall": "https://www.homedepot.ca/product/certainteed-1-2-inch-x-4-ft-x-8-ft-drywall/1000116648",
+    "Wood framing (stud walls)": "https://www.homedepot.ca/product/2-inch-x-4-inch-x-8-ft-construction-grade-lumber/1000115194",
+    "Spray foam insulation": "https://www.homedepot.ca/product/great-stuff-big-gap-filler-insulating-foam-sealant-12-oz/1000110899",
+    "Triple-glazed windows": "https://www.homedepot.ca/product/american-craftsman-60-inch-x-60-inch-double-pane-double-hung-vinyl-window/1001202871",
+    "Metal roofing": "https://www.homedepot.ca/product/ondura-36-inch-x-79-inch-black-roof-panel/1000167679",
+    "Solar panel kit": "https://www.homedepot.ca/product/nature-power-110-watt-solar-panel-kit-with-inverter-and-charging-controller/1001037002",
+    "Mini wood stove": "https://www.homedepot.ca/product/englander-1-200-sq-ft-wood-burning-stove/1000829230"
+}
+
+# Show links in Streamlit
+if "materials" in st.session_state:
+    st.subheader("🔗 Explore Suggested Materials")
+
+    cols = st.columns(2)
+
+    for i, mat in enumerate(st.session_state.materials):
+        col = cols[i % 2]
+        with col:
+            url = material_links.get(mat)
+            if url:
+                st.markdown(f"✅ **{mat}**\n\n[🛒 View Product on Home Depot]({url})", unsafe_allow_html=True)
+            else:
+                st.markdown(f"🟡 **{mat}**\n\n_(no direct link available)_")
+