from antigravity_sdk import RemoteGPU
import os

print("Testing antigravity library...")
# Nota: L'URL default è già corretto nella classe, ma lo esplicito per chiarezza
client = RemoteGPU("https://jangai-antigravity.hf.space")

code = """
import os
import sys
print(f"Hello from Antigravity Lib! 🚀")
print(f"Server Host: {os.uname().nodename}")
print(f"Python: {sys.version.split()[0]}")
"""

print("Sending code...")
result = client.run(code)

if "Hello from Antigravity Lib!" in result.output:
    print("✅ TEST PASSED")
else:
    print("❌ TEST FAILED")
    print("Output:", result.output)