Update app.py

app.py

@@ -66,7 +66,7 @@ with tab1:
     # Sidebar
     with st.sidebar:
         st.header("🧪 Pathology Tools")
-        
+
         if st.button("🧹 Clear Chat"):
             for k in ["messages", "thread_id", "image_urls", "pending_prompt"]:
                 st.session_state[k] = [] if k.endswith("s") else None
@@ -98,14 +98,12 @@ with tab1:
         st.markdown("### 💬 Ask a Pathology-Specific Question")
         user_input = st.chat_input("Example: What are features of squamous cell carcinoma?")
 
-        # Append input or pending prompt
         if user_input:
             st.session_state.messages.append({"role": "user", "content": user_input})
         elif st.session_state.pending_prompt:
             st.session_state.messages.append({"role": "user", "content": st.session_state.pending_prompt})
             st.session_state.pending_prompt = None
 
-        # Trigger assistant run
         if st.session_state.messages and st.session_state.messages[-1]["role"] == "user":
             try:
                 if not st.session_state.thread_id:
@@ -135,12 +133,11 @@ with tab1:
                         if m.role == "assistant":
                             reply = m.content[0].text.value.strip()
 
-                            # De-dupe reply
                             if not any(reply in msg["content"] or msg["content"] in reply
                                        for msg in st.session_state.messages if msg["role"] == "assistant"):
                                 st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                                # Image detection
+                                # Extract image URLs
                                 images = re.findall(
                                     r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png',
                                     reply
@@ -153,22 +150,26 @@ with tab1:
             except Exception as e:
                 st.error(f"❌ Error: {e}")
 
-        # Display conversation
         for msg in st.session_state.messages:
             with st.chat_message(msg["role"]):
                 st.markdown(msg["content"], unsafe_allow_html=True)
 
-        # Optional follow-ups
+        # ✅ 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"]
-            if "Some Possible Questions:" in last:
-                questions = re.findall(r"[-•]\s*(.*)", last)
-                if questions:
-                    st.markdown("#### 💡 Follow-Up Suggestions")
-                    for q in questions:
-                        if st.button(f"📌 {q}"):
-                            st.session_state.pending_prompt = q
-                            st.rerun()
+
+            # Extract bullet lines that end in a question mark
+            questions = [
+                line.strip(" -•") for line in last.splitlines()
+                if line.strip().startswith(("-", "•")) and line.strip().endswith("?")
+            ]
+
+            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:
@@ -180,6 +181,7 @@ with tab1:
                 except Exception:
                     st.warning(f"⚠️ Failed to load image: {url}")
 
+
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"