Update app.py

app.py

@@ -35,11 +35,11 @@ if not st.session_state.authenticated:
     st.stop()
 
 # ------------------ App Configuration ------------------
-st.set_page_config(page_title="Pathology Assistant", layout="wide", initial_sidebar_state="collapsed")
+st.set_page_config(page_title="AI Pathology Assistant", layout="wide", initial_sidebar_state="collapsed")
 st.title("🧬 AI Pathology Assistant")
-st.caption("Explore histology, anatomy, and pathology documents using AI + OCR")
+st.caption("AI-powered exploration of pathology, anatomy, and histology documents via OCR + GPT")
 
-# ------------------ Load API Key ------------------
+# ------------------ 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.")
@@ -48,7 +48,7 @@ if not OPENAI_API_KEY:
 client = OpenAI(api_key=OPENAI_API_KEY)
 
 # ------------------ Assistant Configuration ------------------
-ASSISTANT_ID = "asst_jXDSjCG8LI4HEaFEcjFVq8KB"  # Replace with actual ID
+ASSISTANT_ID = "asst_YOUR_PATHOLOGY_ASSISTANT_ID"  # 🔁 Replace with your actual OpenAI assistant ID
 
 # ------------------ Session State ------------------
 if "messages" not in st.session_state:
@@ -64,7 +64,7 @@ if "pending_prompt" not in st.session_state:
 
 # ------------------ Sidebar ------------------
 with st.sidebar:
-    st.header("🔍 Pathology Tools")
+    st.header("🧪 Pathology Tools")
     if st.button("🧹 Clear Chat"):
         st.session_state.messages = []
         st.session_state.thread_id = None
@@ -73,12 +73,12 @@ with st.sidebar:
         st.session_state.pending_prompt = None
         st.rerun()
 
-    show_image = st.toggle("🖼️ Show Slide Image", value=True)
-    keyword = st.text_input("Keyword Search (e.g. mitosis, infarct, carcinoma)")
-    if st.button("🔎 Search Keyword") and keyword:
+    show_image = st.toggle("📸 Show Slide Image", 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}"
 
-    section = st.text_input("🧠 Section Lookup (e.g. Connective Tissue, Inflammation)")
+    section = st.text_input("Section Lookup", placeholder="e.g. Connective Tissue")
     if section:
         st.session_state.pending_prompt = f"Summarize or list key points from section: {section}"
 
@@ -90,11 +90,11 @@ with st.sidebar:
         "Extract diagnostic markers",
         "Summarize embryology stages"
     ]
-    action = st.selectbox("⚙️ Common Queries", actions)
+    action = st.selectbox("Common Pathology Queries", actions)
     if action != actions[0]:
         st.session_state.pending_prompt = action
 
-# ------------------ Main Chat Layout ------------------
+# ------------------ Main Chat UI ------------------
 chat_col, image_col = st.columns([2, 1])
 
 with chat_col:
@@ -123,7 +123,7 @@ with chat_col:
                 assistant_id=ASSISTANT_ID
             )
 
-            with st.spinner("🔬 Analyzing pathology data..."):
+            with st.spinner("🔬 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"):
@@ -140,12 +140,12 @@ with chat_col:
                         if match:
                             doc, page = match.group(1).strip(), int(match.group(2))
                             folder = quote(doc)
-                            img_url = f"https://raw.githubusercontent.com/AndrewLORTech/c2ozschlaegerforrestdale/main/{folder}/{folder}_page_{page:04d}.png"
+                            img_url = f"https://raw.githubusercontent.com/AndrewLORTech/witspathologai/main/{folder}/{folder}_page_{page:04d}.png"
                             st.session_state.image_url = img_url
                             st.session_state.image_updated = True
                         break
             else:
-                st.error("❌ Assistant failed.")
+                st.error("❌ Assistant failed to respond.")
             st.rerun()
         except Exception as e:
             st.error(f"❌ Error: {e}")
@@ -154,7 +154,7 @@ with chat_col:
         with st.chat_message(msg["role"]):
             st.markdown(msg["content"], unsafe_allow_html=True)
 
-# ------------------ Image Viewer ------------------
+# ------------------ Right Column: OCR Page Image ------------------
 with image_col:
     if show_image and st.session_state.image_url:
         try:
@@ -163,4 +163,4 @@ with image_col:
             img = Image.open(BytesIO(r.content))
             st.image(img, caption="🧾 OCR Page Image", use_container_width=True)
         except Exception as e:
-            st.error(f"🖼️ Image failed: {e}")
+            st.error(f"🖼️ Failed to load image: {e}")