Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import time
 import re
 import requests
+import urllib.parse
 from PIL import Image
 from io import BytesIO
 from openai import OpenAI
@@ -47,7 +48,7 @@ if not OPENAI_API_KEY:
 client = OpenAI(api_key=OPENAI_API_KEY)
 
 # ------------------ Assistant Configuration ------------------
-ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"  # Replace with your Assistant ID
+ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"
 
 # ------------------ Session State ------------------
 if "messages" not in st.session_state:
@@ -133,7 +134,7 @@ with chat_col:
                         reply = m.content[0].text.value
                         st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                        # 🔍 Extract GitHub raw image URLs
+                        # Extract raw GitHub image URLs
                         image_matches = re.findall(
                             r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n"]+\.png',
                             reply
@@ -150,15 +151,23 @@ with chat_col:
         with st.chat_message(msg["role"]):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Image Preview (Silent Fails) ------------------
+# ------------------ Image Preview (Smart URL 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:
-                r = requests.get(raw_url, timeout=5)
+                if raw_url.startswith("https://raw.githubusercontent.com/"):
+                    base_url = "https://raw.githubusercontent.com/"
+                    path = raw_url[len(base_url):]
+                    encoded_path = urllib.parse.quote(path, safe="/")
+                    encoded_url = base_url + encoded_path
+                else:
+                    encoded_url = raw_url
+
+                r = requests.get(encoded_url, timeout=5)
                 r.raise_for_status()
                 img = Image.open(BytesIO(r.content))
-                st.image(img, caption=f"📷 {raw_url.split('/')[-1]}", use_container_width=True)
+                st.image(img, caption=f"📷 {encoded_url.split('/')[-1]}", use_container_width=True)
             except Exception:
                 continue  # Silently skip broken images