用Python整一个简单的音乐球

import os
try:
  import pygame
  import pyaudio
  import numpy as np

except:
  os.system("pip install pygame")
  os.system("pip install pyaudio")
  os.system("pip install numpy")


# 初始化Pygame和PyAudio
pygame.init()
pa = pyaudio.PyAudio()

# 设置窗口大小
WINDOW_SIZE = (800, 600)

# 创建两个Pygame Surface：一个用于绘制3D球体，一个用于绘制背景
surf_ball = pygame.Surface(WINDOW_SIZE, pygame.SRCALPHA)
surf_bg = pygame.Surface(WINDOW_SIZE)

# 定义3D球体的参数
BALL_COLOR = (255, 255, 255)
BALL_CENTER = (400, 300)
BALL_RADIUS = 150
BALL_FREQ_RANGE = (20, 2000)  # 只考虑这个频率范围内的声音
BALL_MAX_SIZE = 2 * BALL_RADIUS

# 定义背景的参数
BG_COLOR = (0, 0, 0)
BG_RADIUS = 100
BG_FREQ_RANGE = (40, 400)  # 只考虑这个频率范围内的声音
BG_MAX_SIZE = 2 * BG_RADIUS

# 定义声音处理的参数
SAMPLE_RATE = 44100
BUFFER_SIZE = 1024
FREQ_BINS = BUFFER_SIZE // 2
SPECTRUM_MIN = -60
SPECTRUM_MAX = 0
SPECTRUM_RANGE = SPECTRUM_MAX - SPECTRUM_