Update visualization.py

visualization.py

@@ -15,6 +15,8 @@ from matplotlib.patches import Rectangle
 from utils import seconds_to_timecode
 from anomaly_detection import determine_anomalies
 from scipy import interpolate
+import librosa
+import librosa.display
 import gradio as gr
 import os
 
@@ -256,3 +258,24 @@ def plot_stacked_mse_heatmaps(mse_face, mse_posture, mse_voice, df, title="Combi
     plt.tight_layout()
     plt.close()
     return fig
+
+def plot_audio_waveform(audio_path, title="Audio Waveform"):
+    # Load the audio file
+    y, sr = librosa.load(audio_path)
+    
+    # Create the plot
+    plt.figure(figsize=(20, 4))
+    librosa.display.waveshow(y, sr=sr)
+    
+    # Set the x-axis to display timecodes
+    max_time = librosa.get_duration(y=y, sr=sr)
+    x_ticks = np.arange(0, max_time, max_time/10)  # 10 ticks
+    x_labels = [f"{int(t//3600):02d}:{int((t%3600)//60):02d}:{int(t%60):02d}" for t in x_ticks]
+    plt.xticks(x_ticks, x_labels, rotation=45)
+    
+    plt.title(title)
+    plt.xlabel("Time")
+    plt.ylabel("Amplitude")
+    plt.tight_layout()
+    
+    return plt.gcf()