Update app.py

app.py

@@ -312,6 +312,54 @@ if st.button("Generate Concept Image from Design Summary"):
 
         except Exception as e:
             st.error(f"Image generation failed: {e}")
+            
+if next_step == "🔍 Continue researching (codes, build types, regulations)":
+        st.markdown("#### 🔍 Research Hub")
+    st.info("Here are some useful topics and tools to help you dig deeper before you build.")
+
+    st.markdown("##### 🧠 Suggested Topics to Explore")
+    research_topics = {
+        "Prefab vs. Traditional Construction": "Compare pros and cons of prefabricated (modular) vs. traditional stick-built homes.",
+        "Land Use & Zoning": "Understand how land use bylaws and zoning affect what and where you can build.",
+        "Permit & Code Requirements": "Find out what permits are typically needed for a small home project in Canada.",
+        "Sustainable Building Materials": "Explore environmentally-friendly materials that are affordable and effective.",
+        "Off-Grid Living Regulations": "Learn the legal considerations if you want to live off-grid.",
+        "Energy Efficiency Standards": "Understand how to meet or exceed Canadian energy code requirements."
+    }
+
+    selected_topic = st.selectbox("Choose a topic to learn more:", ["Select a topic"] + list(research_topics.keys()))
+    if selected_topic != "Select a topic":
+        with st.spinner("Researching..."):
+            response = client.chat.completions.create(
+                model="gpt-4",
+                messages=[
+                    {"role": "system", "content": "You are a Canadian construction research assistant."},
+                    {"role": "user", "content": research_topics[selected_topic]}
+                ],
+                temperature=0.7
+            )
+            st.markdown(f"### 📖 {selected_topic}")
+            st.markdown(response.choices[0].message.content.strip())
+
+    st.markdown("##### 💬 Ask Your Own Research Question")
+    user_question = st.text_input("What else would you like to learn about?")
+    if st.button("Ask"):
+        if user_question:
+            with st.spinner("Getting the answer..."):
+                response = client.chat.completions.create(
+                    model="gpt-4",
+                    messages=[
+                        {"role": "system", "content": "You're a helpful Canadian homebuilding advisor. Keep answers clear and based on current standards."},
+                        {"role": "user", "content": user_question}
+                    ],
+                    temperature=0.7
+                )
+                st.markdown("### 💡 Answer")
+                st.markdown(response.choices[0].message.content.strip())
+        else:
+            st.warning("Please type a question first.")
+
+