Update app.py

app.py

@@ -5,6 +5,7 @@ import json
 from datetime import datetime
 import time
 import os
+import functools
 
 # Set page configuration
 st.set_page_config(
@@ -103,6 +104,129 @@ def get_access_token():
         st.error(f"⚠️ Error getting access token: {str(e)}")
         return None
 
+# Function to get RescueGroups API key
+def get_rescuegroups_api_key():
+    api_key = os.environ.get('RESCUEGROUPS_API_KEY') or st.secrets.get('RESCUEGROUPS_API_KEY')
+    if not api_key:
+        st.error("⚠️ RescueGroups API key is missing. Please set it in environment variables or Streamlit secrets.")
+        return None
+    return api_key
+
+# Function to search pets from RescueGroups.org
+@functools.lru_cache(maxsize=32)
+def search_rescuegroups_pets_cached(type_param, location, distance):
+    api_key = get_rescuegroups_api_key()
+    if not api_key:
+        return None
+
+    url = "https://api.rescuegroups.org/v5/public/animals/search"
+
+    headers = {
+        "Authorization": api_key,
+        "Content-Type": "application/vnd.api+json"
+    }
+
+    body = {
+        "data": {
+            "type": "animals",
+            "attributes": {
+                "species": type_param,
+                "locationPostalcode": location,
+                "distance": distance,
+                "status": "Available",
+                "sort": "distance",
+                "limit": 100
+            }
+        }
+    }
+
+    try:
+        response = requests.post(url, headers=headers, json=body)
+        response.raise_for_status()
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        st.error(f"⚠️ Error searching pets from RescueGroups: {str(e)}")
+        return None
+
+def search_rescuegroups_pets(params):
+    return search_rescuegroups_pets_cached(params.get("type", "Dog"), params.get("location", ""), params.get("distance", 50))
+
+# Function to normalize RescueGroups results
+def normalize_rescuegroups_data(rg_data):
+    pets = []
+    if not rg_data or 'data' not in rg_data:
+        return pets
+
+    for item in rg_data['data']:
+        pet = {
+            'id': f"RG-{item['id']}",
+            'name': item['attributes'].get('name', 'Unknown'),
+            'status': 'adoptable',
+            'age': item['attributes'].get('ageGroup', ''),
+            'gender': item['attributes'].get('sex', ''),
+            'size': item['attributes'].get('sizeGroup', ''),
+            'breeds': {
+                'primary': item['attributes'].get('breedPrimary', ''),
+                'secondary': item['attributes'].get('breedSecondary', ''),
+                'mixed': False
+            },
+            'colors': {
+                'primary': item['attributes'].get('colorPrimary', ''),
+                'secondary': item['attributes'].get('colorSecondary', ''),
+                'tertiary': ''
+            },
+            'photos': [],
+            'description': item['attributes'].get('description', ''),
+            'contact': {
+                'email': '',
+                'phone': '',
+                'address': {
+                    'city': '',
+                    'state': '',
+                    'postcode': ''
+                }
+            },
+            'source': 'RescueGroups'
+        }
+        if 'photos' in item['attributes'] and item['attributes']['photos']:
+            pet['photos'] = [{'medium': item['attributes']['photos'][0]}]
+        pets.append(pet)
+    return pets
+
+# Example modification in your search form to call BOTH APIs
+with st.form("pet_search_form"):
+    # ... your existing form fields ...
+    submitted = st.form_submit_button("Search")
+
+    if submitted:
+        params = {
+            "type": animal_type.split(" ")[0],
+            "location": location,
+            "distance": distance
+        }
+        
+        petfinder_results = search_pets(params)
+        rescuegroups_results = search_rescuegroups_pets(params)
+        
+        combined_pets = []
+        
+        if petfinder_results and 'animals' in petfinder_results:
+            for pet in petfinder_results['animals']:
+                pet['source'] = 'Petfinder'
+                combined_pets.append(pet)
+        if rescuegroups_results:
+            normalized = normalize_rescuegroups_data(rescuegroups_results)
+            combined_pets.extend(normalized)
+
+        if combined_pets:
+            st.session_state.search_results = {'animals': combined_pets}
+            st.session_state.page = 1
+            st.success(f"Found {len(combined_pets)} pets from both sources!")
+        else:
+            st.error("No pets found. Try adjusting your filters.")
+
+
+
 # Function to search pets
 def search_pets(params):
     token = get_access_token()