Update app.py

app.py

@@ -120,6 +120,9 @@ if st.button("Get Cost Estimate"):
         st.warning("Please enter a prompt first.")
         st.markdown("### 🏗️ Building Code Comparison")
 
+st.markdown("---")
+st.markdown("### 🧾 Building Code Comparison Between Provinces")
+
 province_1 = st.selectbox("Select first province:", [
     "British Columbia", "Alberta", "Ontario", "Quebec", "Atlantic Canada", "Prairies"
 ], key="prov1")
@@ -154,5 +157,74 @@ if st.button("Compare Building Codes"):
             st.markdown(comparison)
     else:
         st.warning("Please select two different provinces.")
+        st.markdown("### ➡️ What would you like to do next?")
+
+next_step = st.selectbox(
+    "Now that you have some information and a layout in mind, would you like help with...",
+    [
+        "Select an option",
+        "🎨 Explore design ideas (color palettes, materials for your climate)",
+        "🔍 Continue researching (codes, build types, regulations)"
+    ]
+)
+
+if next_step == "🎨 Explore design ideas (color palettes, materials for your climate)":
+    st.markdown("#### 🎨 Design Recommendations")
+    st.info("Based on your prompt and region, here are some interior and exterior design ideas:")
+    
+    def suggest_designs(prompt, region):
+        system_instruction = (
+            "You are a residential design advisor. Based on the architectural prompt and region, "
+            "suggest color palettes, finish materials, and stylistic choices that suit the climate "
+            "and vibe of the home. Include interior and exterior ideas."
+        )
+
+        user_input = f"Region: {region}\n\nPrompt:\n{prompt}"
+
+        response = client.chat.completions.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": system_instruction},
+                {"role": "user", "content": user_input}
+            ],
+            temperature=0.7
+        )
+
+        return response.choices[0].message.content.strip()
+
+    design_suggestions = suggest_designs(st.session_state.prompt, region)
+    st.markdown(design_suggestions)
+
+elif next_step == "🔍 Continue researching (codes, build types, regulations)":
+    st.markdown("#### 🔍 Further Research Suggestions")
+    st.info("Here are some directions you can explore to better prepare your build:")
+
+    st.markdown("""
+    - 📄 Compare prefab vs. traditional construction  
+    - 🌱 Explore sustainable and passive design principles  
+    - 🧾 Learn about zoning and land-use requirements in your area  
+    - 🛠️ Get familiar with permit processes and builder requirements  
+    """)
+
+    if st.button("Ask a research question"):
+        research_q = st.text_input("What would you like to know more about?")
+        if research_q:
+            with st.spinner("Finding answers..."):
+                system_instruction = (
+                    "You are a research assistant helping someone planning to build a modern home. "
+                    "Answer their question in a clear, helpful way using Canadian context where possible."
+                )
+
+                response = client.chat.completions.create(
+                    model="gpt-4",
+                    messages=[
+                        {"role": "system", "content": system_instruction},
+                        {"role": "user", "content": research_q}
+                    ],
+                    temperature=0.7
+                )
+                st.markdown(response.choices[0].message.content.strip())
+
+