Update app.py

app.py

@@ -33,6 +33,8 @@ 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")
@@ -41,6 +43,7 @@ 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 ------------------
@@ -115,9 +118,15 @@ if st.session_state.results:
             st.caption(item.get("summary", ""))
             for url in item.get("images", [])[:1]:
                 if st.button("🖼️ View Image", key=f"view_{i}"):
-                    st.image(url, use_column_width=True)
-
+                    st.session_state.lightbox_url = url
+
+    if st.session_state.lightbox_url:
+        st.markdown("---")
+        st.image(st.session_state.lightbox_url, use_column_width=True, caption="🔍 Enlarged Preview")
+        if st.button("❌ Close Viewer"):
+            st.session_state.lightbox_url = None
+            st.rerun()
 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)