Update app.py

app.py

@@ -206,9 +206,8 @@ with tab2:
                         response_text = msg.content[0].text.value.strip()
                         st.session_state.image_response = response_text
 
-                        # ✅ Robust parser (works for 🖼️ lines and "Image URL:" labels)
-                        import re
-                        blocks = re.split(r"(?:###\s)?🖼️", response_text)[1:]
+                        # ✅ Reliable parser for markdown blocks
+                        blocks = response_text.split("### 🖼️")[1:]
                         results = []
                         for b in blocks:
                             lines = b.strip().splitlines()
@@ -233,28 +232,22 @@ with tab2:
         except Exception as e:
             st.error(f"❌ Assistant Error: {e}")
 
-    # 🖼️ Display Image Cards in 4-column layout
+    # 🖼️ Display images in vertical stack
     if st.session_state.image_results:
         st.markdown("### 🖼️ Image Results")
-        for i in range(0, len(st.session_state.image_results), 4):
-            row = st.columns(4)
-            for j, col in enumerate(row):
-                idx = i + j
-                if idx < len(st.session_state.image_results):
-                    item = st.session_state.image_results[idx]
-                    with col:
-                        try:
-                            img_data = requests.get(item["url"], timeout=10).content
-                            st.image(Image.open(BytesIO(img_data)), use_container_width=True)
-                        except:
-                            st.image("https://via.placeholder.com/150?text=Image+Error", use_container_width=True)
-
-                        st.markdown(f"**{item.get('title', 'No Title')}**", unsafe_allow_html=True)
-                        st.caption(item.get("desc", "")[:160] + "..." if item.get("desc") else "")
-                        if st.button("🔍 View", key=f"img_{idx}"):
-                            st.session_state.image_lightbox = item["url"]
-
-    # 🔬 Full Image Lightbox Viewer
+        for idx, item in enumerate(st.session_state.image_results):
+            try:
+                img = Image.open(BytesIO(requests.get(item["url"], timeout=10).content))
+                st.image(img, use_container_width=True)
+            except:
+                st.image("https://via.placeholder.com/150?text=Image+Error", use_container_width=True)
+
+            st.markdown(f"**{item.get('title', 'No Title')}**", unsafe_allow_html=True)
+            st.caption(item.get("desc", "")[:200] + "..." if item.get("desc") else "")
+            if st.button("🔍 View", key=f"img_{idx}"):
+                st.session_state.image_lightbox = item["url"]
+
+    # 🔬 Full image lightbox
     if st.session_state.image_lightbox:
         st.markdown("### 🔬 Full Image View")
         try: