Update app.py

app.py

@@ -603,725 +603,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-import os
-import functools
-
-# Set page configuration
-st.set_page_config(
-    page_title="PetMatch - Find Your Perfect Pet",
-    page_icon="🐾",
-    layout="centered"
-)
-
-# Custom CSS
-st.markdown("""
-<style>
-    .main-header {
-        font-size: 2.5rem;
-        color: #ff6b6c;
-        text-align: center;
-        margin-bottom: 1rem;
-    }
-    .sub-header {
-        font-size: 1.5rem;
-        color: #4a4a4a;
-        text-align: center;
-        margin-bottom: 2rem;
-    }
-    .pet-card {
-        border-radius: 10px;
-        border: 1px solid #e0e0e0;
-        padding: 1rem;
-        margin-bottom: 1rem;
-    }
-    .pet-name {
-        font-size: 1.3rem;
-        font-weight: bold;
-        color: #ff6b6b;
-    }
-    .pet-details {
-        margin-top: 0.5rem;
-    }
-    .pet-description {
-        margin-top: 1rem;
-        font-style: italic;
-    }
-    .tag {
-        background-color: #808080;
-        border-radius: 20px;
-        padding: 0.2rem 0.6rem;
-        margin-right: 0.3rem;
-        font-size: 0.8rem;
-    }
-</style>
-""", unsafe_allow_html=True)
-
-# Initialize session state variables
-if 'access_token' not in st.session_state:
-    st.session_state.access_token = None
-if 'token_expires' not in st.session_state:
-    st.session_state.token_expires = 0
-if 'search_results' not in st.session_state:
-    st.session_state.search_results = None
-if 'selected_pet' not in st.session_state:
-    st.session_state.selected_pet = None
-if 'page' not in st.session_state:
-    st.session_state.page = 1
-if 'favorites' not in st.session_state:
-    st.session_state.favorites = []
-
-# Function to get access token
-def get_access_token():
-    # Check if token is still valid
-    if st.session_state.access_token and time.time() < st.session_state.token_expires:
-        return st.session_state.access_token
-    
-    # Get API credentials from environment variables or secrets
-    api_key = os.environ.get('PETFINDER_API_KEY') or st.secrets.get('PETFINDER_API_KEY')
-    api_secret = os.environ.get('PETFINDER_API_SECRET') or st.secrets.get('PETFINDER_API_SECRET')
-    
-    if not api_key or not api_secret:
-        st.error("⚠️ Petfinder API credentials are missing. Please set them in your environment variables or Streamlit secrets.")
-        return None
-    
-    # Get new token
-    url = "https://api.petfinder.com/v2/oauth2/token"
-    data = {
-        "grant_type": "client_credentials",
-        "client_id": api_key,
-        "client_secret": api_secret
-    }
-    
-    try:
-        response = requests.post(url, data=data)
-        response.raise_for_status()
-        token_data = response.json()
-        st.session_state.access_token = token_data['access_token']
-        st.session_state.token_expires = time.time() + token_data['expires_in'] - 60  # Buffer of 60 seconds
-        return st.session_state.access_token
-    except requests.exceptions.RequestException as e:
-        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()
-    if not token:
-        return None
-    
-    url = "https://api.petfinder.com/v2/animals"
-    headers = {"Authorization": f"Bearer {token}"}
-    
-    try:
-        response = requests.get(url, headers=headers, params=params)
-        response.raise_for_status()
-        return response.json()
-    except requests.exceptions.RequestException as e:
-        st.error(f"⚠️ Error searching pets: {str(e)}")
-        return None
-
-# Function to get breeds
-def get_breeds(animal_type):
-    token = get_access_token()
-    if not token:
-        return []
-    
-    url = f"https://api.petfinder.com/v2/types/{animal_type}/breeds"
-    headers = {"Authorization": f"Bearer {token}"}
-    
-    try:
-        response = requests.get(url, headers=headers)
-        response.raise_for_status()
-        return [breed['name'] for breed in response.json()['breeds']]
-    except requests.exceptions.RequestException as e:
-        st.error(f"⚠️ Error getting breeds: {str(e)}")
-        return []
-
-# Function to get organizations
-def get_organizations(location):
-    token = get_access_token()
-    if not token:
-        return []
-    
-    url = "https://api.petfinder.com/v2/organizations"
-    headers = {"Authorization": f"Bearer {token}"}
-    params = {"location": location, "distance": 100, "limit": 100}
-    
-    try:
-        response = requests.get(url, headers=headers, params=params)
-        response.raise_for_status()
-        return [(org['id'], org['name']) for org in response.json()['organizations']]
-    except requests.exceptions.RequestException as e:
-        st.error(f"⚠️ Error getting organizations: {str(e)}")
-        return []
-
-# Function to get pet details
-def get_pet_details(pet_id):
-    token = get_access_token()
-    if not token:
-        return None
-    
-    url = f"https://api.petfinder.com/v2/animals/{pet_id}"
-    headers = {"Authorization": f"Bearer {token}"}
-    
-    try:
-        response = requests.get(url, headers=headers)
-        response.raise_for_status()
-        return response.json()['animal']
-    except requests.exceptions.RequestException as e:
-        st.error(f"⚠️ Error getting pet details: {str(e)}")
-        return None
-
-# Function to format pet card
-def display_pet_card(pet, is_favorite=False, context="search"):
-    col1, col2 = st.columns([1, 2])
-    
-    with col1:
-        if pet['photos'] and len(pet['photos']) > 0:
-            st.image(pet['photos'][0]['medium'], use_container_width=True)
-        else:
-            st.image("https://via.placeholder.com/300x300?text=No+Image", use_container_width=True)
-    
-    with col2:
-        st.markdown(f"<div class='pet-name'>{pet['name']}</div>", unsafe_allow_html=True)
-        
-        # Tags
-        tags_html = ""
-        if pet['status'] == 'adoptable':
-            tags_html += "<span class='tag' style='background-color: #808080;'>Adoptable</span> "
-        else:
-            tags_html += f"<span class='tag' style='background-color: #808080;'>{pet['status'].title()}</span> "
-        
-        if pet['age']:
-            tags_html += f"<span class='tag'>{pet['age']}</span> "
-        if pet['gender']:
-            tags_html += f"<span class='tag'>{pet['gender']}</span> "
-        if pet['size']:
-            tags_html += f"<span class='tag'>{pet['size']}</span> "
-        
-        st.markdown(f"<div>{tags_html}</div>", unsafe_allow_html=True)
-        
-        st.markdown("<div class='pet-details'>", unsafe_allow_html=True)
-        if pet['breeds']['primary']:
-            breed_text = pet['breeds']['primary']
-            if pet['breeds']['secondary']:
-                breed_text += f" & {pet['breeds']['secondary']}"
-            if pet['breeds']['mixed']:
-                breed_text += " (Mixed)"
-            st.markdown(f"<strong>Breed:</strong> {breed_text}", unsafe_allow_html=True)
-        
-        if pet['colors']['primary'] or pet['colors']['secondary'] or pet['colors']['tertiary']:
-            colors = [c for c in [pet['colors']['primary'], pet['colors']['secondary'], pet['colors']['tertiary']] if c]
-            st.markdown(f"<strong>Colors:</strong> {', '.join(colors)}", unsafe_allow_html=True)
-        
-        if 'location' in pet and pet['contact']['address']['city'] and pet['contact']['address']['state']:
-            st.markdown(f"<strong>Location:</strong> {pet['contact']['address']['city']}, {pet['contact']['address']['state']}", unsafe_allow_html=True)
-        
-        st.markdown("</div>", unsafe_allow_html=True)
-        
-        if pet['description']:
-            st.markdown(f"<div class='pet-description'>{pet['description'][:500]}{'...' if len(pet['description']) > 500 else ''}</div>", unsafe_allow_html=True)
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("View Details", key=f"details_{context}_{pet['id']}"):
-                st.session_state.selected_pet = pet['id']
-                st.rerun()
-        with col2:
-            if not is_favorite:
-                if st.button("Add to Favorites", key=f"fav_{context}_{pet['id']}"):
-                    if pet['id'] not in [p['id'] for p in st.session_state.favorites]:
-                        st.session_state.favorites.append(pet)
-                        st.success(f"Added {pet['name']} to favorites!")
-                        st.rerun()
-            else:
-                if st.button("Remove from Favorites", key=f"unfav_{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()
-
-# Function to generate pet compatibility message
-def get_compatibility_message(pet):
-    messages = []
-    
-    # Check for kids
-    if 'children' in pet['environment'] and pet['environment']['children'] is not None:
-        if pet['environment']['children']:
-            messages.append("✅ Good with children")
-        else:
-            messages.append("❌ Not recommended for homes with children")
-    
-    # Check for dogs
-    if 'dogs' in pet['environment'] and pet['environment']['dogs'] is not None:
-        if pet['environment']['dogs']:
-            messages.append("✅ Good with dogs")
-        else:
-            messages.append("❌ Not recommended for homes with dogs")
-    
-    # Check for cats
-    if 'cats' in pet['environment'] and pet['environment']['cats'] is not None:
-        if pet['environment']['cats']:
-            messages.append("✅ Good with cats")
-        else:
-            messages.append("❌ Not recommended for homes with cats")
-    
-    # Handling care needs
-    if pet['attributes']:
-        if 'special_needs' in pet['attributes'] and pet['attributes']['special_needs']:
-            messages.append("⚠️ Has special needs")
-        
-        if 'house_trained' in pet['attributes'] and pet['attributes']['house_trained']:
-            messages.append("✅ House-trained")
-        elif 'house_trained' in pet['attributes']:
-            messages.append("❌ Not house-trained")
-        
-        if 'shots_current' in pet['attributes'] and pet['attributes']['shots_current']:
-            messages.append("✅ Vaccinations up to date")
-        
-        if 'spayed_neutered' in pet['attributes'] and pet['attributes']['spayed_neutered']:
-            messages.append("✅ Spayed/neutered")
-    
-    return messages
-
-# Function to display pet details page
-# Changes to make keys unique across different tabs
-
-# Function to display pet details page with unique tab identifier
-def display_pet_details(pet_id, context="search", tab_id="tab1"):
-    pet = get_pet_details(pet_id)
-    if not pet:
-        st.error("Unable to retrieve pet details. Please try again.")
-        return
-    
-    # Back button with unique key that includes tab identifier
-    if st.button("← Back to Search Results", key=f"back_{tab_id}_{context}_{pet_id}"):
-        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)
-    
-    # Pet photos
-    if pet['photos'] and len(pet['photos']) > 0:
-        photo_cols = st.columns(min(3, len(pet['photos'])))
-        for i, col in enumerate(photo_cols):
-            if i < len(pet['photos']):
-                col.image(pet['photos'][i]['large'], use_container_width=True)
-    else:
-        st.image("https://via.placeholder.com/500x300?text=No+Image", use_container_width=True)
-    
-    # Pet details
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        st.markdown("### Details")
-        # Fix the breed line
-        breed_text = pet['breeds']['primary']
-        if pet['breeds']['secondary']:
-            breed_text += f" & {pet['breeds']['secondary']}"
-        if pet['breeds']['mixed']:
-            breed_text += " (Mixed)"
-    
-        details = [
-            f"**Type:** {pet['type']}",
-            f"**Breed:** {breed_text}",
-            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)}")
-    
-        for detail in details:
-            st.markdown(detail)
-    
-    with col2:
-        st.markdown("### Compatibility")
-        compatibility = get_compatibility_message(pet)
-        for msg in compatibility:
-            st.markdown(msg)
-    
-    # Description
-    if pet['description']:
-        if pet['description']:
-            st.markdown("### About")
-            #st.markdown(pet['description'])
-            st.markdown(f"<div class='pet-description'>{pet['description'][:500]}{'...' if len(pet['description']) > 500 else ''}</div>", unsafe_allow_html=True)
-    
-    # Contact information
-    st.markdown("### Adoption Information")
-    
-    # Organization info
-    if pet['organization_id']:
-        st.markdown(f"**Organization:** {pet['organization_id']}")
-    
-    # Contact details
-    contact_info = []
-    if pet['contact']['email']:
-        contact_info.append(f"**Email:** {pet['contact']['email']}")
-    if pet['contact']['phone']:
-        contact_info.append(f"**Phone:** {pet['contact']['phone']}")
-    if pet['contact']['address']['city'] and pet['contact']['address']['state']:
-        contact_info.append(f"**Location:** {pet['contact']['address']['city']}, {pet['contact']['address']['state']} {pet['contact']['address']['postcode'] or ''}")
-    
-    for info in contact_info:
-        st.markdown(info)
-    
-    # URL to pet on Petfinder
-    if pet['url']:
-        st.markdown(f"[View on Petfinder]({pet['url']})")
-    
-    # Add to favorites with unique key
-    is_favorite = pet['id'] in [p['id'] for p in st.session_state.favorites]
-    if not is_favorite:
-        if st.button("Add to Favorites", key=f"add_fav_{tab_id}_{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", key=f"rem_fav_{tab_id}_{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()
-
-# Function to format pet card with unique tab identifier
-def display_pet_card(pet, is_favorite=False, context="search", tab_id="tab1"):
-    col1, col2 = st.columns([1, 2])
-    
-    with col1:
-        if pet['photos'] and len(pet['photos']) > 0:
-            st.image(pet['photos'][0]['medium'], use_container_width=True)
-        else:
-            st.image("https://via.placeholder.com/300x300?text=No+Image", use_container_width=True)
-    
-    with col2:
-        st.markdown(f"<div class='pet-name'>{pet['name']}</div>", unsafe_allow_html=True)
-        
-        # Tags
-        tags_html = ""
-        if pet['status'] == 'adoptable':
-            tags_html += "<span class='tag' style='background-color: #808080;'>Adoptable</span> "
-        else:
-            tags_html += f"<span class='tag' style='background-color: #808080;'>{pet['status'].title()}</span> "
-        
-        if pet['age']:
-            tags_html += f"<span class='tag'>{pet['age']}</span> "
-        if pet['gender']:
-            tags_html += f"<span class='tag'>{pet['gender']}</span> "
-        if pet['size']:
-            tags_html += f"<span class='tag'>{pet['size']}</span> "
-        
-        st.markdown(f"<div>{tags_html}</div>", unsafe_allow_html=True)
-        
-        st.markdown("<div class='pet-details'>", unsafe_allow_html=True)
-        if pet['breeds']['primary']:
-            breed_text = pet['breeds']['primary']
-            if pet['breeds']['secondary']:
-                breed_text += f" & {pet['breeds']['secondary']}"
-            if pet['breeds']['mixed']:
-                breed_text += " (Mixed)"
-            st.markdown(f"<strong>Breed:</strong> {breed_text}", unsafe_allow_html=True)
-        
-        if pet['colors']['primary'] or pet['colors']['secondary'] or pet['colors']['tertiary']:
-            colors = [c for c in [pet['colors']['primary'], pet['colors']['secondary'], pet['colors']['tertiary']] if c]
-            st.markdown(f"<strong>Colors:</strong> {', '.join(colors)}", unsafe_allow_html=True)
-        
-        if 'location' in pet and pet['contact']['address']['city'] and pet['contact']['address']['state']:
-            st.markdown(f"<strong>Location:</strong> {pet['contact']['address']['city']}, {pet['contact']['address']['state']}", unsafe_allow_html=True)
-        
-        st.markdown("</div>", unsafe_allow_html=True)
-        
-        if pet['description']:
-            st.markdown(f"<div class='pet-description'>{pet['description'][:300]}{'...' if len(pet['description']) > 300 else ''}</div>", unsafe_allow_html=True)
-        
-        col1, col2 = st.columns(2)
-        with col1:
-            if st.button("View Details", key=f"details_{tab_id}_{context}_{pet['id']}"):
-                st.session_state.selected_pet = pet['id']
-                st.rerun()
-        with col2:
-            if not is_favorite:
-                if st.button("Add to Favorites", key=f"fav_{tab_id}_{context}_{pet['id']}"):
-                    if pet['id'] not in [p['id'] for p in st.session_state.favorites]:
-                        st.session_state.favorites.append(pet)
-                        st.success(f"Added {pet['name']} to favorites!")
-                        st.rerun()
-            else:
-                if st.button("Remove from Favorites", key=f"unfav_{tab_id}_{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()
-
-# Main app with updated function calls
-def main():
-    # Title
-    st.markdown("<h1 class='main-header'>🐾 PetMatch</h1>", unsafe_allow_html=True)
-    st.markdown("<p class='sub-header'>Find your perfect pet companion</p>", unsafe_allow_html=True)
-    
-    # Create tabs
-    tab1, tab2, tab3 = st.tabs(["Search", "Favorites", "About"])
-    
-    with tab1:
-        # If a pet is selected, show details
-        if st.session_state.selected_pet:
-            display_pet_details(st.session_state.selected_pet, context="search", tab_id="tab1")
-        else:
-            # Search form
-            with st.expander("Search Options", expanded=True):
-                with st.form("pet_search_form"):
-                    col1, col2 = st.columns(2)
-                    
-                    with col1:
-                        animal_type = st.selectbox(
-                            "Animal Type",
-                            ["Dog", "Cat", "Rabbit", "Small & Furry", "Horse", "Bird", "Scales, Fins & Other", "Barnyard"]
-                        )
-                        
-                        location = st.text_input("Location (ZIP code or City, State)", "")
-                        
-                        distance = st.slider("Distance (miles)", min_value=10, max_value=500, value=50, step=10)
-                    
-                    with col2:
-                        age_options = ["", "Baby", "Young", "Adult", "Senior"]
-                        age = st.selectbox("Age", age_options)
-                        
-                        size_options = ["", "Small", "Medium", "Large", "XLarge"]
-                        size = st.selectbox("Size", size_options)
-                        
-                        gender_options = ["", "Male", "Female"]
-                        gender = st.selectbox("Gender", gender_options)
-                        good_with_children = st.checkbox("Good with children")
-                        good_with_dogs = st.checkbox("Good with dogs")
-                        good_with_cats = st.checkbox("Good with cats")
-                        house_trained = st.checkbox("House-trained")
-                        special_needs = st.checkbox("Special needs")
-        
-                    submitted = st.form_submit_button("Search")
-                    
-                    if submitted:
-                        # Build search parameters
-                        params = {
-                            "type": animal_type.split(" ")[0],  # Take first word for types like "Small & Furry"
-                            "location": location,
-                            "distance": distance,
-                            "status": "adoptable",
-                            "sort": "distance",
-                            "limit": 100
-                        }
-                        
-                        if age and age != "":
-                            params["age"] = age
-                        if size and size != "":
-                            params["size"] = size
-                        if gender and gender != "":
-                            params["gender"] = gender
-                        
-                        # Add advanced filters
-                        if good_with_children:
-                            params["good_with_children"] = 1
-                        if good_with_dogs:
-                            params["good_with_dogs"] = 1
-                        if good_with_cats:
-                            params["good_with_cats"] = 1
-                        if house_trained:
-                            params["house_trained"] = 1
-                        if special_needs:
-                            params["special_needs"] = 1
-                        
-                        # Perform search
-                        results = search_pets(params)
-                        if results and 'animals' in results:
-                            st.session_state.search_results = results
-                            st.session_state.page = 1
-                            st.success(f"Found {len(results['animals'])} pets!")
-                        else:
-                            st.error("No pets found with those criteria. Try expanding your search.")
-            
-            # Display search results
-            if st.session_state.search_results and 'animals' in st.session_state.search_results:
-                st.markdown("### Search Results")
-                
-                # Pagination
-                results = st.session_state.search_results['animals']
-                total_pages = (len(results) + 9) // 10  # 10 items per page
-                
-                # Display page selector
-                if total_pages > 1:
-                    col1, col2, col3 = st.columns([1, 3, 1])
-                    with col2:
-                        page = st.slider("Page", 1, total_pages, st.session_state.page)
-                        if page != st.session_state.page:
-                            st.session_state.page = page
-                
-                # Display pets for current page
-                start_idx = (st.session_state.page - 1) * 10
-                end_idx = min(start_idx + 10, len(results))
-                
-                for pet in results[start_idx:end_idx]:
-                    st.markdown("---")
-                    display_pet_card(pet, tab_id="tab1")
-    
-    with tab2:
-        st.markdown("### Your Favorite Pets")
-        
-        if not st.session_state.favorites:
-            st.info("You haven't added any pets to your favorites yet. Start searching to find your perfect match!")
-        else:
-            # Check if a pet is selected from favorites
-            if st.session_state.selected_pet:
-                display_pet_details(st.session_state.selected_pet, context="favorites", tab_id="tab2")
-            else:
-                for pet in st.session_state.favorites:
-                    st.markdown("---")
-                    display_pet_card(pet, is_favorite=True, context="favorites", tab_id="tab2")
-    
-    with tab3:
-        st.markdown("### About PetMatch")
-        st.markdown("""
-        PetMatch helps you find your perfect pet companion from thousands of adoptable animals across the country.
-        
-        **How to use PetMatch:**
-        1. Search for pets based on your preferences and location
-        2. Browse through the results and click "View Details" to learn more about each pet
-        3. Add pets to your favorites to keep track of the ones you're interested in
-        4. Contact the shelter or rescue organization directly using the provided information
-        
-        **Data Source:**
-        PetMatch uses the Petfinder API to provide up-to-date information on adoptable pets. Petfinder is North America's largest adoption website with hundreds of thousands of adoptable pets listed by more than 11,500 animal shelters and rescue organizations.
-        
-        **Privacy:**
-        PetMatch does not store any personal information or search history. Your favorites are stored locally in your browser and are not shared with any third parties.
-        """)
-
-
-if __name__ == "__main__":
-    main()