Update app.py

app.py

@@ -116,13 +116,17 @@ with tab1:
 
                 with st.spinner("🔬 Analyzing..."):
                     while True:
-                        status = client.beta.threads.runs.retrieve(thread_id=st.session_state.thread_id, run_id=run.id)
+                        status = client.beta.threads.runs.retrieve(
+                            thread_id=st.session_state.thread_id, run_id=run.id
+                        )
                         if status.status in ("completed", "failed", "cancelled"):
                             break
                         time.sleep(1)
 
                 if status.status == "completed":
-                    responses = client.beta.threads.messages.list(thread_id=st.session_state.thread_id).data
+                    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()
@@ -130,6 +134,7 @@ with tab1:
                                        for msg in st.session_state.messages if msg["role"] == "assistant"):
                                 st.session_state.messages.append({"role": "assistant", "content": reply})
 
+                                # Extract image URLs from response
                                 images = re.findall(
                                     r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png',
                                     reply
@@ -142,20 +147,27 @@ 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)
 
+        # Display follow-up questions
         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)
+                # Filter only lines that are actual questions
+                all_lines = re.findall(r"[-•]\s*(.*)", last)
+                questions = [line for line in all_lines if 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()
+                else:
+                    st.markdown("#### 💡 No follow-up questions detected in the assistant's response.")
 
     with image_col:
         if show_image and st.session_state.image_urls:
@@ -167,7 +179,6 @@ with tab1:
                 except Exception:
                     st.warning(f"⚠️ Failed to load image: {url}")
 
-
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"