Update app.py

app.py

@@ -213,36 +213,22 @@ with tab2:
         except Exception as e:
             st.error(f"Error: {e}")
 
-    # Render assistant response + build GitHub image URLs from filenames
+    # Render assistant response and display images using 'Image URL' field
     if st.session_state.image_response:
         st.markdown("### 🧠 Assistant Response")
         st.markdown(st.session_state.image_response, unsafe_allow_html=True)
 
-        # Extract image filenames from assistant response
-        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_
-            match = re.match(r'^(.+?)_page_', fname)
-            if match:
-                folder = match.group(1).strip()
-            else:
-                st.warning(f"⚠️ Could not determine folder for: {fname}")
-                continue
-
-            # ✅ Use URL-safe encoding for both folder and filename
-            encoded_folder = quote(folder)
-            encoded_fname = quote(fname)
-
-            url = f"https://raw.githubusercontent.com/AndrewLORTech/witspathologaihisto/main/{encoded_folder}/{encoded_fname}"
+        # Extract full GitHub URLs from assistant output
+        url_matches = re.findall(r'Image URL:\s*(https?://[^\s]+)', st.session_state.image_response)
 
+        for url in url_matches:
             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)
+                st.image(img, caption=url.split("/")[-1], use_container_width=True)
             except Exception as e:
                 st.warning(f"⚠️ Failed to load: {url}")
                 st.error(f"🛑 Error: {str(e)}")