Update app.py

app.py

@@ -284,6 +284,7 @@ def get_compatibility_message(pet):
     return messages
 
 # Function to display pet details page
+# The fix is in the display_pet_details function
 def display_pet_details(pet_id, context="search"):
     pet = get_pet_details(pet_id)
     if not pet:
@@ -292,11 +293,11 @@ def display_pet_details(pet_id, context="search"):
     
     # Back button
     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
-            st.markdown(f"<h1 class='main-header'>{pet['name']}</h1>", unsafe_allow_html=True)
+        st.session_state.selected_pet = None
+        st.rerun()  # Force immediate rerun
+    
+    # Pet name and status
+    st.markdown(f"<h1 class='main-header'>{pet['name']}</h1>", unsafe_allow_html=True)
     
     status_color = "#c8e6c9" if pet['status'] == 'adoptable' else "#ffcdd2"
     st.markdown(f"<div style='text-align: center;'><span class='tag' style='background-color: {status_color}; font-size: 1rem;'>{pet['status'].title()}</span></div>", unsafe_allow_html=True)
@@ -328,15 +329,15 @@ def display_pet_details(pet_id, context="search"):
             f"**Age:** {pet['age']}",
             f"**Gender:** {pet['gender']}",
             f"**Size:** {pet['size']}"
-    ]
+        ]
     
-    # Fix the colors line as well, to be safe
-    colors = [c for c in [pet['colors']['primary'], pet['colors']['secondary'], pet['colors']['tertiary']] if c]
-    if colors:
-        details.append(f"**Colors:** {', '.join(colors)}")
+        # Fix the colors line as well, to be safe
+        colors = [c for c in [pet['colors']['primary'], pet['colors']['secondary'], pet['colors']['tertiary']] if c]
+        if colors:
+            details.append(f"**Colors:** {', '.join(colors)}")
     
-    for detail in details:
-        st.markdown(detail)
+        for detail in details:
+            st.markdown(detail)
     
     with col2:
         st.markdown("### Compatibility")
@@ -375,12 +376,12 @@ def display_pet_details(pet_id, context="search"):
     # Add to favorites
     is_favorite = pet['id'] in [p['id'] for p in st.session_state.favorites]
     if not is_favorite:
-        if st.button("Add to Favorites"):
+        if st.button("Add to Favorites", key=f"add_fav_{context}_{pet_id}"):
             st.session_state.favorites.append(pet)
             st.success(f"Added {pet['name']} to favorites!")
             st.rerun()
     else:
-        if st.button("Remove from Favorites"):
+        if st.button("Remove from Favorites", key=f"rem_fav_{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()