#!/bin/bash
# setup_free_gaia.sh - Setup script for Free GAIA Agent

echo "🆓 Free GAIA Agent Setup"
echo "========================"
echo "No API keys required!"

# Check Python version
python_version=$(python3 --version 2>&1)
echo "🐍 Python version: $python_version"

# Create virtual environment (optional but recommended)
echo "📦 Creating virtual environment..."
python3 -m venv free_gaia_env
source free_gaia_env/bin/activate

# Install minimal requirements
echo "📥 Installing requirements..."
pip install --upgrade pip
pip install datasets pandas requests wikipedia

# Test internet connection
echo "🌐 Testing internet connection..."
python3 -c "
import requests
try:
    response = requests.get('https://api.duckduckgo.com/', timeout=5)
    print('✅ Internet connection working')
except:
    print('❌ Internet connection failed')
    print('⚠️  This agent requires internet access')
"

# Test dataset access
echo "🧪 Testing GAIA dataset access..."
python3 -c "
from datasets import load_dataset
try:
    dataset = load_dataset('gaia-benchmark/GAIA', '2023_all')
    print('✅ GAIA dataset access successful!')
    print(f'   Validation set: {len(dataset[\"validation\"])} questions')
    print(f'   Test set: {len(dataset[\"test\"])} questions')
except Exception as e:
    print(f'❌ Dataset access failed: {e}')
    print('💡 You may need to request access at:')
    print('   https://huggingface.co/datasets/gaia-benchmark/GAIA')
"

echo ""
echo "🎉 Setup complete!"
echo ""
echo "📋 Next steps:"
echo "1. Run: python3 free_gaia_agent.py"
echo "2. Choose test mode first (5 questions)"
echo "3. If successful, try larger batches"
echo "4. Submit generated .jsonl file to GAIA leaderboard"
echo ""
echo "🔗 GAIA Leaderboard: https://huggingface.co/spaces/gaia-benchmark/leaderboard"