Update app.py

app.py

@@ -5,7 +5,6 @@ import re
 import requests
 from PIL import Image
 from io import BytesIO
-from urllib.parse import quote
 from openai import OpenAI
 
 # ------------------ Authentication ------------------
@@ -133,7 +132,7 @@ with chat_col:
                     if m.role == "assistant":
                         reply = m.content[0].text.value
 
-                        # ✂️ Split output into main and references (if structured this way)
+                        # ✂️ Split output into main and references
                         split_marker = "---"
                         if split_marker in reply:
                             main_answer, references = reply.split(split_marker, 1)
@@ -142,7 +141,7 @@ with chat_col:
                         else:
                             st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                        # Extract GitHub image URLs
+                        # Extract GitHub image URLs directly
                         image_matches = re.findall(
                             r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n]+\.png',
                             reply
@@ -159,19 +158,14 @@ with chat_col:
         with st.chat_message("assistant" if msg["role"] != "references" else "📄"):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Image Viewer (Scrollable Format) ------------------
+# ------------------ Image Viewer ------------------
 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 "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 = "https://raw.githubusercontent.com/" + "/".join(encoded_segments)
-
-                st.image(encoded_url, caption=f"Slide {i + 1}", use_column_width=True)
+                # Optional: Log image URLs for debugging
+                # st.write(f"Image URL {i+1}: {raw_url}")
+                st.image(raw_url, caption=f"Slide {i + 1}", use_container_width=True)
             except Exception as e:
                 st.error(f"❌ Failed to load image: {e}")