add waveform and bg_color params

app.py

@@ -6,7 +6,7 @@ import io
 import librosa
 import tempfile
 
-def extract_waveform_animation(audio_file, window_seconds=5):
+def extract_waveform_animation(audio_file, window_seconds=5, waveform_color="blue", bg_color="#00FFFF00"):
     y, sr = librosa.load(audio_file, sr=None)
     duration = librosa.get_duration(y=y, sr=sr)
     FPS = 1
@@ -18,7 +18,8 @@ def extract_waveform_animation(audio_file, window_seconds=5):
     first_window = y[:window_length]
     x_vals = np.linspace(0, duration, num=len(y))
     ax.set_axis_off()
-    fig.set_facecolor("#00FFFF00")
+    ax.set_facecolor(waveform_color)
+    fig.set_facecolor(bg_color)
     
     def init():
         ax.set_xlim(0, window_seconds)
@@ -53,7 +54,9 @@ iface = gr.Interface(
     fn=extract_waveform_animation,
     inputs=[
         gr.Audio(type="filepath"),
-        gr.Slider(1, 10, value=5, step=1, label="Window Size (seconds)")
+        gr.Slider(1, 10, value=5, step=1, label="Window Size (seconds)"),
+        gr.ColorPicker(value="blue", label="Waveform Color"),
+        gr.ColorPicker(value="#00FFFF00", label="Background Color")
     ],
     outputs=gr.Image(),
     description="Scroll through audio waveform with a moving window."