import requests
import os
import base64
from typing import Optional

from . import APIProvider, register

MIME_MAP = {".wav": "audio/wav", ".mp3": "audio/mpeg", ".m4a": "audio/mp4", ".flac": "audio/flac"}


@register("zoom")
class ZoomProvider(APIProvider):
    ENDPOINT = "https://api.zoom.us/v2/aiservices/scribe/transcribe"

    def transcribe(
        self,
        model_variant: str,
        audio_file_path: Optional[str],
        sample: dict,
        use_url: bool = False,
        language: str = "en",
    ) -> str:
        api_key = os.getenv("ZOOM_API_KEY")
        if not api_key or api_key == "your_api_key":
            raise ValueError("ZOOM_API_KEY environment variable not set")

        if use_url:
            file_payload = sample["row"]["audio"][0]["src"]
            audio_duration = sample["row"]["audio_length_s"]
        else:
            audio_duration = (
                len(sample["audio"]["array"]) / sample["audio"]["sampling_rate"]
            )
            with open(audio_file_path, "rb") as f:
                audio_bytes = f.read()
            mime = MIME_MAP.get(os.path.splitext(audio_file_path)[1].lower(), "audio/wav")
            file_payload = f"data:{mime};base64,{base64.b64encode(audio_bytes).decode('ascii')}"

        if audio_duration <= 29.9:
            segmentation_mode = "none"
        else:
            segmentation_mode = "auto"

        resp = requests.post(
            self.ENDPOINT,
            headers={"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"},
            json={"file": file_payload,
                  "config": {
                    "language": "en-US",
                    "segmentation_mode": segmentation_mode,
                    "experimental_feature": {"model_pro": True},
                    "timestamps": True,
                  }},
            timeout=300,
        )
        resp.raise_for_status()
        return resp.json().get("result", {}).get("text_display", "") or "."