Update app.py

app.py

@@ -118,4 +118,41 @@ if st.button("Get Cost Estimate"):
             st.markdown(estimate)
     else:
         st.warning("Please enter a prompt first.")
+        st.markdown("### 🏗️ Building Code Comparison")
+
+province_1 = st.selectbox("Select first province:", [
+    "British Columbia", "Alberta", "Ontario", "Quebec", "Atlantic Canada", "Prairies"
+], key="prov1")
+
+province_2 = st.selectbox("Select second province:", [
+    "British Columbia", "Alberta", "Ontario", "Quebec", "Atlantic Canada", "Prairies"
+], key="prov2")
+
+if st.button("Compare Building Codes"):
+    if province_1 != province_2:
+        with st.spinner("Comparing codes across provinces..."):
+            def compare_building_codes(p1, p2, prompt):
+                system_instruction = (
+                    "You are a Canadian building code expert. Compare the key differences in building code requirements "
+                    f"between {p1} and {p2} for residential construction, especially related to small or modern homes like the one described below. "
+                    "Focus on differences in energy efficiency, foundations, insulation, accessibility, and design limitations. "
+                    "Keep your response clear and under 10 bullet points."
+                )
+
+                response = client.chat.completions.create(
+                    model="gpt-4",
+                    messages=[
+                        {"role": "system", "content": system_instruction},
+                        {"role": "user", "content": prompt}
+                    ],
+                    temperature=0.7
+                )
+                return response.choices[0].message.content.strip()
+
+            comparison = compare_building_codes(province_1, province_2, st.session_state.prompt)
+            st.markdown(f"### 🏛️ {province_1} vs. {province_2}")
+            st.markdown(comparison)
+    else:
+        st.warning("Please select two different provinces.")
+