Update app.py

app.py

@@ -163,7 +163,7 @@ with tab1:
                 except Exception:
                     continue
 
-# ------------------ Tab 2: Visual Search ------------------
+# ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     st.title("🔍 Visual Reference Search")
     user_query = st.text_input("Enter keyword to search images (e.g. ovary, thyroid, mitosis)")
@@ -220,24 +220,29 @@ with tab2:
         filename_matches = re.findall(r'Image Filename:\s*(.*?\.png)', st.session_state.image_response)
 
         for fname in filename_matches:
-            # Extract folder from filename prefix (before _page_)
-            parts = fname.split("_page_")
-            if len(parts) > 1:
-                folder = parts[0]
+            # Safely extract folder by splitting BEFORE the _page_ part
+            match = re.match(r'^(.+?)_page_', fname)
+            if match:
+                folder = match.group(1)
             else:
                 st.warning(f"⚠️ Could not determine folder for: {fname}")
                 continue
 
-            # Only encode spaces — leave parentheses untouched to match GitHub structure
+            # Encode only spaces – DO NOT encode parentheses or underscores
             encoded_folder = folder.replace(" ", "%20")
             encoded_fname = fname.replace(" ", "%20")
 
             url = f"https://raw.githubusercontent.com/AndrewLORTech/witspathologaihisto/main/{encoded_folder}/{encoded_fname}"
 
+            # Optional: show the raw image URL for debugging
+            st.markdown(f"🔗 [View image directly]({url})")
+
             try:
                 r = requests.get(url, timeout=5)
                 r.raise_for_status()
                 img = Image.open(BytesIO(r.content))
                 st.image(img, caption=fname, use_container_width=True)
-            except Exception:
+            except Exception as e:
                 st.warning(f"⚠️ Failed to load: {url}")
+                st.error(f"🛑 Error: {str(e)}")
+