Update app.py

app.py

@@ -92,16 +92,27 @@ with st.sidebar:
 
     st.markdown("---")
     if st.button("🗑️ Clear Chat & Reset"):
-        # Cleanup Pinecone
-        cleanup_session_index(st.session_state.session_id)
-        # Reset State
-        st.session_state.session_id = str(uuid.uuid4())
-        st.session_state.doc_store = SessionDocStore()
-        st.session_state.messages = []
-        st.session_state.bm25 = None
-        st.session_state.processed_file = None
-        st.session_state.uploader_key = str(uuid.uuid4())
-        st.rerun()
+        try:
+            # 1. Attempt Cleanup with visual feedback
+            with st.spinner(f"Deleting vector data for {st.session_state.session_id}..."):
+                cleanup_session_index(st.session_state.session_id)
+            
+            # 2. Reset Local State (Only if cleanup succeeded)
+            st.session_state.session_id = str(uuid.uuid4())
+            st.session_state.doc_store = SessionDocStore()
+            st.session_state.messages = []
+            st.session_state.bm25 = None
+            st.session_state.processed_file = None
+            st.session_state.uploader_key = str(uuid.uuid4())
+            
+            # 3. Re-attach Janitor for new session
+            st.session_state.janitor = SessionJanitor(st.session_state.session_id)
+            
+            st.success("Cache cleared successfully!")
+            st.rerun()
+            
+        except Exception as e:
+            st.error(f"Cleanup failed! Check Pinecone Console.\nError: {e}")
 
     st.markdown("### ℹ️ How to Use")
     st.info(