Update app.py

app.py

@@ -225,8 +225,9 @@ 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.experimental_rerun()  # <-- replace st.stop() with this
+                st.experimental_rerun()  # Force immediate rerun
 
         with col2:
             if not is_favorite:
@@ -293,8 +294,9 @@ def display_pet_details(pet_id):
     
     # Back button
     if st.button("← Back to Search Results"):
+    if st.session_state.selected_pet is not None:
         st.session_state.selected_pet = None
-        st.experimental_rerun()  # <-- replace st.stop() with this
+        st.experimental_rerun()  # Force immediate rerun
 
     # Pet name and status
     st.markdown(f"<h1 class='main-header'>{pet['name']}</h1>", unsafe_allow_html=True)