Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,106 @@
|
|
|
|
|
|
1 |
import streamlit as st
|
|
|
2 |
from openai import OpenAI
|
|
|
|
|
3 |
|
4 |
-
#
|
5 |
st.set_page_config(page_title="AI Architecture Assistant", layout="centered")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
|
7 |
|
8 |
-
|
|
|
9 |
if "prompt" not in st.session_state:
|
10 |
st.session_state.prompt = ""
|
11 |
|
|
|
|
|
12 |
sample_prompt = (
|
13 |
"A highly detailed, realistic architectural floor plan for a modern tiny home. "
|
14 |
"The design features an open-concept layout with a multi-functional living space "
|
@@ -18,6 +110,7 @@ sample_prompt = (
|
|
18 |
"with subtle accents. Annotate key dimensions and furniture placements. Professional architectural rendering style."
|
19 |
)
|
20 |
|
|
|
21 |
# === UI START ===
|
22 |
st.title("🏠 AI Architecture Assistant")
|
23 |
st.caption("Design smarter. Visualize faster. Build better.")
|
@@ -370,6 +463,47 @@ st.markdown("""
|
|
370 |
- [Quebec Construction Code](https://www.rbq.gouv.qc.ca/en/technical-buildings/buildings/construction-code.html)
|
371 |
- [Tiny Home Legal Info (Canada)](https://tinyhousecanada.ca/)
|
372 |
""")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
373 |
|
374 |
|
375 |
|
|
|
1 |
+
# === IMPORTS ===
|
2 |
+
|
3 |
import streamlit as st
|
4 |
+
from PIL import Image
|
5 |
from openai import OpenAI
|
6 |
+
from pathlib import Path
|
7 |
+
|
8 |
|
9 |
+
# ✅ 1. Page Configuration — MUST be first Streamlit command
|
10 |
st.set_page_config(page_title="AI Architecture Assistant", layout="centered")
|
11 |
+
|
12 |
+
|
13 |
+
# ✅ 2. Inline custom CSS with Poppins font and olive green theme
|
14 |
+
st.markdown("""
|
15 |
+
<style>
|
16 |
+
@import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;500;700&display=swap');
|
17 |
+
|
18 |
+
.stApp {
|
19 |
+
background-color: #f4f7f5 !important;
|
20 |
+
font-family: 'Poppins', sans-serif;
|
21 |
+
color: #1e1e1e;
|
22 |
+
padding: 1rem;
|
23 |
+
}
|
24 |
+
|
25 |
+
/* Headings */
|
26 |
+
h1 {
|
27 |
+
font-family: 'Poppins', sans-serif;
|
28 |
+
color: #3a5a40 !important;
|
29 |
+
font-weight: 700;
|
30 |
+
font-size: 2.3rem;
|
31 |
+
}
|
32 |
+
|
33 |
+
h2, h3 {
|
34 |
+
font-family: 'Poppins', sans-serif;
|
35 |
+
color: #344e41 !important;
|
36 |
+
font-weight: 600;
|
37 |
+
font-size: 1.4rem;
|
38 |
+
}
|
39 |
+
|
40 |
+
/* Inputs */
|
41 |
+
.stTextInput input, .stTextArea textarea {
|
42 |
+
background-color: #ffffff !important;
|
43 |
+
color: #1e1e1e !important;
|
44 |
+
border: 1px solid #c8d6c1;
|
45 |
+
border-radius: 10px;
|
46 |
+
padding: 14px;
|
47 |
+
font-size: 1rem;
|
48 |
+
font-family: 'Poppins', sans-serif;
|
49 |
+
}
|
50 |
+
|
51 |
+
/* Buttons */
|
52 |
+
.stButton > button {
|
53 |
+
background-color: #588157;
|
54 |
+
color: white;
|
55 |
+
font-weight: 600;
|
56 |
+
font-family: 'Poppins', sans-serif;
|
57 |
+
padding: 0.7rem 1.6rem;
|
58 |
+
font-size: 1rem;
|
59 |
+
border: none;
|
60 |
+
border-radius: 10px;
|
61 |
+
transition: all 0.3s ease;
|
62 |
+
box-shadow: 0 2px 5px rgba(0,0,0,0.1);
|
63 |
+
}
|
64 |
+
|
65 |
+
.stButton > button:hover {
|
66 |
+
background-color: #3a5a40;
|
67 |
+
box-shadow: 0 4px 10px rgba(0,0,0,0.15);
|
68 |
+
}
|
69 |
+
|
70 |
+
/* Links */
|
71 |
+
a {
|
72 |
+
color: #3a5a40;
|
73 |
+
font-family: 'Poppins', sans-serif;
|
74 |
+
text-decoration: none;
|
75 |
+
font-weight: 500;
|
76 |
+
}
|
77 |
+
|
78 |
+
a:hover {
|
79 |
+
text-decoration: underline;
|
80 |
+
color: #2c4030;
|
81 |
+
}
|
82 |
+
</style>
|
83 |
+
""", unsafe_allow_html=True)
|
84 |
+
|
85 |
+
|
86 |
+
# ✅ 3. Logo Display (must come after styling)
|
87 |
+
try:
|
88 |
+
logo = Image.open("logo.jpg")
|
89 |
+
st.image(logo, width=300)
|
90 |
+
except FileNotFoundError:
|
91 |
+
st.warning("Logo not found. Please ensure 'logo.jpg' is in the same folder as app.py.")
|
92 |
+
|
93 |
+
|
94 |
+
# ✅ 4. Connect to OpenAI
|
95 |
client = OpenAI(api_key=st.secrets["OPENAI_API_KEY"])
|
96 |
|
97 |
+
|
98 |
+
# ✅ 5. Session Setup
|
99 |
if "prompt" not in st.session_state:
|
100 |
st.session_state.prompt = ""
|
101 |
|
102 |
+
|
103 |
+
# ✅ 6. Sample Prompt
|
104 |
sample_prompt = (
|
105 |
"A highly detailed, realistic architectural floor plan for a modern tiny home. "
|
106 |
"The design features an open-concept layout with a multi-functional living space "
|
|
|
110 |
"with subtle accents. Annotate key dimensions and furniture placements. Professional architectural rendering style."
|
111 |
)
|
112 |
|
113 |
+
|
114 |
# === UI START ===
|
115 |
st.title("🏠 AI Architecture Assistant")
|
116 |
st.caption("Design smarter. Visualize faster. Build better.")
|
|
|
463 |
- [Quebec Construction Code](https://www.rbq.gouv.qc.ca/en/technical-buildings/buildings/construction-code.html)
|
464 |
- [Tiny Home Legal Info (Canada)](https://tinyhousecanada.ca/)
|
465 |
""")
|
466 |
+
# === 9. Eco-Friendly Upgrades Agent ===
|
467 |
+
|
468 |
+
st.markdown("---")
|
469 |
+
st.markdown("### 🌱 Suggest Eco-Friendly Upgrades")
|
470 |
+
|
471 |
+
if st.button("Get Eco-Friendly Suggestions"):
|
472 |
+
if st.session_state.prompt.strip():
|
473 |
+
with st.spinner("Analyzing your layout for green upgrades..."):
|
474 |
+
|
475 |
+
def suggest_eco_upgrades(prompt):
|
476 |
+
system_instruction = (
|
477 |
+
"You are a sustainability expert in residential architecture. "
|
478 |
+
"Based on the following home design prompt, suggest 5 to 7 eco-friendly upgrades. "
|
479 |
+
"Be specific and include options like energy efficiency, water conservation, material choices, and renewable energy use. "
|
480 |
+
"Avoid general advice—tailor suggestions to the design and layout described."
|
481 |
+
)
|
482 |
+
|
483 |
+
response = client.chat.completions.create(
|
484 |
+
model="gpt-4",
|
485 |
+
messages=[
|
486 |
+
{"role": "system", "content": system_instruction},
|
487 |
+
{"role": "user", "content": prompt}
|
488 |
+
],
|
489 |
+
temperature=0.7
|
490 |
+
)
|
491 |
+
|
492 |
+
return response.choices[0].message.content.strip()
|
493 |
+
|
494 |
+
eco_suggestions = suggest_eco_upgrades(st.session_state.prompt)
|
495 |
+
|
496 |
+
st.markdown("### ✅ Recommended Eco-Friendly Features")
|
497 |
+
st.markdown(eco_suggestions)
|
498 |
+
|
499 |
+
# Save to session for summary
|
500 |
+
st.session_state.eco_suggestions = eco_suggestions
|
501 |
+
|
502 |
+
else:
|
503 |
+
st.warning("Please enter a prompt first.")
|
504 |
+
|
505 |
+
|
506 |
+
|
507 |
|
508 |
|
509 |
|