import requests, os
import matplotlib.pyplot as plt
from datetime import datetime
BASE_URL = os.environ["POLYSIM_BASE_URL"]
headers = {"X-API-Key": os.environ["POLYSIM_API_KEY"]}
equity = requests.get(
f"{BASE_URL}/v1/account/equity",
headers=headers,
params={"days": 30}, # 30 days
).json()
times = [datetime.fromisoformat(e["timestamp"]) for e in equity]
values = [float(e["total_value"]) for e in equity]
plt.figure(figsize=(12, 6))
plt.plot(times, values)
# Starting balance = your API-wallet baseline, NOT the $1,000 UI MAIN
# wallet: Pro = 10_000, Pro+ = 25_000 (Free keys are read-only). Set it
# to your tier's baseline so the reference line is meaningful.
API_WALLET_BASELINE = 10_000 # Pro; use 25_000 on Pro+
plt.axhline(y=API_WALLET_BASELINE, color="gray", linestyle="--", label="Starting balance")
plt.title("Portfolio Equity Curve")
plt.ylabel("Portfolio Value ($)")
plt.legend()
plt.show()