Update app.py

app.py

@@ -1,4 +1,3 @@
-# ------------------ Import ------------------
 import streamlit as st
 import os
 import time
@@ -18,7 +17,7 @@ VALID_USERS = {
 }
 
 def login():
-    st.title("🔐 Login Required")
+    st.title("\U0001F512 Login Required")
     email = st.text_input("Email")
     password = st.text_input("Password", type="password")
     if st.button("Login"):
@@ -26,7 +25,7 @@ def login():
             st.session_state.authenticated = True
             st.rerun()
         else:
-            st.error("❌ Incorrect email or password.")
+            st.error("\u274C Incorrect email or password.")
 
 if "authenticated" not in st.session_state:
     st.session_state.authenticated = False
@@ -36,12 +35,12 @@ if not st.session_state.authenticated:
 
 # ------------------ Configuration ------------------
 st.set_page_config(page_title="AI Pathology Assistant", layout="wide", initial_sidebar_state="collapsed")
-st.title("🧬 AI Pathology Assistant")
+st.title("\U0001F9EC AI Pathology Assistant")
 st.caption("AI-powered exploration of pathology, anatomy, and histology documents via OCR + GPT")
 
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 if not OPENAI_API_KEY:
-    st.error("❌ Missing OPENAI_API_KEY environment variable.")
+    st.error("\u274C Missing OPENAI_API_KEY environment variable.")
     st.stop()
 
 client = OpenAI(api_key=OPENAI_API_KEY)
@@ -59,17 +58,17 @@ if "pending_prompt" not in st.session_state:
 
 # ------------------ Sidebar ------------------
 with st.sidebar:
-    st.header("🧪 Pathology Tools")
-    if st.button("🧹 Clear Chat"):
+    st.header("\U0001F9EA Pathology Tools")
+    if st.button("\U0001F9F9 Clear Chat"):
         st.session_state.messages = []
         st.session_state.thread_id = None
         st.session_state.image_urls = []
         st.session_state.pending_prompt = None
         st.rerun()
 
-    show_image = st.toggle("📸 Show Images", value=True)
+    show_image = st.toggle("\U0001F4F8 Show Slide Images", value=True)
     keyword = st.text_input("Keyword Search", placeholder="e.g. mitosis, carcinoma")
-    if st.button("🔎 Search") and keyword:
+    if st.button("\U0001F50E Search") and keyword:
         st.session_state.pending_prompt = f"Find clauses or references related to: {keyword}"
 
     section = st.text_input("Section Lookup", placeholder="e.g. Connective Tissue")
@@ -92,7 +91,7 @@ with st.sidebar:
 chat_col, image_col = st.columns([2, 1])
 
 with chat_col:
-    st.markdown("### 💬 Ask a Pathology-Specific Question")
+    st.markdown("### \U0001F4AC Ask a Pathology-Specific Question")
     user_input = st.chat_input("Example: What are features of squamous cell carcinoma?")
     if user_input:
         st.session_state.messages.append({"role": "user", "content": user_input})
@@ -117,7 +116,7 @@ with chat_col:
                 assistant_id=ASSISTANT_ID
             )
 
-            with st.spinner("🔬 Analyzing..."):
+            with st.spinner("\U0001F52C Analyzing..."):
                 while True:
                     status = client.beta.threads.runs.retrieve(thread_id=st.session_state.thread_id, run_id=run.id)
                     if status.status in ("completed", "failed", "cancelled"):
@@ -128,29 +127,35 @@ with chat_col:
                 messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
                 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})
+                        raw_reply = m.content[0].text.value
 
-                        matches = re.findall(
-                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^\s]+\.png',
-                            reply
+                        image_matches = re.findall(
+                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^"]+?\.png',
+                            raw_reply
                         )
-                        st.session_state.image_urls = matches
+                        st.session_state.image_urls = image_matches
+
+                        reply_cleaned = re.sub(
+                            r'https://raw\.githubusercontent\.com/AndrewLORTech/witspathologai/main/[^"]+?\.png',
+                            '[Image reference available in Slide Previews →]',
+                            raw_reply
+                        )
+                        st.session_state.messages.append({"role": "assistant", "content": reply_cleaned})
                         break
             else:
-                st.error("❌ Assistant failed to respond.")
+                st.error("\u274C Assistant failed to respond.")
             st.rerun()
         except Exception as e:
-            st.error(f"❌ Error: {e}")
+            st.error(f"\u274C Error: {e}")
 
     for msg in st.session_state.messages:
         with st.chat_message(msg["role"]):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Scrollable Image Carousel ------------------
+# ------------------ Scrollable Image Preview ------------------
 with image_col:
     if show_image and st.session_state.image_urls:
-        st.markdown("### 🖼️ Image(s)")
+        st.markdown("### \U0001F5BC Image(s)")
         st.markdown("""
         <style>
         .carousel-wrapper {
@@ -161,15 +166,15 @@ with image_col:
             padding: 0.5rem 0;
         }
         .carousel-wrapper::-webkit-scrollbar {
-            height: 10px;
+            height: 8px;
         }
         .carousel-wrapper::-webkit-scrollbar-thumb {
-            background: #888;
-            border-radius: 5px;
+            background: #999;
+            border-radius: 6px;
         }
         .carousel-wrapper img {
             scroll-snap-align: start;
-            height: 500px;
+            height: 420px;
             border-radius: 8px;
         }
         </style>
@@ -178,14 +183,12 @@ with image_col:
         html = '<div class="carousel-wrapper">'
         for raw_url in 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)
                 html += f'<img src="{encoded_url}" alt="slide preview">'
             except Exception as e:
-                st.error(f"❌ Failed to load image: {e}")
+                st.error(f"\u274C Failed to load image: {e}")
         html += "</div>"
-        st.markdown(html, unsafe_allow_html=True)
+        st.markdown(html, unsafe_allow_html=True)