add color picker for waveform and bg

app.py

@@ -2,24 +2,30 @@ import gradio as gr
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.animation import FuncAnimation
-import matplotlib.colors as mcolors
-import io
 import librosa
 import tempfile
 
-def extract_waveform_animation(audio_file, window_seconds=5):
+def rgba_to_hex(r, g, b, a):
+    return f'#{r:02x}{g:02x}{b:02x}{a:02x}'
+
+def check_rgba(string:str):
+    if '#' in string:
+        return string
+    else:
+        clean_str_list = string.split("(")[1].split(")")[0].split(",")
+        hex = rgba_to_hex(clean_str_list[0], clean_str_list[1], clean_str_list[2], clean_str_list[3])
+        return hex
+
+def extract_waveform_animation(audio_file, window_seconds, waveform_color, background_color):
     y, sr = librosa.load(audio_file, sr=None)
     duration = librosa.get_duration(y=y, sr=sr)
     FPS = 1
     fig, ax = plt.subplots()
-    line, = ax.plot([], [], lw=2, color="#0021e4")
+    line, = ax.plot([], [], lw=2, color=waveform_color)
     window_length = int(window_seconds * sr)
-    
-    # Initialize with first window
-    first_window = y[:window_length]
     x_vals = np.linspace(0, duration, num=len(y))
     ax.set_axis_off()
-    bg_color ="#00FFFF00"
+    bg_color = background_color
     fig.set_facecolor(bg_color)
     
     def init():
@@ -55,7 +61,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(label="Waveform color",  value="#0021e4"),
+        gr.ColorPicker(label="Background color",  value="#00FFFF00")
     ],
     outputs=gr.Image(),
     description="Scroll through audio waveform with a moving window."