Update app.py

app.py

@@ -155,6 +155,20 @@ with tab1:
             with st.chat_message(msg["role"]):
                 st.markdown(msg["content"], unsafe_allow_html=True)
 
+        # 🧠 Follow-up Suggestions
+        if st.session_state.messages and st.session_state.messages[-1]["role"] == "assistant":
+            latest_reply = st.session_state.messages[-1]["content"]
+            match = re.search(r"###\s*\*\*Some Possible Questions:\*\*(.*?)###", latest_reply + "###", re.DOTALL)
+            if match:
+                block = match.group(1)
+                questions = [line.strip(" -•") for line in block.splitlines() if line.strip().startswith(("-", "•"))]
+                if questions:
+                    st.markdown("#### 💡 Follow-Up Suggestions")
+                    for q in questions:
+                        if st.button(f"📌 {q}"):
+                            st.session_state.pending_prompt = q
+                            st.rerun()
+
     with image_col:
         if show_image and st.session_state.image_urls:
             st.markdown("### 🖼️ Image(s)")
@@ -167,6 +181,7 @@ with tab1:
                 except Exception:
                     continue
 
+
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"