Update app.py

app.py

@@ -173,7 +173,7 @@ def get_pet_details(pet_id):
         return None
 
 # Function to format pet card
-def display_pet_card(pet, is_favorite=False):
+def display_pet_card(pet, is_favorite=False, context="search"):
     col1, col2 = st.columns([1, 2])
     
     with col1:
@@ -224,20 +224,18 @@ def display_pet_card(pet, is_favorite=False):
         
         col1, col2 = st.columns(2)
         with col1:
-            if st.button("View Details", key=f"details_{pet['id']}"):
-                if st.session_state.selected_pet != pet['id']:
-                    st.session_state.selected_pet = pet['id']
-                    st.rerun()  # Force immediate rerun
-
+            if st.button("View Details", key=f"details_{context}_{pet['id']}"):
+                st.session_state.selected_pet = pet['id']
+                st.stop()
         with col2:
             if not is_favorite:
-                if st.button("Add to Favorites", key=f"fav_{pet['id']}"):
+                if st.button("Add to Favorites", key=f"fav_{context}_{pet['id']}"):
                     if pet['id'] not in [p['id'] for p in st.session_state.favorites]:
                         st.session_state.favorites.append(pet)
                         st.success(f"Added {pet['name']} to favorites!")
                         st.rerun()
             else:
-                if st.button("Remove from Favorites", key=f"unfav_{pet['id']}"):
+                if st.button("Remove from Favorites", key=f"unfav_{context}_{pet['id']}"):
                     st.session_state.favorites = [p for p in st.session_state.favorites if p['id'] != pet['id']]
                     st.success(f"Removed {pet['name']} from favorites!")
                     st.rerun()