Update app.py

app.py

@@ -1,20 +1,14 @@
 import streamlit as st
-import requests
-from bs4 import BeautifulSoup
-import os
 from openai import OpenAI
 
-# === CONFIGURATION ===
+# === SETUP ===
+st.set_page_config(page_title="AI Architecture Assistant", layout="centered")
+client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
 
-# Load API keys
-ZYTE_API_KEY = os.getenv("ZYTE_API_KEY")
-ZYTE_API_URL = "https://api.zyte.com/v1/extract"
-OPENAI_API_KEY = st.secrets["OPENAI_API_KEY"]
+# === PROMPT SETUP ===
+if "prompt" not in st.session_state:
+    st.session_state.prompt = ""
 
-# Initialize OpenAI client
-client = OpenAI(api_key=OPENAI_API_KEY)
-
-# Sample prompt for OpenAI image generation
 sample_prompt = (
     "A highly detailed, realistic architectural floor plan for a modern tiny home. "
     "The design features an open-concept layout with a multi-functional living space "
@@ -24,91 +18,71 @@ sample_prompt = (
     "with subtle accents. Annotate key dimensions and furniture placements. Professional architectural rendering style."
 )
 
-# === ZYTE SCRAPER SETUP ===
-
-LAMINATE_FLOORING_URL = "https://www.homedepot.com/b/Flooring-Laminate-Flooring/N-5yc1vZare1"
-
-def zyte_scrape(url):
-    headers = {"Content-Type": "application/json"}
-    payload = {"url": url, "browserHtml": True}
-    response = requests.post(ZYTE_API_URL, headers=headers, json=payload, auth=(ZYTE_API_KEY, ""))
-    if response.status_code == 200:
-        return response.json()["browserHtml"]
-    else:
-        st.error(f"Zyte error: {response.status_code}")
-        return None
-
-def parse_home_depot(html):
-    soup = BeautifulSoup(html, 'html.parser')
-    product_cards = soup.find_all('div', class_='product-pod')
-    
-    results = []
-    for card in product_cards:
-        title_tag = card.find('span', class_='product-pod__title__product')
-        price_dollars = card.find('span', class_='price__dollars')
-        price_cents = card.find('span', class_='price__cents')
-        if title_tag and price_dollars:
-            name = title_tag.text.strip()
-            price = f"${price_dollars.text.strip()}.{price_cents.text.strip() if price_cents else '00'}"
-            results.append({"name": name, "price": price})
-    return results
-
-# === STREAMLIT UI ===
-
-st.set_page_config(page_title="AI Architecture Agents", layout="centered")
-st.title("🏗️ AI Architecture Assistant")
+# === UI START ===
+st.title("🏠 AI Architecture Assistant")
+st.caption("Design smarter. Visualize faster. Build better.")
 
-agent = st.selectbox("Choose an Agent:", ["Agent 1: Floor Plan Generator", "Agent 3: Material Explorer"])
-
-# === AGENT 1: IMAGE GENERATOR ===
-if agent == "Agent 1: Floor Plan Generator":
-    st.header("🖼️ Generate a Tiny Home Floor Plan")
-
-    if "prompt" not in st.session_state:
-        st.session_state.prompt = ""
-
-    st.text_area("Enter your prompt:", key="prompt_input", value=st.session_state.prompt, height=200)
+# === 1. Prompt Input ===
+st.markdown("### ✍️ Enter Your Floor Plan Prompt")
+st.text_area("Prompt", key="prompt_input", value=st.session_state.prompt, height=200)
 
+col1, col2 = st.columns(2)
+with col1:
     if st.button("Insert Sample Prompt"):
         st.session_state.prompt = sample_prompt
         st.rerun()
 
-    st.session_state.prompt = st.session_state.prompt_input
-    st.text_area("Current prompt being used:", st.session_state.prompt, height=200, disabled=True)
-
-    if st.button("Generate Image"):
-        if st.session_state.prompt.strip():
-            with st.spinner("Generating image..."):
-                try:
-                    response = client.images.generate(
-                        model="dall-e-3",
-                        prompt=st.session_state.prompt,
-                        size="1024x1024",
-                        quality="standard",
-                        n=1
-                    )
-                    image_url = response.data[0].url
-                    st.image(image_url, caption="Generated Image", use_column_width=True)
-                    st.markdown(f"[Download Image]({image_url})", unsafe_allow_html=True)
-                except Exception as e:
-                    st.error(f"Error: {e}")
-        else:
-            st.warning("Please enter a valid prompt.")
-
-# === AGENT 3: MATERIAL SCRAPER ===
-elif agent == "Agent 3: Material Explorer":
-    st.header("📦 Building Material Browser")
-    st.markdown("Browse current **laminate flooring** options from Home Depot.")
+st.session_state.prompt = st.session_state.prompt_input
+
+st.text_area("Current Prompt Being Used", st.session_state.prompt, height=200, disabled=True)
+
+# === 2. Generate Image ===
+st.markdown("### 🖼️ Generate a Floor Plan Image")
+
+if st.button("Generate Image"):
+    if st.session_state.prompt.strip():
+        with st.spinner("Generating image..."):
+            try:
+                response = client.images.generate(
+                    model="dall-e-3",
+                    prompt=st.session_state.prompt,
+                    size="1024x1024",
+                    quality="standard",
+                    n=1
+                )
+                image_url = response.data[0].url
+                st.image(image_url, caption="Generated Floor Plan", use_column_width=True)
+                st.markdown(f"[Download Image]({image_url})", unsafe_allow_html=True)
+            except Exception as e:
+                st.error(f"Error: {e}")
+    else:
+        st.warning("Please enter a valid prompt.")
+
+# === 3. Recommend Materials ===
+st.markdown("### 🧱 Recommended Building Materials")
+
+if st.button("Get Material Recommendations"):
+    if st.session_state.prompt.strip():
+        with st.spinner("Analyzing design and recommending materials..."):
+            def generate_material_recommendations(prompt):
+                system_instruction = (
+                    "You are a smart building material recommender. Based on the architectural prompt, "
+                    "return a list of 5 to 10 building materials implied by the layout or concept. "
+                    "Respond in clear, plain English using bullet points. Do not return JSON."
+                )
+
+                response = client.chat.completions.create(
+                    model="gpt-4",
+                    messages=[
+                        {"role": "system", "content": system_instruction},
+                        {"role": "user", "content": prompt}
+                    ],
+                    temperature=0.7
+                )
+                return response.choices[0].message.content.strip()
+
+            result = generate_material_recommendations(st.session_state.prompt)
+            st.markdown(result)
+    else:
+        st.warning("Please enter a prompt first.")
 
-    if st.button("Get Laminate Flooring Options"):
-        with st.spinner("Scraping Home Depot via Zyte..."):
-            html = zyte_scrape(LAMINATE_FLOORING_URL)
-            if html:
-                data = parse_home_depot(html)
-                if data:
-                    st.success("Here are some laminate flooring options:")
-                    st.table(data)
-                else:
-                    st.warning("No products found. Parsing may need updating.")
-            else:
-                st.error("Failed to scrape the page.")