Update app.py

app.py

@@ -206,8 +206,9 @@ with tab2:
                         response_text = msg.content[0].text.value.strip()
                         st.session_state.image_response = response_text
 
-                        # ✅ Robust markdown block parser
-                        blocks = response_text.split("### 🖼️")[1:]
+                        # ✅ Robust parser (works for 🖼️ lines and "Image URL:" labels)
+                        import re
+                        blocks = re.split(r"(?:###\s)?🖼️", response_text)[1:]
                         results = []
                         for b in blocks:
                             lines = b.strip().splitlines()
@@ -217,10 +218,10 @@ with tab2:
                                 "url": ""
                             }
                             for line in lines:
-                                if line.startswith("- **Description:**"):
-                                    card["desc"] = line.replace("- **Description:**", "").strip()
-                                elif line.startswith("- **Image URL:**"):
-                                    card["url"] = line.replace("- **Image URL:**", "").strip()
+                                if "Description:" in line:
+                                    card["desc"] = line.split("Description:")[1].strip()
+                                elif "Image URL:" in line:
+                                    card["url"] = line.split("Image URL:")[1].strip()
                                 elif line.startswith("http") and not card["url"]:
                                     card["url"] = line.strip()
                             if card["url"]:
@@ -243,16 +244,17 @@ with tab2:
                     item = st.session_state.image_results[idx]
                     with col:
                         try:
-                            img = Image.open(BytesIO(requests.get(item["url"]).content))
-                            st.image(img, use_container_width=True)
+                            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", "")[:150] + "..." if item.get("desc") else "")
+                        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
     if st.session_state.image_lightbox:
         st.markdown("### 🔬 Full Image View")
         try: