Update app.py

app.py

@@ -1,12 +1,20 @@
 import streamlit as st
+import requests
+from bs4 import BeautifulSoup
+import os
 from openai import OpenAI
 
+# === SETUP ===
 
+# 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"]
 
-# Initialize OpenAI client using Hugging Face secrets
-client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
+# Initialize OpenAI client
+client = OpenAI(api_key=OPENAI_API_KEY)
 
-# Sample architectural prompt
+# Sample prompt for Agent 1
 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 "
@@ -16,87 +24,12 @@ sample_prompt = (
     "with subtle accents. Annotate key dimensions and furniture placements. Professional architectural rendering style."
 )
 
-# Session state for user prompt
-if "prompt" not in st.session_state:
-    st.session_state.prompt = ""
-
-st.title("🏠 OpenAI Image Generator: Tiny Home Floor Plan")
-
-# Prompt input area
-st.text_area("Enter your prompt:", key="prompt_input", value=st.session_state.prompt, height=200)
-
-# Button to insert sample
-if st.button("Insert Sample Prompt"):
-    st.session_state.prompt = sample_prompt
-    st.rerun()
-
-# Sync prompt state
-st.session_state.prompt = st.session_state.prompt_input
-
-# Display the current prompt as read-only
-st.text_area("Current prompt being used:", st.session_state.prompt, height=200, disabled=True)
-
-# Generate 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 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 Explorer ===
-import re
-
-your_dalle_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
-that combines a compact living area, dining space, and efficient kitchen with smart storage.
-A cozy loft bedroom is accessible via a sleek staircase or ladder. The minimalist bathroom
-includes a shower. Emphasize large windows with natural light, clean lines, and neutral tones
-with subtle accents. Annotate key dimensions and furniture placements.
-Professional architectural rendering style.
-"""
-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"]
-"""
-
-
-import streamlit as st
-import requests
-from bs4 import BeautifulSoup
-import os
-
-# Load Zyte API key from environment
-ZYTE_API_KEY = os.getenv("ZYTE_API_KEY")
-ZYTE_API_URL = "https://api.zyte.com/v1/extract"
-
-# Hardcoded Home Depot URL
+# === Agent 3 Setup ===
 LAMINATE_FLOORING_URL = "https://www.homedepot.com/b/Flooring-Laminate-Flooring/N-5yc1vZare1"
 
-# Function to call Zyte API and return rendered HTML
 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"]
@@ -104,7 +37,6 @@ def zyte_scrape(url):
         st.error(f"Zyte error: {response.status_code}")
         return None
 
-# Function to parse the Home Depot laminate flooring page
 def parse_home_depot(html):
     soup = BeautifulSoup(html, 'html.parser')
     product_cards = soup.find_all('div', class_='product-pod')
@@ -114,17 +46,55 @@ def parse_home_depot(html):
         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 with no user input
-def main():
-    st.title("Agent 3: Building Material Browser")
+# === Streamlit App ===
+
+st.set_page_config(page_title="AI Architecture Agents", layout="centered")
+st.title("🏗️ AI Architecture Assistant")
+
+agent = st.selectbox("Choose an Agent:", ["Agent 1: Floor Plan Image Generator", "Agent 3: Material Explorer"])
+
+if agent == "Agent 1: Floor Plan Image 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)
+
+    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.")
+
+elif agent == "Agent 3: Material Explorer":
+    st.header("📦 Building Material Browser")
     st.markdown("Browse current **laminate flooring** options from Home Depot.")
 
     if st.button("Get Laminate Flooring Options"):
@@ -139,6 +109,3 @@ def main():
                     st.warning("No products found. Parsing may need updating.")
             else:
                 st.error("Failed to scrape the page.")
-
-if __name__ == "__main__":
-    main()