"""
Busca de Clusters Multi-GPU e GPUs Flagship no Vast.ai.
"""

import subprocess
import json

def search_clusters():
    cmd = [
        '/home/j/.local/bin/vastai',
        'search',
        'offers',
        'reliability > 0.95 inet_down > 150 (num_gpus >= 2 or gpu_name = H100 or gpu_name = H100_SXM or gpu_name = H200)',
        '-o',
        'dph',
        '--raw'
    ]
    res = subprocess.run(cmd, capture_output=True, text=True)
    if res.returncode != 0:
        print("Erro:", res.stderr)
        return

    offers = json.loads(res.stdout)
    categories = {}

    for o in offers:
        gpu_name = o.get('gpu_name', 'Unknown')
        num_gpus = o.get('num_gpus', 1)
        dph = o.get('dph_total', 999.0)
        total_vram = (o.get('gpu_ram', 0)) / 1024.0
        dlperf = o.get('dlperf', 0)
        inet_down = o.get('inet_down', 0)
        reliability = (o.get('reliability2') or o.get('reliability') or 0.98) * 100
        offer_id = o.get('id')

        key = f"{num_gpus}x {gpu_name}" if num_gpus > 1 else gpu_name

        if key not in categories or dph < categories[key]['dph']:
            categories[key] = {
                'id': offer_id,
                'name': key,
                'num_gpus': num_gpus,
                'total_vram_gb': round(total_vram, 1),
                'dph': round(dph, 3),
                'dlperf': round(dlperf, 1),
                'inet_down_mbps': round(inet_down, 0),
                'reliability_pct': round(reliability, 1)
            }

    print(json.dumps(categories, indent=2))

if __name__ == "__main__":
    search_clusters()