Update app.py

app.py

@@ -151,23 +151,25 @@ with chat_col:
         with st.chat_message(msg["role"]):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Image Preview with Safe Encoding ------------------
+# ------------------ Image Preview (Safe Encoding) ------------------
 with image_col:
     if show_image and st.session_state.image_urls:
         st.markdown("### 🖼️ Slide Previews")
         for raw_url in st.session_state.image_urls:
             try:
-                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
-
+                # Split and encode each part of the URL after domain
+                if "githubusercontent.com" not in raw_url:
+                    continue
+                _, raw_path = raw_url.split("githubusercontent.com/", 1)
+                segments = raw_path.strip().split("/")
+                encoded_segments = [quote(seg) for seg in segments]
+                encoded_url = f"https://raw.githubusercontent.com/{'/'.join(encoded_segments)}"
+
+                # Load and display image
                 r = requests.get(encoded_url)
                 r.raise_for_status()
                 img = Image.open(BytesIO(r.content))
-                st.image(img, caption=f"📷 {encoded_url.split('/')[-1]}", use_container_width=True)
+                st.image(img, caption=f"📷 {encoded_segments[-1]}", use_container_width=True)
+
             except Exception as e:
                 st.error(f"❌ Failed to load image from {raw_url}: {e}")
-