Create tab1to3_unified.py

tab/tab1to3_unified.py

@@ -0,0 +1,84 @@
+import gradio as gr
+from cryptography.fernet import Fernet
+import qrcode
+from PIL import Image
+import io
+import base64
+import random
+
+# Function to simulate QKD key (random bits)
+def generate_qkd_key(length):
+    return ''.join(random.choice(['0', '1']) for _ in range(length))
+
+# Function to create a Fernet key from QKD bits
+def qkd_bits_to_fernet_key(qkd_bits):
+    # Pad/truncate to 256 bits (32 bytes)
+    bits_256 = qkd_bits[:256].ljust(256, '0')
+    # Convert to bytes and then base64 encode for Fernet
+    byte_arr = int(bits_256, 2).to_bytes(32, byteorder='big')
+    return base64.urlsafe_b64encode(byte_arr)
+
+# Encrypt message with Fernet key
+def encrypt_message(msg, qkd_bits):
+    key = qkd_bits_to_fernet_key(qkd_bits)
+    f = Fernet(key)
+    encrypted = f.encrypt(msg.encode()).decode()
+    qr = qrcode.make(key.decode())
+    return encrypted, key.decode(), qr
+
+# Decrypt message
+def decrypt_message(encrypted_msg, key):
+    try:
+        f = Fernet(key.encode())
+        decrypted = f.decrypt(encrypted_msg.encode()).decode()
+        return decrypted
+    except Exception as e:
+        return f"❌ Decryption Failed: {str(e)}"
+
+# Unified Gradio Tab
+def get_tab1to3_unified():
+    with gr.Tab("🔐 Quantum Encryption Suite"):
+        gr.Markdown("""
+        ## 🛡️ QuantumCrypt: Complete Encryption + Decryption Suite
+        This tab unifies:
+        - 🔑 QKD-based Quantum Key Generation
+        - 🔒 Message Encryption
+        - 🔓 Message Decryption
+        """)
+
+        # Shared state to hold QKD key
+        qkd_key_state = gr.State("")
+
+        with gr.Accordion("🔑 1. Generate Quantum Key (QKD)", open=True):
+            key_slider = gr.Slider(minimum=72, maximum=512, value=128, step=8, label="Select QKD Key Length (bits)")
+            gen_key_btn = gr.Button("🎲 Generate QKD Key")
+            qkd_key_output = gr.Textbox(label="Generated QKD Key", lines=1)
+
+        with gr.Accordion("🔒 2. Encrypt a Message", open=False):
+            msg_input = gr.Textbox(label="Message to Encrypt", placeholder="Enter your secret message")
+            encrypt_btn = gr.Button("🔐 Encrypt with QKD Key")
+            encrypted_output = gr.Textbox(label="Encrypted Message")
+            fernet_output = gr.Textbox(label="Fernet Key Used (Base64)")
+            qr_output = gr.Image(label="QR Code of Key", type="pil")
+
+        with gr.Accordion("🔓 3. Decrypt a Message", open=False):
+            encrypted_input = gr.Textbox(label="Encrypted Message")
+            decrypt_key_input = gr.Textbox(label="Fernet Key")
+            decrypt_btn = gr.Button("🔓 Decrypt")
+            decrypted_output = gr.Textbox(label="Decrypted Message")
+
+        # Event: Generate QKD Key
+        def handle_qkd(length):
+            key = generate_qkd_key(length)
+            return key, key
+
+        gen_key_btn.click(fn=handle_qkd, inputs=[key_slider], outputs=[qkd_key_output, qkd_key_state])
+
+        # Event: Encrypt Message
+        encrypt_btn.click(fn=encrypt_message, inputs=[msg_input, qkd_key_state],
+                          outputs=[encrypted_output, fernet_output, qr_output])
+
+        # Event: Decrypt Message
+        decrypt_btn.click(fn=decrypt_message,
+                          inputs=[encrypted_input, decrypt_key_input],
+                          outputs=[decrypted_output])