Update app.py

app.py

@@ -284,15 +284,15 @@ def get_compatibility_message(pet):
     return messages
 
 # Function to display pet details page
-def display_pet_details(pet_id):
+def display_pet_details(pet_id, context="search"):
     pet = get_pet_details(pet_id)
     if not pet:
         st.error("Unable to retrieve pet details. Please try again.")
         return
     
     # Back button
-    if st.button("← Back to Search Results", key=f"back_search_{pet_id}"):
-        if st.session_state.selected_pet is not None:
+    if st.button("← Back to Search Results", key=f"back_{context}_{pet_id}"):
+        #if st.session_state.selected_pet is not None:
             st.session_state.selected_pet = None
             st.rerun()  # Force immediate rerun
         # Pet name and status
@@ -397,7 +397,7 @@ def main():
     with tab1:
         # If a pet is selected, show details
         if st.session_state.selected_pet:
-            display_pet_details(st.session_state.selected_pet)
+            display_pet_details(st.session_state.selected_pet, context="search")
         else:
             # Search form
             with st.expander("Search Options", expanded=True):
@@ -502,7 +502,7 @@ def main():
         else:
             # Check if a pet is selected from favorites
             if st.session_state.selected_pet:
-                display_pet_details(st.session_state.selected_pet)
+                display_pet_details(st.session_state.selected_pet, context="search")
             else:
                 for pet in st.session_state.favorites:
                     st.markdown("---")