Create tab5_randomness_visualizer.py

tab/tab5_randomness_visualizer.py

@@ -0,0 +1,28 @@
+import gradio as gr
+import matplotlib.pyplot as plt
+from io import BytesIO
+from PIL import Image
+
+def plot_key_bits(key_str):
+    bits = [int(b) for b in key_str.strip() if b in '01']
+    fig, ax = plt.subplots(figsize=(10, 2))
+    ax.bar(range(len(bits)), bits, color='skyblue')
+    ax.set_title("Bit Distribution of QKD Key")
+    ax.set_xlabel("Bit Index")
+    ax.set_ylabel("Bit Value")
+    plt.tight_layout()
+    buf = BytesIO()
+    plt.savefig(buf, format='png')
+    plt.close()
+    buf.seek(0)
+    return Image.open(buf)
+
+def get_tab5_randomness():
+    with gr.Tab("📊 QKD Key Randomness Visualizer"):
+        binary_input = gr.Textbox(label="Enter QKD Key (binary)", lines=3)
+        visualize_btn = gr.Button("Plot Bit Distribution")
+        graph_output = gr.Image(label="Randomness Graph")
+
+        visualize_btn.click(plot_key_bits,
+                            inputs=[binary_input],
+                            outputs=[graph_output])