Transcrição de voz em tempo real

# Dependência: pip install "openai[realtime]"
import asyncio
import base64
from openai import AsyncOpenAI

# Reutilizar o SDK oficial requer apenas três adaptações:
#   1) websocket_base_url apontando para o gateway da AiHubMix (em vez do endereço padrão da OpenAI)
#   2) connect() passando model explicitamente: veja a Note acima, obrigatório no handshake do gateway
#   3) extra_query trazendo intent=transcription
client = AsyncOpenAI(
    api_key="sk-***",  # substitua pela sua chave de API da AiHubMix
    websocket_base_url="wss://aihubmix.com/v1",
)

async def main():
    async with client.realtime.connect(
        model="gpt-live-transcribe",           # obrigatório: entra na URL do handshake, o gateway usa para roteamento + cobrança
        extra_query={"intent": "transcription"},
    ) as conn:
        # 1) Configurar a sessão de transcrição
        await conn.session.update(session={
            "type": "transcription",
            "audio": {"input": {
                "format": {"type": "audio/pcm", "rate": 24000},
                "transcription": {
                    "model": "gpt-live-transcribe",
                    "languages": ["en", "zh"],
                    "prompt": "会议录音,含产品名与英文缩写",
                    "keywords": ["AiHubMix", "gpt-live-transcribe"],
                },
                "turn_detection": None,
                "noise_reduction": {"type": "near_field"},
            }},
        })

        # 2) Ler o áudio local PCM16 / 24kHz / mono cru e enviar em fragmentos
        async def send_audio():
            with open("audio_pcm16_24k.raw", "rb") as f:
                pcm = f.read()
            chunk = 24000 * 2 // 10  # 100ms = taxa de amostragem × 2 bytes ÷ 10
            for i in range(0, len(pcm), chunk):
                await conn.input_audio_buffer.append(
                    audio=base64.b64encode(pcm[i:i + chunk]).decode()
                )
                await asyncio.sleep(0.1)  # simula o ritmo em tempo real
            # Sem VAD, faça commit manual após enviar todo o áudio para acionar a finalização da transcrição
            await conn.input_audio_buffer.commit()

        asyncio.create_task(send_audio())

        # 3) Receber os resultados da transcrição
        async for evt in conn:
            etype = getattr(evt, "type", "")
            if etype.endswith("transcription.delta"):
                print(getattr(evt, "delta", ""), end="", flush=True)
            elif etype.endswith("transcription.completed"):
                print("\n[concluído]", getattr(evt, "transcript", ""))
                break  # pode sair ao obter o resultado completo
            elif etype == "error":
                print("\n[erro]", evt.to_dict())
                break

asyncio.run(main())

import asyncio
import base64
import json
import websockets

API_KEY = "sk-***"  # substitua pela sua chave de API da AiHubMix
URL = (
    "wss://aihubmix.com/v1/realtime"
    "?intent=transcription&model=gpt-live-transcribe"
)

async def main():
    # websockets >= 13 usa additional_headers; versões antigas usam extra_headers
    async with websockets.connect(
        URL, additional_headers={"Authorization": f"Bearer {API_KEY}"}
    ) as ws:
        # 1) Configurar a sessão de transcrição
        await ws.send(json.dumps({
            "type": "session.update",
            "session": {
                "type": "transcription",
                "audio": {"input": {
                    "format": {"type": "audio/pcm", "rate": 24000},
                    "transcription": {"model": "gpt-live-transcribe", "language": "en"},
                    "turn_detection": None,
                    "noise_reduction": {"type": "near_field"},
                }},
            },
        }))

        # 2) Ler o áudio local PCM16 / 24kHz / mono cru e enviar em fragmentos
        async def send_audio():
            with open("audio_pcm16_24k.raw", "rb") as f:
                pcm = f.read()
            chunk = 24000 * 2 // 10  # 100ms = taxa de amostragem × 2 bytes ÷ 10
            for i in range(0, len(pcm), chunk):
                await ws.send(json.dumps({
                    "type": "input_audio_buffer.append",
                    "audio": base64.b64encode(pcm[i:i + chunk]).decode(),
                }))
                await asyncio.sleep(0.1)  # simula o ritmo em tempo real
            # Sem VAD, faça commit manual após enviar todo o áudio para acionar a finalização da transcrição
            await ws.send(json.dumps({"type": "input_audio_buffer.commit"}))

        asyncio.create_task(send_audio())

        # 3) Receber os resultados da transcrição
        async for msg in ws:
            evt = json.loads(msg)
            etype = evt.get("type", "")
            if etype.endswith("transcription.delta"):
                print(evt.get("delta", ""), end="", flush=True)
            elif etype.endswith("transcription.completed"):
                print("\n[concluído]", evt.get("transcript", ""))
                break  # pode sair ao obter o resultado completo
            elif etype == "error":
                print("\n[erro]", evt.get("error"))
                break

asyncio.run(main())

# Use wscat para validar rapidamente a conectividade e a autenticação (instale antes com npm i -g wscat)
wscat -c "wss://aihubmix.com/v1/realtime?intent=transcription&model=gpt-live-transcribe" \
  -H "Authorization: Bearer $AIHUBMIX_API_KEY"

# Após conectar, cole um frame session.update para começar (o áudio precisa ser codificado em base64 e enviado com input_audio_buffer.append)