Update tab/tab5_randomness_visualizer.py

tab/tab5_randomness_visualizer.py

@@ -1,11 +1,13 @@
+# tab5_randomness_visualizer.py
+
 import gradio as gr
 import matplotlib.pyplot as plt
 from io import BytesIO
 from PIL import Image
 import random
-from matplotlib.patches import Patch
+from matplotlib.patches import Patch  # For legend
 
-# ---------- Helper Function 1: Randomness Plot (0 = Blue, 1 = Orange) ----------
+# ---------- Randomness Graph Function ----------
 def plot_key_randomness(key_str):
     bits = [int(b) for b in key_str.strip() if b in '01']
     total_bits = len(bits)
@@ -22,7 +24,6 @@ def plot_key_randomness(key_str):
     ax.set_xlabel("Bit Position")
     ax.set_yticks([])
 
-    # Legend
     legend_handles = [
         Patch(color='#1f77b4', label='0 (Blue)'),
         Patch(color='#ff7f0e', label='1 (Orange)')
@@ -35,7 +36,6 @@ def plot_key_randomness(key_str):
     plt.close()
     buf.seek(0)
 
-    # Verdict based on balance
     if diff <= total_bits * 0.1:
         verdict = f"✅ Balanced Key: {zero_count} zeros & {one_count} ones — good randomness!"
     else:
@@ -43,8 +43,7 @@ def plot_key_randomness(key_str):
 
     return Image.open(buf), verdict
 
-
-# ---------- Helper Function 2: Add Noise and Compare ----------
+# ---------- Noise Addition & Visualizer ----------
 def add_noise_to_key(key_str, flip_percent=0.1):
     bits = list(key_str.strip())
     total_bits = len(bits)
@@ -53,8 +52,8 @@ def add_noise_to_key(key_str, flip_percent=0.1):
 
     for i in flip_indices:
         bits[i] = '1' if bits[i] == '0' else '0'
-    return ''.join(bits), flip_indices
 
+    return ''.join(bits), flip_indices
 
 def compare_original_vs_noisy(key_str):
     key_str = ''.join([b for b in key_str.strip() if b in '01'])
@@ -91,25 +90,19 @@ def compare_original_vs_noisy(key_str):
 
     return Image.open(buf), summary
 
-
 # ---------- Combined Gradio Tab ----------
-def get_combined_tab():
+def get_tab5_randomness():
     with gr.Tab("📊 QKD Key Analysis (Randomness + Noise Simulation)"):
-        # --- Input Binary Key ---
         input_key = gr.Textbox(label="🔑 Enter Your QKD Key (binary)", lines=3, placeholder="Example: 0101011100...")
 
-        # --- Action Button ---
         process_btn = gr.Button("Analyze Key")
 
-        # --- Output 1: Randomness Graph + Verdict ---
         randomness_graph = gr.Image(label="🧪 Bit Distribution Graph")
         randomness_text = gr.Textbox(label="Randomness Insight", lines=2)
 
-        # --- Output 2: Noise Simulation Graph + Summary ---
         noise_graph = gr.Image(label="🔍 Noise Simulation (Original vs Flipped Bits)")
         noise_summary = gr.Textbox(label="Noise Impact Summary", lines=2)
 
-        # --- Connect everything ---
         def full_analysis(binary_key):
             return (
                 *plot_key_randomness(binary_key),