import requests, json, os
from datetime import datetime, date

API_KEY = "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"

def fetch_costs():
    headers = {"Authorization": f"Bearer {API_KEY}"}
    
    # Get key info - has daily usage built-in!
    key_info = requests.get("https://openrouter.ai/api/v1/auth/key", headers=headers).json()
    
    today = date.today().isoformat()
    
    # Use usage_daily from key info for today's cost
    today_cost = key_info.get("data", {}).get("usage_daily", 0)
    
    output = {
        "updated_at": datetime.now().isoformat(),
        "daily_limit": 5.00,
        "monthly_limit": 100.00,
        "today": {
            "total": round(today_cost, 6),
            "by_model": {}
        },
        "credit_balance": key_info.get("data", {}).get("limit_remaining", 0),
        "credit_used": key_info.get("data", {}).get("usage", 0)
    }
    
    with open("/data/reports/daily/costs.json", "w") as f:
        json.dump(output, f, indent=2)
    
    print(f"Updated: ${today_cost:.4f} spent today")
    print(f"Total used: ${key_info.get('data', {}).get('usage', 0):.4f}")
    print(f"Balance: ${key_info.get('data', {}).get('limit_remaining', 0):.4f}")

fetch_costs()