Update app.py

app.py

@@ -167,12 +167,10 @@ with tab1:
                 except Exception:
                     continue
 
-
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
-    ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"  # Your visual reference assistant ID
+    ASSISTANT_ID = "asst_9v09zgizdcuuhNdcFQpRo9RO"
 
-    # Session states
     if "image_thread_id" not in st.session_state:
         st.session_state.image_thread_id = None
     if "image_response" not in st.session_state:
@@ -182,7 +180,6 @@ with tab2:
     if "image_lightbox" not in st.session_state:
         st.session_state.image_lightbox = None
 
-    # User input
     image_input = st.chat_input("Ask for histology visual references (e.g. ovary histology, mitosis)")
     if image_input:
         st.session_state.image_response = None
@@ -222,7 +219,6 @@ with tab2:
                         response_text = msg.content[0].text.value
                         st.session_state.image_response = response_text
 
-                        # 🔍 Robustly extract "Image URL" entries
                         lines = response_text.splitlines()
                         image_urls = []
                         for i, line in enumerate(lines):
@@ -239,7 +235,6 @@ with tab2:
         except Exception as e:
             st.error(f"❌ Visual Assistant Error: {e}")
 
-    # Layout split
     text_col, image_col = st.columns([2, 1])
 
     with text_col:
@@ -254,9 +249,13 @@ with tab2:
                 image_url = item.get("image")
                 if image_url:
                     try:
-                        st.image(image_url, caption=image_url.split("/")[-1], use_container_width=True)
-                        if st.button("🖼️ View Full Image", key=f"img_{i}"):
-                            st.session_state.image_lightbox = image_url
+                        encoded_url = requests.utils.requote_uri(image_url)
+                        r = requests.get(encoded_url, timeout=10)
+                        r.raise_for_status()
+                        img = Image.open(BytesIO(r.content))
+                        st.image(img, caption=encoded_url.split("/")[-1], use_container_width=True)
+                        if st.button("🔍 View Full Image", key=f"full_img_{i}"):
+                            st.session_state.image_lightbox = encoded_url
                     except Exception as e:
                         st.warning(f"⚠️ Could not load: {image_url}")
                         st.error(f"{e}")
@@ -264,7 +263,17 @@ with tab2:
             st.info("ℹ️ No image references found yet.")
 
     if st.session_state.image_lightbox:
-        st.image(st.session_state.image_lightbox, caption="🔍 Full Image View", use_container_width=True)
+        st.markdown("### 🔬 Full Image View")
+        try:
+            img_url = st.session_state.image_lightbox
+            r = requests.get(img_url, timeout=10)
+            r.raise_for_status()
+            full_img = Image.open(BytesIO(r.content))
+            st.image(full_img, caption=img_url.split("/")[-1], use_container_width=True)
+        except Exception as e:
+            st.warning("⚠️ Could not load full image.")
+            st.error(str(e))
         if st.button("❌ Close Preview"):
             st.session_state.image_lightbox = None
             st.rerun()
+