Update app.py

app.py

@@ -26,8 +26,6 @@ if "json_links" not in st.session_state:
     st.session_state.json_links = []
 if "json_link_details" not in st.session_state:
     st.session_state.json_link_details = {}
-if "expanded_json" not in st.session_state:
-    st.session_state.expanded_json = set()
 
 st.set_page_config(page_title="Chat with Your JSON Vectors (Hybrid, Clean)", layout="wide")
 st.title("Chat with Your Vectorized JSON Files")
@@ -243,19 +241,13 @@ def show_tiny_json_links():
     # Only show for the last assistant answer if there are matching JSONs
     if not st.session_state.json_links:
         return
-    st.write("")
     for idx, link_key in enumerate(st.session_state.json_links):
         label = st.session_state.json_link_details[link_key]['label']
         rec = st.session_state.json_link_details[link_key]['record']
-        unique_id = f"{link_key}_{idx}"
-        link_text = f"<a href='javascript:void(0);' style='font-size: 11px; color: #555; text-decoration: underline;' onclick=\"document.getElementById('{unique_id}').style.display = (document.getElementById('{unique_id}').style.display === 'none' ? 'block' : 'none')\">[view JSON]</a> <span style='font-size: 10px; color: #999'>{label}</span>"
-        st.markdown(link_text, unsafe_allow_html=True)
-        if unique_id not in st.session_state.expanded_json:
-            st.session_state.expanded_json.remove(unique_id) if unique_id in st.session_state.expanded_json else None
-        if st.session_state.get("show_" + unique_id, False):
-            st.code(json.dumps(rec, indent=2), language="json", key=unique_id)
-        else:
-            st.markdown(f"<div id='{unique_id}' style='display:none'>{json.dumps(rec, indent=2)}</div>", unsafe_allow_html=True)
+        expander_label = f"<span style='font-size:11px; color:#444; text-decoration:underline;'>[view JSON]</span> <span style='font-size:10px; color:#aaa'>{label}</span>"
+        with st.expander(label="", expanded=False):
+            st.markdown(expander_label, unsafe_allow_html=True)
+            st.code(json.dumps(rec, indent=2), language="json")
     st.session_state.json_links = []
     st.session_state.json_link_details = {}
 
@@ -294,6 +286,5 @@ if st.button("Clear chat"):
     st.session_state.messages = []
     st.session_state.json_links = []
     st.session_state.json_link_details = {}
-    st.session_state.expanded_json = set()
 
 st.info(f"Batches ingested so far (this session): {st.session_state.ingested_batches}")