Update app.py

app.py

@@ -168,16 +168,102 @@ next_step = st.selectbox(
     ]
 )
 
+if next_step == st.markdown("#### 🎨 Design Recommendations")
+st.markdown("We'll give you some smart and stylish ideas for your project based on your region and layout.")
+
+# Select region (re-use if already chosen earlier)
+design_region = st.selectbox("Where will your home be built?", [
+    "British Columbia", "Alberta", "Ontario", "Quebec", "Atlantic Canada", "Prairies"
+], key="design_region")
+
+# Select design style
+design_style = st.selectbox("What style do you like?", [
+    "Scandinavian", "Modern Minimalist", "Rustic Cabin", "Japandi", "West Coast Contemporary"
+], key="design_style")
+
+# Select categories
+selected_categories = st.multiselect(
+    "What kind of design advice would you like?",
+    [
+        "Climate-Responsive Material Suggestions",
+        "Interior Color Palette",
+        "Lighting & Window Placement Tips",
+        "Space-Saving Built-ins"
+    ],
+    default=["Climate-Responsive Material Suggestions", "Interior Color Palette"]
+)
+
+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."
-        )
+    st.markdown("We'll give you some smart and stylish ideas for your project based on your region and layout.")
+
+    # Select region
+    design_region = st.selectbox("Where will your home be built?", [
+        "British Columbia", "Alberta", "Ontario", "Quebec", "Atlantic Canada", "Prairies"
+    ], key="design_region")
+
+    # Select design style
+    design_style = st.selectbox("What style do you like?", [
+        "Scandinavian", "Modern Minimalist", "Rustic Cabin", "Japandi", "West Coast Contemporary"
+    ], key="design_style")
+
+    # Select categories
+    selected_categories = st.multiselect(
+        "What kind of design advice would you like?",
+        [
+            "Climate-Responsive Material Suggestions",
+            "Interior Color Palette",
+            "Lighting & Window Placement Tips",
+            "Space-Saving Built-ins"
+        ],
+        default=["Climate-Responsive Material Suggestions", "Interior Color Palette"]
+    )
+
+    if st.button("Generate Design Ideas"):
+        with st.spinner("Designing your dream space..."):
+            def generate_design_ideas(prompt, region, style, categories):
+                system_instruction = (
+                    "You are a residential design assistant. Based on the home description, region, and preferred design style, "
+                    "give tailored recommendations for the selected categories. Keep it concise, stylish, and easy to implement for a small-scale home."
+                )
+
+                user_input = (
+                    f"Prompt:\n{prompt}\n\n"
+                    f"Region: {region}\n"
+                    f"Style: {style}\n"
+                    f"Categories: {', '.join(categories)}"
+                )
+
+                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_results = generate_design_ideas(
+                st.session_state.prompt,
+                design_region,
+                design_style,
+                selected_categories
+            )
+
+            st.markdown("### 💡 Design Suggestions")
+            st.markdown(design_results)
+
+
 
         user_input = f"Region: {region}\n\nPrompt:\n{prompt}"