Update app.py

app.py

@@ -78,6 +78,16 @@ with col2:
             with st.chat_message(msg["role"]):
                 st.write(msg["content"])
 
+                # ✅ Extract and render follow-up questions as buttons (from assistant)
+                if msg["role"] == "assistant":
+                    followups = re.findall(r"[-•*] (What|How|Are|Is|Does|Can|When|Where|Why)[^\n]+", msg["content"])
+                    if followups:
+                        st.markdown("#### 💡 Ask a Follow-up:")
+                        for q in followups:
+                            if st.button(f"🔁 {q}", key=f"followup-{q}"):
+                                st.session_state.messages.append({"role": "user", "content": q})
+                                st.rerun()
+
     if prompt:
         st.session_state.messages.append({"role": "user", "content": prompt})