Update app.py

app.py

@@ -104,6 +104,7 @@ with tab1:
             st.session_state.messages.append({"role": "user", "content": st.session_state.pending_prompt})
             st.session_state.pending_prompt = None
 
+        # Assistant logic
         if st.session_state.messages and st.session_state.messages[-1]["role"] == "user":
             try:
                 if not st.session_state.thread_id:
@@ -131,7 +132,20 @@ with tab1:
                     responses = client.beta.threads.messages.list(thread_id=st.session_state.thread_id).data
                     for m in reversed(responses):
                         if m.role == "assistant":
-                            reply = m.content[0].text.value.strip()
+                            full_reply = m.content[0].text.value.strip()
+
+                            # ✂️ Extract and remove "Some Possible Questions"
+                            question_block_match = re.search(
+                                r"Some Possible Questions:\s*(.*?)(\n{2,}|Other References:|$)",
+                                full_reply,
+                                re.DOTALL
+                            )
+                            question_block = None
+                            if question_block_match:
+                                question_block = question_block_match.group(1)
+                                full_reply = full_reply.replace(f"Some Possible Questions:\n{question_block}", "").strip()
+
+                            reply = full_reply
 
                             if not any(reply in msg["content"] or msg["content"] in reply
                                        for msg in st.session_state.messages if msg["role"] == "assistant"):
@@ -150,20 +164,17 @@ with tab1:
             except Exception as e:
                 st.error(f"❌ Error: {e}")
 
+        # Display messages
         for msg in st.session_state.messages:
             with st.chat_message(msg["role"]):
                 st.markdown(msg["content"], unsafe_allow_html=True)
 
-        # ✅ Follow-Up Suggestions (Questions Only)
-        if st.session_state.messages and st.session_state.messages[-1]["role"] == "assistant":
-            last = st.session_state.messages[-1]["content"]
-
-            # Extract bullet lines that end in a question mark
+        # ✅ Follow-Up Suggestions (from stripped question block)
+        if 'question_block' in locals() and question_block:
             questions = [
-                line.strip(" -•") for line in last.splitlines()
+                line.strip(" -•") for line in question_block.splitlines()
                 if line.strip().startswith(("-", "•")) and line.strip().endswith("?")
             ]
-
             if questions:
                 st.markdown("#### 💡 Follow-Up Suggestions")
                 for q in questions:
@@ -181,7 +192,6 @@ with tab1:
                 except Exception:
                     st.warning(f"⚠️ Failed to load image: {url}")
 
-
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"