Update tab/tab5_randomness_visualizer.py

tab/tab5_randomness_visualizer.py

@@ -2,6 +2,7 @@ import gradio as gr
 import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image
+from matplotlib.patches import Patch  # for legend
 
 # ---------- Function to plot bit distribution ----------
 def plot_key_bits(key_str):
@@ -14,21 +15,32 @@ def plot_key_bits(key_str):
     diff = abs(zero_count - one_count)
     total_bits = len(bits)
 
-    # Create figure
-    fig, ax = plt.subplots(figsize=(10, 2))
+    # Set fixed height = 1 for all bars (so 0s are visible!)
+    heights = [1] * total_bits
 
     # Color code bits: Blue for 0, Orange for 1
     colors = ['#1f77b4' if b == 0 else '#ff7f0e' for b in bits]
-    ax.bar(range(len(bits)), bits, color=colors)
+
+    # Create figure
+    fig, ax = plt.subplots(figsize=(12, 2.5))
+
+    # Plot bar chart with edge line for clarity
+    ax.bar(range(total_bits), heights, color=colors, edgecolor='black', linewidth=0.2)
 
     # Add user-friendly titles
     ax.set_title("🔍 Visual Check: 0s vs 1s in Your QKD Key", fontsize=12)
     ax.set_xlabel("Bit Position in Key", fontsize=10)
     ax.set_ylabel("Bit (0 = Blue, 1 = Orange)", fontsize=10)
+    ax.set_yticks([])  # remove y ticks since height is always 1
 
-    # Hide y-ticks for cleaner look
-    ax.set_yticks([0, 1])
+    # Add legend for 0 and 1 colors
+    legend_handles = [
+        Patch(color='#1f77b4', label='0 (Blue)'),
+        Patch(color='#ff7f0e', label='1 (Orange)')
+    ]
+    ax.legend(handles=legend_handles, loc='upper right')
 
+    # Final layout adjustment
     plt.tight_layout()
 
     # Save figure to buffer