Update app.py

app.py

@@ -213,23 +213,17 @@ with tab2:
         except Exception as e:
             st.error(f"Error: {e}")
 
-    # Render assistant response and display images using 'Image URL' field
+    # Render assistant response and load each image using the actual URL
     if st.session_state.image_response:
         st.markdown("### 🧠 Assistant Response")
         st.markdown(st.session_state.image_response, unsafe_allow_html=True)
 
-        # Extract full GitHub URLs from assistant output
+        # Extract valid GitHub URLs from assistant response
         url_matches = re.findall(r'Image URL:\s*(https?://[^\s]+)', st.session_state.image_response)
 
         for url in url_matches:
-            st.markdown(f"🔗 [View image directly]({url})")
-
             try:
-                r = requests.get(url, timeout=5)
-                r.raise_for_status()
-                img = Image.open(BytesIO(r.content))
-                st.image(img, caption=url.split("/")[-1], use_container_width=True)
+                st.image(url, caption=url.split("/")[-1], use_container_width=True)
             except Exception as e:
-                st.warning(f"⚠️ Failed to load: {url}")
+                st.warning(f"⚠️ Could not load image: {url}")
                 st.error(f"🛑 Error: {str(e)}")
-