Update app.py

app.py

@@ -219,23 +219,26 @@ with tab2:
                         response_text = msg.content[0].text.value
                         st.session_state.image_response = response_text
 
-                        # ✅ Robust URL extraction from markdown-formatted response
+                        # ✅ FINAL FIXED PARSER: Handles Image URL on next line
                         lines = response_text.splitlines()
                         image_urls = []
-                        for i, line in enumerate(lines):
-                            clean_line = re.sub(r"[*`]", "", line).strip().lower()
-                            if "image url:" in clean_line:
-                                url_candidate = line.split("Image URL:")[-1].strip()
-                                if not url_candidate.startswith("http") and i + 1 < len(lines):
-                                    next_line = re.sub(r"[*`]", "", lines[i + 1].strip())
-                                    if next_line.startswith("http"):
-                                        url_candidate = next_line
-                                if url_candidate.startswith("http"):
-                                    image_urls.append(url_candidate)
+                        expecting_url = False
+
+                        for line in lines:
+                            line_clean = line.strip().replace("**", "")
+                            if "Image URL:" in line_clean:
+                                parts = line_clean.split("Image URL:")
+                                if len(parts) > 1 and parts[1].strip().startswith("http"):
+                                    image_urls.append(parts[1].strip())
+                                else:
+                                    expecting_url = True  # Flag next line
+                            elif expecting_url:
+                                if line_clean.startswith("http"):
+                                    image_urls.append(line_clean)
+                                expecting_url = False
 
                         st.session_state.image_results = [{"image": url} for url in image_urls]
 
-                        # Optional: auto-preview first image
                         if image_urls and not st.session_state.image_lightbox:
                             st.session_state.image_lightbox = image_urls[0]
                         break
@@ -248,7 +251,7 @@ with tab2:
         if st.session_state.image_response:
             st.markdown("### 🧠 Assistant Response")
             st.markdown(st.session_state.image_response, unsafe_allow_html=True)
-            # Optional: Debug raw output
+            # Optional: Debug
             # st.code(st.session_state.image_response, language="markdown")
 
     with image_col: