Update app.py

app.py

@@ -219,13 +219,15 @@ with tab2:
                         response_text = msg.content[0].text.value
                         st.session_state.image_response = response_text
 
+                        # ✅ FIXED: Robustly parse "Image URL" from markdown-formatted response
                         lines = response_text.splitlines()
                         image_urls = []
                         for i, line in enumerate(lines):
-                            if "Image URL:" in line:
-                                url = line.split("Image URL:")[-1].strip()
+                            clean_line = re.sub(r"\*\*", "", line).strip()
+                            if "Image URL:" in clean_line:
+                                url = clean_line.split("Image URL:")[-1].strip()
                                 if not url.startswith("http") and i + 1 < len(lines):
-                                    next_line = lines[i + 1].strip()
+                                    next_line = re.sub(r"\*\*", "", lines[i + 1].strip())
                                     if next_line.startswith("http"):
                                         url = next_line
                                 if url.startswith("http"):
@@ -276,4 +278,3 @@ with tab2:
         if st.button("❌ Close Preview"):
             st.session_state.image_lightbox = None
             st.rerun()
-