Update app.py

app.py

@@ -121,6 +121,63 @@ def search_pets(params):
         st.error(f"⚠️ Error searching pets: {str(e)}")
         return None
 
+# Function to search pets from RescueGroups.org
+def search_rescuegroups_pets(params):
+    api_key = os.environ.get('RESCUEGROUPS_API_KEY') or st.secrets.get('RESCUEGROUPS_API_KEY')
+    if not api_key:
+        st.error("⚠️ RescueGroups API key missing. Please set it.")
+        return []
+
+    url = "https://api.rescuegroups.org/v5/public/animals/search/available"
+
+    headers = {
+        "Authorization": f"Bearer {api_key}",
+        "Content-Type": "application/vnd.api+json"
+    }
+
+    # Build RescueGroups.org query (adjust as needed based on API spec)
+    query_params = {
+        "filter[species]": params.get("type", ""),
+        "filter[location]": params.get("location", ""),
+        "page[limit]": 100
+    }
+
+    try:
+        response = requests.get(url, headers=headers, params=query_params)
+        response.raise_for_status()
+        animals = response.json().get('data', [])
+        formatted_pets = []
+
+        # Normalize result
+        for pet in animals:
+            formatted_pet = {
+                'id': f"rg_{pet['id']}",
+                'name': pet['attributes']['name'],
+                'status': 'adoptable',
+                'age': pet['attributes']['ageGroup'],
+                'gender': pet['attributes']['sex'],
+                'size': pet['attributes']['size'],
+                'photos': [{'medium': pet['attributes'].get('pictureThumbnailUrl', '')}] if pet['attributes'].get('pictureThumbnailUrl') else [],
+                'description': pet['attributes']['descriptionText'],
+                'breeds': {'primary': pet['attributes']['breedPrimary'], 'secondary': pet['attributes']['breedSecondary'], 'mixed': False},
+                'colors': {'primary': None, 'secondary': None, 'tertiary': None},
+                'contact': {
+                    'address': {'city': pet['attributes'].get('locationCity', ''), 'state': pet['attributes'].get('locationState', ''), 'postcode': ''},
+                    'email': pet['attributes'].get('contactEmail', ''),
+                    'phone': pet['attributes'].get('contactPhone', '')
+                },
+                'organization_id': pet['relationships'].get('orgs', {}).get('data', [{}])[0].get('id', ''),
+                'source': 'RescueGroups'
+            }
+            formatted_pets.append(formatted_pet)
+
+        return formatted_pets
+
+    except requests.exceptions.RequestException as e:
+        st.error(f"⚠️ Error searching RescueGroups pets: {str(e)}")
+        return []
+
+
 # Function to get breeds
 def get_breeds(animal_type):
     token = get_access_token()
@@ -199,6 +256,9 @@ def display_pet_card(pet, is_favorite=False, context="search"):
             tags_html += f"<span class='tag'>{pet['gender']}</span> "
         if pet['size']:
             tags_html += f"<span class='tag'>{pet['size']}</span> "
+        # Attribution source tag
+        if 'source' in pet:
+            st.markdown(f"<span class='tag' style='background-color: #444;'>Source: {pet['source']}</span>", unsafe_allow_html=True)
         
         st.markdown(f"<div>{tags_html}</div>", unsafe_allow_html=True)
         
@@ -417,7 +477,10 @@ def display_pet_card(pet, is_favorite=False, context="search", tab_id="tab1"):
             tags_html += f"<span class='tag'>{pet['gender']}</span> "
         if pet['size']:
             tags_html += f"<span class='tag'>{pet['size']}</span> "
-        
+        # Attribution source tag
+        if 'source' in pet:
+            st.markdown(f"<span class='tag' style='background-color: #444;'>Source: {pet['source']}</span>", unsafe_allow_html=True)
+    
         st.markdown(f"<div>{tags_html}</div>", unsafe_allow_html=True)
         
         st.markdown("<div class='pet-details'>", unsafe_allow_html=True)
@@ -537,9 +600,25 @@ def main():
                         
                         # Perform search
                         results = search_pets(params)
+                        combined_results = []
+
                         if results and 'animals' in results:
-                            st.session_state.search_results = results
+                            pf_pets = results['animals']
+                            # Add source attribution
+                            for pet in pf_pets:
+                                pet['source'] = 'PetFinder'
+                        combined_results.extend(pf_pets)
+
+                        # Fetch from RescueGroups
+                        rg_pets = search_rescuegroups_pets(params)
+                        combined_results.extend(rg_pets)
+
+                        if combined_results:
+                            st.session_state.search_results = {'animals': combined_results}
                             st.session_state.page = 1
+                            st.success(f"Found {len(combined_results)} pets from PetFinder and RescueGroups!")
+                        else:
+                            st.error("No pets found with those criteria.")
                             st.success(f"Found {len(results['animals'])} pets!")
                         else:
                             st.error("No pets found with those criteria. Try expanding your search.")