Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import collections
 import scipy.signal
 import numpy as np
+import matplotlib.pyplot as plt
 from functools import partial
 from openwakeword.model import Model
 
@@ -43,28 +44,32 @@ def process_audio(audio, state=collections.defaultdict(partial(collections.deque
             # Average last few frames for smoother display
             scores.append(np.mean(list(state[key])))
     
-    # Create dataframe for bar plot
-    df = pd.DataFrame({
-        "Model": model_names,
-        "Score": scores
-    })
-    
     # Sort by score for better visibility
-    df = df.sort_values('Score', ascending=False)
+    sorted_indices = np.argsort(scores)[::-1]
+    model_names = [model_names[i] for i in sorted_indices]
+    scores = [scores[i] for i in sorted_indices]
+    
+    # Create matplotlib figure
+    fig, ax = plt.subplots(figsize=(10, 6))
+    
+    # Create horizontal bar chart for better label readability
+    bars = ax.barh(model_names, scores, color='#4A90E2')
+    
+    # Customize appearance
+    ax.set_xlim(0, 1)
+    ax.set_xlabel('Detection Score', fontsize=12)
+    ax.set_title('Real-time Wake Word Detection', fontsize=14, fontweight='bold')
+    ax.grid(axis='x', alpha=0.3)
+    
+    # Add score labels on bars
+    for i, (bar, score) in enumerate(zip(bars, scores)):
+        if score > 0.05:
+            ax.text(score + 0.02, i, f'{score:.3f}', 
+                   va='center', fontsize=9)
     
-    # Create bar plot
-    plot = gr.BarPlot(
-        value=df,
-        x="Model",
-        y="Score",
-        y_lim=[0, 1],
-        width=700,
-        height=400,
-        x_title="",
-        y_title="Detection Score"
-    )
+    plt.tight_layout()
     
-    return plot, state
+    return fig, state
 
 # Create Gradio interface and launch
 desc = """This is a demo of the pre-trained models included in the latest release
@@ -94,7 +99,7 @@ gr_int = gr.Interface(
         "state"
     ],
     outputs=[
-        gr.BarPlot(show_label=False),
+        gr.Plot(show_label=False),
         "state"
     ],
     live=True