Update app.py

app.py

@@ -1,3 +1,4 @@
+# ------------------ Import ------------------
 import streamlit as st
 import os
 import time
@@ -29,28 +30,24 @@ def login():
 
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
-
 if not st.session_state.authenticated:
     login()
     st.stop()
 
-# ------------------ App Configuration ------------------
+# ------------------ Configuration ------------------
 st.set_page_config(page_title="AI Pathology Assistant", layout="wide", initial_sidebar_state="collapsed")
 st.title("🧬 AI Pathology Assistant")
 st.caption("AI-powered exploration of pathology, anatomy, and histology documents via OCR + GPT")
 
-# ------------------ Load OpenAI API Key ------------------
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 if not OPENAI_API_KEY:
     st.error("❌ Missing OPENAI_API_KEY environment variable.")
     st.stop()
 
 client = OpenAI(api_key=OPENAI_API_KEY)
+ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"
 
-# ------------------ Assistant Configuration ------------------
-ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"  # Replace with your assistant ID
-
-# ------------------ Session State ------------------
+# ------------------ State Setup ------------------
 if "messages" not in st.session_state:
     st.session_state.messages = []
 if "thread_id" not in st.session_state:
@@ -70,7 +67,7 @@ with st.sidebar:
         st.session_state.pending_prompt = None
         st.rerun()
 
-    show_image = st.toggle("📸 Show Slide Images", value=True)
+    show_image = st.toggle("📸 Show Images", value=True)
     keyword = st.text_input("Keyword Search", placeholder="e.g. mitosis, carcinoma")
     if st.button("🔎 Search") and keyword:
         st.session_state.pending_prompt = f"Find clauses or references related to: {keyword}"
@@ -91,7 +88,7 @@ with st.sidebar:
     if action != actions[0]:
         st.session_state.pending_prompt = action
 
-# ------------------ Main Chat UI ------------------
+# ------------------ Chat UI ------------------
 chat_col, image_col = st.columns([2, 1])
 
 with chat_col:
@@ -132,22 +129,13 @@ 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})
 
-                        # ✂️ Optional: split out references
-                        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-hosted image links
-                        image_matches = re.findall(
-                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s\n]+\.png',
+                        matches = re.findall(
+                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s]+\.png',
                             reply
                         )
-                        st.session_state.image_urls = image_matches
+                        st.session_state.image_urls = matches
                         break
             else:
                 st.error("❌ Assistant failed to respond.")
@@ -156,31 +144,38 @@ 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)
 
-# ------------------ Slide Viewer (Horizontal Scroll) ------------------
+# ------------------ Scrollable Image Carousel ------------------
 with image_col:
     if show_image and st.session_state.image_urls:
-        st.markdown("### 🖼️ Slide Previews")
-
+        st.markdown("### 🖼️ Image(s)")
         st.markdown("""
         <style>
-        .image-scroll-container {
+        .carousel-wrapper {
             display: flex;
             overflow-x: auto;
-            gap: 16px;
+            scroll-snap-type: x mandatory;
+            gap: 1rem;
             padding: 0.5rem 0;
         }
-        .image-scroll-container img {
+        .carousel-wrapper::-webkit-scrollbar {
+            height: 10px;
+        }
+        .carousel-wrapper::-webkit-scrollbar-thumb {
+            background: #888;
+            border-radius: 5px;
+        }
+        .carousel-wrapper img {
+            scroll-snap-align: start;
             height: 500px;
-            border-radius: 10px;
+            border-radius: 8px;
         }
         </style>
         """, unsafe_allow_html=True)
 
-        st.markdown('<div class="image-scroll-container">', unsafe_allow_html=True)
-
+        html = '<div class="carousel-wrapper">'
         for raw_url in st.session_state.image_urls:
             try:
                 if "githubusercontent.com" not in raw_url:
@@ -189,9 +184,8 @@ with image_col:
                 segments = raw_path.strip().split("/")
                 encoded_segments = [quote(seg) for seg in segments]
                 encoded_url = "https://raw.githubusercontent.com/" + "/".join(encoded_segments)
-
-                st.markdown(f'<img src="{encoded_url}" alt="slide" />', unsafe_allow_html=True)
+                html += f'<img src="{encoded_url}" alt="slide preview">'
             except Exception as e:
                 st.error(f"❌ Failed to load image: {e}")
-
-        st.markdown('</div>', unsafe_allow_html=True)
+        html += "</div>"
+        st.markdown(html, unsafe_allow_html=True)