find frequencies from audio file in python

main.py
import numpy as np
import wave

# Open the WAV file
with wave.open('audio_file.wav', 'rb') as wav_file:
    # Get the number of frames, channels and sample rate
    num_frames = wav_file.getnframes()
    num_channels = wav_file.getnchannels()
    sample_rate = wav_file.getframerate()
    
    # Read the audio frames
    wav_frames = wav_file.readframes(num_frames)
    wav_frames = np.frombuffer(wav_frames, dtype=np.int16)
    
    # Apply FFT to each channel
    fft_frames = []
    for channel in range(num_channels):
        # Get the channel data
        channel_data = wav_frames[channel::num_channels]
        # Apply FFT on the channel data
        fft_data = np.fft.rfft(channel_data)
        fft_frames.append(fft_data.real)
    
    # Get the frequency axis
    freq_axis = np.fft.rfftfreq(num_frames, d=1/sample_rate)
    
# The FFT frames contain the frequency spectra for each channel
print(fft_frames)
919 chars
29 lines