Update app.py

app.py

@@ -53,3 +53,66 @@ if st.button("Generate Image"):
                 st.error(f"Error: {e}")
     else:
         st.warning("Please enter a valid prompt.")
+        # === Agent 3: Material Explorer ===
+import json
+
+st.header("Agent 3: Material Explorer")
+
+# Optionally use DALL·E prompt if needed (or use layout summary from Agent 1/2)
+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"]
+        """
+
+        model = genai.GenerativeModel("gemini-1.5-pro")
+        response = model.generate_content(material_prompt)
+        raw = response.text.strip()
+
+        import re
+        cleaned = re.sub(r"```json|```", "", raw).strip()
+
+        try:
+            materials = json.loads(cleaned)
+            st.session_state.materials = materials
+            st.success("✅ Materials extracted.")
+        except Exception as e:
+            st.error("❌ Could not parse Gemini response.")
+            st.code(raw)
+            materials = []
+# Hardcoded URL map
+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 material links
+if "materials" in st.session_state:
+    st.subheader("🔗 Explore Real-World Materials")
+
+    for mat in st.session_state.materials:
+        url = material_links.get(mat)
+        if url:
+            st.markdown(f"✅ **{mat}**: [View on Home Depot]({url})", unsafe_allow_html=True)
+        else:
+            st.markdown(f"🟡 **{mat}**: _(no link available)_")
+