Update app.py

app.py

@@ -162,7 +162,6 @@ with tab1:
                     st.image(img, caption=f"📷 {raw_url.split('/')[-1]}", use_container_width=True)
                 except Exception:
                     continue
-
 # ------------------ Tab 2: Visual Reference Search ------------------
 with tab2:
     st.title("🔍 Visual Reference Search")
@@ -220,23 +219,25 @@ with tab2:
         filename_matches = re.findall(r'Image Filename:\s*(.*?\.png)', st.session_state.image_response)
 
         for fname in filename_matches:
-            # Safely extract folder by splitting BEFORE the _page_ part
+            # Extract folder name from filename prefix (clean + safe)
             match = re.match(r'^(.+?)_page_', fname)
             if match:
-                folder = match.group(1)
+                folder = match.group(1).strip()
             else:
                 st.warning(f"⚠️ Could not determine folder for: {fname}")
                 continue
 
-            # Encode only spaces – DO NOT encode parentheses or underscores
+            # Encode only spaces – parentheses/underscores remain as-is
             encoded_folder = folder.replace(" ", "%20")
             encoded_fname = fname.replace(" ", "%20")
 
+            # Final raw GitHub URL
             url = f"https://raw.githubusercontent.com/AndrewLORTech/witspathologaihisto/main/{encoded_folder}/{encoded_fname}"
 
-            # Optional: show the raw image URL for debugging
+            # Debug: show link
             st.markdown(f"🔗 [View image directly]({url})")
 
+            # Try loading and displaying image
             try:
                 r = requests.get(url, timeout=5)
                 r.raise_for_status()