Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 import os
 import time
 import re
+import requests
+from PIL import Image
+from io import BytesIO
 from urllib.parse import quote
 from openai import OpenAI
 
@@ -130,7 +133,6 @@ with chat_col:
                     if m.role == "assistant":
                         reply = m.content[0].text.value
 
-                        # ✂️ Split output into main and references
                         split_marker = "---"
                         if split_marker in reply:
                             main_answer, references = reply.split(split_marker, 1)
@@ -139,7 +141,7 @@ with chat_col:
                         else:
                             st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                        # Extract raw GitHub image URLs
+                        # Extract raw GitHub URLs
                         image_matches = re.findall(
                             r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n]+\.png',
                             reply
@@ -156,17 +158,24 @@ with chat_col:
         with st.chat_message("assistant" if msg["role"] != "references" else "📄"):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Image Viewer (with encoded URLs) ------------------
+# ------------------ Image Viewer (requests + PIL fallback) ------------------
 with image_col:
     if show_image and st.session_state.image_urls:
         st.markdown("### 🖼️ Slide Previews")
         for i, raw_url in enumerate(st.session_state.image_urls):
             try:
-                if "raw.githubusercontent.com" in raw_url:
-                    base = "https://raw.githubusercontent.com/"
-                    path = raw_url.replace(base, "")
-                    encoded_path = "/".join(quote(p) for p in path.split("/"))
-                    encoded_url = base + encoded_path
-                    st.image(encoded_url, caption=f"Slide {i + 1}", use_container_width=True)
+                # Encode the full URL path
+                base = "https://raw.githubusercontent.com/"
+                path = raw_url.replace(base, "")
+                encoded_path = "/".join(quote(p) for p in path.split("/"))
+                encoded_url = base + encoded_path
+
+                # Fetch image with requests
+                response = requests.get(encoded_url)
+                if response.status_code == 200:
+                    img = Image.open(BytesIO(response.content))
+                    st.image(img, caption=f"Slide {i + 1}", use_container_width=True)
+                else:
+                    st.warning(f"⚠️ Could not load image {i + 1}")
             except Exception as e:
                 st.error(f"❌ Failed to load image {i + 1}: {e}")