Update app.py

app.py

@@ -134,9 +134,9 @@ with chat_col:
                         reply = m.content[0].text.value
                         st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                        # 🧠 Extract ALL image URLs from assistant reply
+                        # 🔍 Extract all GitHub image URLs
                         image_matches = re.findall(
-                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^ \n]+\.png',
+                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n]+\.png',
                             reply
                         )
                         st.session_state.image_urls = image_matches
@@ -151,15 +151,23 @@ with chat_col:
         with st.chat_message(msg["role"]):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Right Column: Image Previews ------------------
+# ------------------ Image Preview with Safe Encoding ------------------
 with image_col:
     if show_image and st.session_state.image_urls:
         st.markdown("### 🖼️ Slide Previews")
-        for url in st.session_state.image_urls:
+        for raw_url in st.session_state.image_urls:
             try:
-                r = requests.get(url)
+                url_parts = raw_url.split("githubusercontent.com/")
+                if len(url_parts) == 2:
+                    encoded_path = quote(url_parts[1])
+                    encoded_url = f"https://raw.githubusercontent.com/{encoded_path}"
+                else:
+                    encoded_url = raw_url
+
+                r = requests.get(encoded_url)
                 r.raise_for_status()
                 img = Image.open(BytesIO(r.content))
-                st.image(img, caption=f"📷 {url.split('/')[-1]}", use_container_width=True)
+                st.image(img, caption=f"📷 {encoded_url.split('/')[-1]}", use_container_width=True)
             except Exception as e:
-                st.error(f"❌ Failed to load image from {url}: {e}")
+                st.error(f"❌ Failed to load image from {raw_url}: {e}")
+