Update app.py

app.py

@@ -2,43 +2,46 @@ import streamlit as st
 import requests
 from bs4 import BeautifulSoup
 import os
+from openai import OpenAI
 
 # === CONFIGURATION ===
 
-# Set page settings
-st.set_page_config(page_title="AI Architecture Agents", layout="centered")
-
-# Load Zyte API key from environment
+# 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
+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 "
+    "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."
+)
+
+# === ZYTE SCRAPER SETUP ===
 
-# Hardcoded Home Depot URL for laminate flooring
 LAMINATE_FLOORING_URL = "https://www.homedepot.com/b/Flooring-Laminate-Flooring/N-5yc1vZare1"
 
-# === ZYTE SCRAPER FUNCTION ===
 def zyte_scrape(url):
     headers = {"Content-Type": "application/json"}
     payload = {"url": url, "browserHtml": True}
-    try:
-        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 API error: {response.status_code}")
-            return None
-    except Exception as e:
-        st.error(f"Zyte request failed: {e}")
+    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
 
-# === PARSER FOR HOME DEPOT ===
 def parse_home_depot(html):
     soup = BeautifulSoup(html, 'html.parser')
     product_cards = soup.find_all('div', class_='product-pod')
-
-    if not product_cards:
-        st.warning("No product cards found. Parsing may need to be updated.")
-        return []
-
+    
     results = []
     for card in product_cards:
         title_tag = card.find('span', class_='product-pod__title__product')
@@ -50,22 +53,55 @@ def parse_home_depot(html):
             results.append({"name": name, "price": price})
     return results
 
-# === MAIN STREAMLIT APP ===
+# === STREAMLIT UI ===
+
+st.set_page_config(page_title="AI Architecture Agents", layout="centered")
 st.title("🏗️ AI Architecture Assistant")
 
-agent = st.selectbox("Choose an Agent:", ["Agent 1: Placeholder", "Agent 3: Material Explorer"])
+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 = ""
 
-if agent == "Agent 1: Placeholder":
-    st.subheader("🖼️ Agent 1 (placeholder)")
-    st.markdown("This section is for the image generator or prompt-based tools.")
-    st.info("Agent 1 is currently under development or simplified here.")
+    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.")
 
+# === AGENT 3: MATERIAL SCRAPER ===
 elif agent == "Agent 3: Material Explorer":
-    st.subheader("📦 Agent 3: Laminate Flooring Options")
-    st.markdown("Scrapes **Home Depot** for current **laminate flooring** product listings.")
+    st.header("📦 Building Material Browser")
+    st.markdown("Browse current **laminate flooring** options from Home Depot.")
 
-    if st.button("Scrape Home Depot"):
-        with st.spinner("Scraping product data via Zyte..."):
+    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)
@@ -73,6 +109,6 @@ elif agent == "Agent 3: Material Explorer":
                     st.success("Here are some laminate flooring options:")
                     st.table(data)
                 else:
-                    st.warning("Scraped page, but found no product data.")
+                    st.warning("No products found. Parsing may need updating.")
             else:
-                st.error("Failed to get page content from Zyte.")
+                st.error("Failed to scrape the page.")