Update app.py

app.py

@@ -1,17 +1,15 @@
 # === IMPORTS ===
+
 import streamlit as st
 from PIL import Image
-import openai
+from openai import OpenAI
 from pathlib import Path
 import io
 
-# ✅ 1. Page Configuration
+# ✅ 1. Page Configuration — MUST be first Streamlit command
 st.set_page_config(page_title="AI Architecture Assistant", layout="centered")
 
-# ✅ 2. Set API Key from Streamlit secrets
-openai.api_key = st.secrets["OPENAI_API_KEY"]
-
-# ✅ 3. Inline custom CSS with Poppins font and olive green theme
+# ✅ 2. Inline custom CSS with Poppins font and olive green theme
 st.markdown("""
 <style>
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;500;700&display=swap');
@@ -21,6 +19,7 @@ st.markdown("""
     color: #1e1e1e;
     padding: 1rem;
 }
+/* Headings */
 h1 {
     font-family: 'Poppins', sans-serif;
     color: #3a5a40 !important;
@@ -33,6 +32,7 @@ h2, h3 {
     font-weight: 600;
     font-size: 1.4rem;
 }
+/* Inputs */
 .stTextInput input, .stTextArea textarea {
     background-color: #ffffff !important;
     color: #1e1e1e !important;
@@ -42,6 +42,7 @@ h2, h3 {
     font-size: 1rem;
     font-family: 'Poppins', sans-serif;
 }
+/* Buttons */
 .stButton > button {
     background-color: #588157;
     color: white;
@@ -58,6 +59,7 @@ h2, h3 {
     background-color: #3a5a40;
     box-shadow: 0 4px 10px rgba(0,0,0,0.15);
 }
+/* Links */
 a {
     color: #3a5a40;
     font-family: 'Poppins', sans-serif;
@@ -71,17 +73,25 @@ a:hover {
 </style>
 """, unsafe_allow_html=True)
 
-# ✅ 4. Logo Display
+
+# ✅ 3. Logo Display (must come after styling)
 try:
     logo = Image.open("ThinkTiny.jpg")
     st.image(logo, width=300)
 except FileNotFoundError:
     st.warning("Logo not found. Please ensure 'ThinkTiny.jpg' is in the same folder as app.py.")
 
+
+# ✅ 4. Connect to OpenAI
+client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
+
+
 # ✅ 5. Session Setup
 if "prompt" not in st.session_state:
     st.session_state.prompt = ""
 
+
+# ✅ 6. Sample Prompt
 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 "
@@ -91,10 +101,13 @@ sample_prompt = (
     "with subtle accents. Annotate key dimensions and furniture placements. Professional architectural rendering style."
 )
 
-st.title("\U0001F3E0 AI Architecture Assistant")
+
+# === UI START ===
+st.title("🏠 AI Architecture Assistant")
 st.caption("Design smarter. Visualize faster. Build better.")
 
-st.markdown("### \u270D\ufe0f Enter Your Floor Plan Prompt")
+# === 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)
@@ -104,22 +117,24 @@ with col1:
         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)
 
-st.markdown("### \U0001F5BC\ufe0f Generate a Floor Plan Image")
+# === 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 = openai.Image.create(
+                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']
+                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:
@@ -127,7 +142,8 @@ if st.button("Generate Image"):
     else:
         st.warning("Please enter a valid prompt.")
 
-st.markdown("### \U0001F9F1 Recommended Building Materials")
+# === 3. Recommend Materials ===
+st.markdown("### 🧱 Recommended Building Materials")
 
 if st.button("Get Material Recommendations"):
     if st.session_state.prompt.strip():
@@ -138,7 +154,8 @@ if st.button("Get Material Recommendations"):
                     "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 = openai.ChatCompletion.create(
+
+                response = client.chat.completions.create(
                     model="gpt-4",
                     messages=[
                         {"role": "system", "content": system_instruction},
@@ -147,13 +164,12 @@ if st.button("Get Material Recommendations"):
                     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.")
 
-# More code continues below for costs, code comparison, etc. (unchanged structure, just update API calls as shown)
-