import os
import sys
from antigravity_sdk.client import RemoteGPU

SCRIPT_PATH = "examples/deep_nanogpt_resumable.py"

def main():
    if not os.path.exists(SCRIPT_PATH):
        print(f"❌ Script not found: {SCRIPT_PATH}")
        sys.exit(1)

    with open(SCRIPT_PATH, 'r') as f:
        code = f.read()

    print(f"🚀 DEBUG RUN: Deep-NanoGPT Phase 2...")
    gpu = RemoteGPU()
    
    # Run once, blocking, verbose
    result = gpu.run(code, download_files=True, verbose=True)
    
    print("\n---------- RAW OUTPUT START ----------")
    print(result.output)
    print("----------- RAW OUTPUT END -----------")

if __name__ == "__main__":
    main()