Update app.py

app.py

@@ -25,6 +25,18 @@ COLLECTION_NAME = "spotify_songs"
 
 st.set_page_config(page_title="Spotify Similarity Search", page_icon="🎵", layout="wide")
 
+def reset_environment():
+    # Clear all st.cache_resource and st.cache_data
+    st.cache_resource.clear()
+    st.cache_data.clear()
+    
+    # Clear all items in session state
+    for key in list(st.session_state.keys()):
+        del st.session_state[key]
+    
+    # Generate a new session ID
+    st.session_state.session_id = str(uuid.uuid4())
+
 @st.cache_resource
 def load_resources():
     return AudioEncoder()
@@ -260,6 +272,8 @@ def main():
     
     st.title("Spotify Similarity Search")
 
+    if 'session_id' not in st.session_state:
+        reset_environment()
     
     qdrant_client = get_qdrant_client()