Update app.py

app.py

@@ -48,7 +48,7 @@ if not OPENAI_API_KEY:
 client = OpenAI(api_key=OPENAI_API_KEY)
 
 # ------------------ Assistant Configuration ------------------
-ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"  # Replace with your actual assistant ID
+ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"  # Replace with your Assistant ID
 
 # ------------------ Session State ------------------
 if "messages" not in st.session_state:
@@ -132,16 +132,9 @@ with chat_col:
                 for m in reversed(messages.data):
                     if m.role == "assistant":
                         reply = m.content[0].text.value
+                        st.session_state.messages.append({"role": "assistant", "content": reply})
 
-                        split_marker = "---"
-                        if split_marker in reply:
-                            main_answer, references = reply.split(split_marker, 1)
-                            st.session_state.messages.append({"role": "assistant", "content": main_answer.strip()})
-                            st.session_state.messages.append({"role": "references", "content": references.strip()})
-                        else:
-                            st.session_state.messages.append({"role": "assistant", "content": reply})
-
-                        # Extract raw GitHub URLs
+                        # 🔍 Extract all GitHub image URLs
                         image_matches = re.findall(
                             r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n]+\.png',
                             reply
@@ -155,27 +148,25 @@ with chat_col:
             st.error(f"❌ Error: {e}")
 
     for msg in st.session_state.messages:
-        with st.chat_message("assistant" if msg["role"] != "references" else "📄"):
+        with st.chat_message(msg["role"]):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Image Viewer (requests + PIL fallback) ------------------
+# ------------------ Image Preview with Safe Encoding ------------------
 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):
+        for raw_url in st.session_state.image_urls:
             try:
-                # 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)
+                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:
-                    st.warning(f"⚠️ Could not load image {i + 1}")
+                    encoded_url = raw_url
+
+                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)
             except Exception as e:
-                st.error(f"❌ Failed to load image {i + 1}: {e}")
+                st.error(f"❌ Failed to load image from {raw_url}: {e}")