varco_client.py

"""VARCO Sound API client module."""

import base64
import requests
from pathlib import Path

BASE_URL = "https://openapi.ai.nc.com/sound/varco/v1/api"
TIMEOUT = 120


def _headers(api_key: str) -> dict:
    return {"OPENAPI_KEY": api_key.strip(), "Content-Type": "application/json"}


def _audio_to_base64(file_bytes: bytes) -> str:
    return base64.b64encode(file_bytes).decode("utf-8")


def _decode_audio(b64: str) -> bytes:
    return base64.b64decode(b64)


def _post(api_key: str, endpoint: str, payload: dict) -> requests.Response:
    url = f"{BASE_URL}/{endpoint}"
    resp = requests.post(url, json=payload, headers=_headers(api_key), timeout=TIMEOUT)
    resp.raise_for_status()
    return resp


# ── Text to Sound ──────────────────────────────────────────────
def text_to_sound(
    api_key: str,
    prompt: str,
    num_sample: int = 1,
    version: str = "v2",
) -> list[bytes]:
    payload = {"prompt": prompt, "num_sample": num_sample, "version": version}
    data = _post(api_key, "text2sound", payload).json()
    return [_decode_audio(item["audio"]) for item in data]


# ── Variation ──────────────────────────────────────────────────
def variation(
    api_key: str,
    source: bytes,
    num_sample: int = 1,
    strength: float = 1.0,
    begin: float | None = None,
    end: float | None = None,
) -> list[bytes]:
    payload: dict = {
        "source": _audio_to_base64(source),
        "num_sample": num_sample,
        "strength": strength,
    }
    if begin is not None and end is not None:
        payload["include"] = {"begin": begin, "end": end}
    data = _post(api_key, "variation", payload).json()
    return [_decode_audio(item["audio"]) for item in data]


# ── Mono to Stereo ─────────────────────────────────────────────
def mono_to_stereo(api_key: str, source: bytes) -> bytes:
    payload = {"source": _audio_to_base64(source)}
    data = _post(api_key, "mono2stereo", payload).json()
    return _decode_audio(data["audio"])


# ── Looping ────────────────────────────────────────────────────
def looping(
    api_key: str,
    source: bytes,
    begin: float | None = None,
    end: float | None = None,
) -> bytes:
    payload: dict = {"source": _audio_to_base64(source)}
    if begin is not None and end is not None:
        payload["preserve"] = {"begin": begin, "end": end}
    data = _post(api_key, "looping", payload).json()
    return _decode_audio(data["audio"])


# ── Conversion ─────────────────────────────────────────────────
def conversion(
    api_key: str,
    source: bytes,
    reference: bytes,
    ratio: float = 1.0,
    enhance: bool = False,
) -> bytes:
    payload = {
        "source": _audio_to_base64(source),
        "reference": _audio_to_base64(reference),
        "ratio": ratio,
        "enhance": enhance,
    }
    data = _post(api_key, "conversion", payload).json()
    return _decode_audio(data["audio"])


# ── Enhance ────────────────────────────────────────────────────
def enhance(api_key: str, source: bytes) -> bytes:
    payload = {"source": _audio_to_base64(source)}
    data = _post(api_key, "enhance", payload).json()
    return _decode_audio(data["audio"])