Update app.py

app.py

@@ -33,8 +33,6 @@ if "pending_prompt" not in st.session_state:
     st.session_state.pending_prompt = None
 if "results" not in st.session_state:
     st.session_state.results = []
-if "lightbox_url" not in st.session_state:
-    st.session_state.lightbox_url = None
 
 # ------------------ Sidebar ------------------
 st.sidebar.header("ℹ️ Information")
@@ -43,7 +41,6 @@ if st.sidebar.button("🧹 Clear Chat"):
     st.session_state.thread_id = None
     st.session_state.pending_prompt = None
     st.session_state.results = []
-    st.session_state.lightbox_url = None
     st.rerun()
 
 # ------------------ Chat Input ------------------
@@ -117,9 +114,8 @@ if st.session_state.results:
             st.markdown(f"_Discipline: {item['discipline']}_")
             st.caption(item.get("summary", ""))
             for url in item.get("images", [])[:1]:
-                st.image(url, use_column_width=True)
-
+                st.image(url, caption=f"{item['drawing_number']} – Page 1", use_column_width=True)
 else:
     for msg in st.session_state.messages:
         with st.chat_message(msg["role"]):
-            st.markdown(msg["content"], unsafe_allow_html=True)
+            st.markdown(msg["content"], unsafe_allow_html=True)