我如何用pyaudio制作我的音频循环？

import pyaudio
import wave

def play_audio(file_path, loop=False):
    chunk = 1024
    wf = wave.open(file_path, 'rb')
    p = pyaudio.PyAudio()

    stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
                    channels=wf.getnchannels(),
                    rate=wf.getframerate(),
                    output=True)

    data = wf.readframes(chunk)
    while data != b'':
        stream.write(data)
        data = wf.readframes(chunk)

    if not loop:
        stream.stop_stream()
        stream.close()
        p.terminate()

# 播放音频文件，不循环
play_audio('audio.wav')

# 播放音频文件，循环
play_audio('audio.wav', loop=True)