import base64
from openai import OpenAI

client = OpenAI(
    api_key="YOUR_API_KEY",
    base_url="https://crazyrouter.com/v1"
)

# 读取音频文件并编码为 Base64
with open("question.wav", "rb") as f:
    audio_base64 = base64.b64encode(f.read()).decode("utf-8")

response = client.chat.completions.create(
    model="gpt-4o-audio-preview",
    messages=[
        {
            "role": "user",
            "content": [
                {"type": "text", "text": "请听这段音频并回答问题"},
                {
                    "type": "input_audio",
                    "input_audio": {
                        "data": audio_base64,
                        "format": "wav"
                    }
                }
            ]
        }
    ]
)

print(response.choices[0].message.content)

response = client.chat.completions.create(
    model="gpt-4o-audio-preview",
    modalities=["text", "audio"],
    audio={"voice": "nova", "format": "wav"},
    messages=[
        {"role": "user", "content": "用中文讲一个简短的笑话"}
    ]
)

# 获取文本回复
print(response.choices[0].message.content)

# 获取音频回复
if response.choices[0].message.audio:
    audio_data = base64.b64decode(response.choices[0].message.audio.data)
    with open("response.wav", "wb") as f:
        f.write(audio_data)