import requests
import json

def test_bee_search():
    base_url = 'http://localhost:3015'
    headers = {'X-API-Key': 'jleu1212', 'Content-Type': 'application/json'}
    
    # Test search for bee-related content
    queries = ['bee', 'photo of a bee', 'insect', 'image classification']
    
    for query in queries:
        print(f"\n🔍 Testing search for: '{query}'")
        try:
            response = requests.post(f'{base_url}/api/search', headers=headers, json={'query': query, 'top_k': 10}, timeout=20)
            print(f'Status: {response.status_code}')
            
            if response.status_code == 200:
                results = response.json()
                chunks = results.get('chunks', [])
                entities = results.get('entities', [])
                
                print(f'Chunks found: {len(chunks)}')
                print(f'Entities found: {len(entities)}')
                
                if entities:
                    print('Entities:')
                    for entity in entities:
                        print(f'  - {entity.get("entity_name")}: {entity.get("entity_type")}')
                
                if chunks:
                    print('First chunk content preview:')
                    if chunks[0].get('content'):
                        print(f'  {chunks[0]["content"][:200]}...')
            else:
                print(f'Error: {response.text}')
                
        except Exception as e:
            print(f'Request failed: {e}')

if __name__ == '__main__':
    test_bee_search()