import jwt
from datetime import datetime, timedelta

ak = ""  # Your Access Key
sk = ""  # Your Secret Key

def generate_token(ak: str, sk: str) -> str:
    now = datetime.now()

    headers = {
        'alg': 'HS256',  # Signing algorithm: HMAC-SHA256
        'typ': 'JWT'     # Token type
    }

    payload = {
        'iss': ak,                                              # Issuer
        'nbf': int((now - timedelta(seconds=5)).timestamp()),   # Not valid before (5s grace)
        'exp': int((now + timedelta(days=10)).timestamp())      # Expires in 10 days
    }

    secret_key = sk.encode('utf-8')
    token = jwt.encode(payload, secret_key, headers=headers)
    return token

api_token = generate_token(ak, sk)
print(api_token)

import requests

def create_video_task(payload: dict, api_token: str) -> dict:
    headers = {
        "Authorization": f"Bearer {api_token}",
        "Content-Type": "application/json"
    }

    response = requests.post(
        "https://api-gen-na.bach.art/api/vdr/submit",
        headers=headers,
        json=payload
    )

    if response.status_code  200:
        return response.json()
    elif response.status_code  400:
        error_data = response.json()
        raise ValueError(f"Parameter error: {error_data.get('message')}")
    elif response.status_code  401:
        raise PermissionError("Authentication failed. Check your API token.")
    elif response.status_code  429:
        raise ResourceWarning("Rate limit exceeded. Please retry later.")
    else:
        raise Exception(f"Unexpected error: {response.status_code}")

import time

def wait_for_completion(task_id: str, api_token: str, max_attempts: int = 60) -> dict:
    headers = {
        "Authorization": f"Bearer {api_token}",
        "Content-Type": "application/json"
    }

    for _ in range(max_attempts):
        response = requests.get(
            f"https://api-gen-na.bach.art/api/vdr/status/{task_id}",
            headers=headers
        )

        result = response.json()
        status = result.get("status")

        if status  "TASK_SUCCEEDED":
            return result
        elif status  "TASK_FAILED":
            raise Exception("Video generation failed")

        time.sleep(10)  # Poll every 10 seconds

    raise TimeoutError("Task did not complete within expected time")