Update tab/tab5_randomness_visualizer.py

tab/tab5_randomness_visualizer.py

@@ -20,7 +20,7 @@ def plot_key_randomness(key_str):
 
     fig, ax = plt.subplots(figsize=(12, 2.5))
     ax.bar(range(total_bits), heights, color=colors, edgecolor='black', linewidth=0.2)
-    ax.set_title("🔍 Bit Distribution: 0s (Blue) vs 1s (Orange)", fontsize=12)
+    ax.set_title("Visual Analysis of Bit Distribution: 0s vs 1s", fontsize=13)
     ax.set_xlabel("Bit Position")
     ax.set_yticks([])
 
@@ -68,13 +68,13 @@ def compare_original_vs_noisy(key_str):
     fig, axs = plt.subplots(2, 1, figsize=(12, 3.8), sharex=True)
 
     axs[0].bar(x, heights, color='green', edgecolor='black', linewidth=0.2)
-    axs[0].set_title("🟢 Before Noise: Original QKD Key", fontsize=11)
+    axs[0].set_title("Before Noise: Original QKD Key", fontsize=11)
     axs[0].set_yticks([])
     axs[0].set_ylabel("Bit")
 
     colors = ['red' if i in flipped_indices else 'green' for i in range(total_bits)]
     axs[1].bar(x, heights, color=colors, edgecolor='black', linewidth=0.2)
-    axs[1].set_title("🔴 After Noise: Key with Eavesdropper Flips", fontsize=11)
+    axs[1].set_title("After Noise: Key with Eavesdropper Flips", fontsize=11)
     axs[1].set_xlabel("Bit Index")
     axs[1].set_yticks([])
     axs[1].set_ylabel("Bit")
@@ -97,10 +97,13 @@ def get_tab5_randomness():
 
         process_btn = gr.Button("Analyze Key")
 
-        randomness_graph = gr.Image(label="🧪 Bit Distribution Graph")
+        # Headings above graphs (as Markdown HTML boxes)
+        randomness_heading = gr.Markdown("### 📈 Graph 1: Bit Randomness Distribution")
+        randomness_graph = gr.Image(label="🧪 Distribution of 0s and 1s in your key")
         randomness_text = gr.Textbox(label="Randomness Insight", lines=2)
 
-        noise_graph = gr.Image(label="🔍 Noise Simulation (Original vs Flipped Bits)")
+        noise_heading = gr.Markdown("### 🧨 Graph 2: Noise Simulation – Flipped Bits due to Eavesdropping")
+        noise_graph = gr.Image(label="🔍 Original vs Flipped Bits")
         noise_summary = gr.Textbox(label="Noise Impact Summary", lines=2)
 
         def full_analysis(binary_key):
@@ -116,4 +119,16 @@ def get_tab5_randomness():
                 randomness_graph, randomness_text,
                 noise_graph, noise_summary
             ]
-        )
+        )
+
+        # Layout all elements together
+        gr.Column([
+            input_key,
+            process_btn,
+            randomness_heading,
+            randomness_graph,
+            randomness_text,
+            noise_heading,
+            noise_graph,
+            noise_summary
+        ])