import requests
import json

BASE = "https://xrplme.online/wc/v1"
COUNTRY = "thailand"   # or any of the 24 countries
ENDPOINT = "bot"      # or "bot/ai" for AI-enhanced (0.5 XRP)
QUERY = "What are Thailand's business registration requirements?"

def query_xrplme(query, country=COUNTRY, endpoint=ENDPOINT):
    url = f"{BASE}/{country}/{endpoint}"
    payload = {
        "message": query,
        "session_id": "agent-session-001"
    }

    resp = requests.post(url, json=payload, timeout=15)

    if resp.status_code == 402:
        data = resp.json()
        payment = data["accepts"][0]
        print(f"Payment required: {payment['amount']} drops")
        print(f"Pay to: {payment['payTo']}")
        print(f"Invoice ID: {payment['extra']['invoiceId']}")
        print("Pay via Xaman/Joey wallet, then call this function again.")
        return

    if resp.status_code == 200:
        data = resp.json()
        print(f"Response: {json.dumps(data, indent=2)}")
        return data

    print(f"Error {resp.status_code}: {resp.text}")
    return None

# Example
query_xrplme("What are the VAT rates in Thailand?")

async function queryXRPLME(query, country = "thailand", endpoint = "bot") {
  const url = `https://xrplme.online/wc/v1/${country}/${endpoint}`;
  const response = await fetch(url, {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify({
      message: query,
      session_id: "web-session-001"
    })
  });

  if (response.status === 402) {
    const data = await response.json();
    const payment = data.accepts[0];
    console.log("Payment required:", payment.amount / 1000000, "XRP");
    console.log("Pay to:", payment.payTo);
    console.log("Invoice ID:", payment.extra.invoiceId);
    // Redirect user to wallet for payment, then retry
    return;
  }

  if (response.status === 200) {
    const data = await response.json();
    console.log("Data:", data);
    return data;
  }
}

// Example
queryXRPLME("Compare Thailand vs Vietnam for tech startup");

# Step 1: Send query (get 402 with payment details)
curl -X POST https://xrplme.online/wc/v1/thailand/bot \
  -H "Content-Type: application/json" \
  -d '{"message":"What are Thailand business registration requirements?","session_id":"agent-001"}'

# Step 2: Pay the XRP invoice (via Xaman/Joey wallet)
# Step 3: Resend the same query to get data
curl -X POST https://xrplme.online/wc/v1/thailand/bot \
  -H "Content-Type: application/json" \
  -d '{"message":"What are Thailand business registration requirements?","session_id":"agent-001"}'

curl https://xrplme.online/wc/v1/thailand/health