Update app.py

app.py

@@ -2,9 +2,6 @@ import streamlit as st
 import os
 import time
 import re
-import requests
-from PIL import Image
-from io import BytesIO
 from openai import OpenAI
 
 # ------------------ Authentication ------------------
@@ -141,12 +138,14 @@ with chat_col:
                         else:
                             st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                        # Extract GitHub image URLs directly
+                        # Extract GitHub image URLs
                         image_matches = re.findall(
-                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n]+\.png',
+                            r'https://github\.com/AndrewLORTech/witspathologai/blob/main/[^\s\n]+\.png',
                             reply
                         )
-                        st.session_state.image_urls = image_matches
+                        # Transform to raw.githubusercontent.com URLs
+                        raw_urls = [url.replace("github.com", "raw.githubusercontent.com").replace("/blob/", "/") for url in image_matches]
+                        st.session_state.image_urls = raw_urls
                         break
             else:
                 st.error("❌ Assistant failed to respond.")
@@ -164,8 +163,6 @@ with image_col:
         st.markdown("### 🖼️ Slide Previews")
         for i, raw_url in enumerate(st.session_state.image_urls):
             try:
-                # 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}")