Update app.py

app.py

@@ -9,6 +9,7 @@ import base64
 import matplotlib.pyplot as plt
 import io
 from PIL import Image
+import qrcode
 
 # 1. Quantum Key Distribution (BB84 Protocol)
 def generate_qkd_key(length=128):
@@ -33,19 +34,28 @@ def bb84_to_fernet_key(qkey_binary_str):
         raise ValueError("Generated Fernet key is invalid.")
     return fernet_key
 
-# 3. Encrypt message using QKD key
+# 3. Generate QR Image of the Key
+def generate_qr_image(data):
+    qr = qrcode.QRCode(version=1, box_size=8, border=4)
+    qr.add_data(data)
+    qr.make(fit=True)
+    img = qr.make_image(fill_color="black", back_color="white")
+    return img
+
+# 4. Encrypt message using QKD key
 def quantum_encrypt(message, qkd_key):
-    if not qkd_key or len(qkd_key) < 256:
-        return "\u274c QKD key too short. Minimum 256 bits required.", "", None
+    if not qkd_key or len(qkd_key) < 72:
+        return "❌ QKD key too short. Minimum 72 bits required.", "", None
     try:
         fernet_key = bb84_to_fernet_key(qkd_key)
         f = Fernet(fernet_key)
         encrypted_message = f.encrypt(message.encode())
-        return encrypted_message.decode(), fernet_key.decode(), None
+        qr_img = generate_qr_image(fernet_key.decode())
+        return encrypted_message.decode(), fernet_key.decode(), qr_img
     except Exception as e:
         return f"Encryption error: {str(e)}", "", None
 
-# 4. Decrypt message using QKD key
+# 5. Decrypt message using QKD key
 def quantum_decrypt(encrypted_message, key):
     try:
         f = Fernet(key.encode())
@@ -54,7 +64,7 @@ def quantum_decrypt(encrypted_message, key):
     except Exception as e:
         return f"Decryption error: {str(e)}"
 
-# 5. Quantum Intrusion Detection using PennyLane
+# 6. Quantum Intrusion Detection using PennyLane
 def quantum_intrusion_detection(data_stream):
     dev = qml.device("default.qubit", wires=2)
 
@@ -68,11 +78,11 @@ def quantum_intrusion_detection(data_stream):
     try:
         x = [float(i) for i in data_stream.split(',')]
         anomaly_score = abs(quantum_classifier(x))
-        return "\ud83d\udea8 Intrusion Detected" if anomaly_score > 0.7 else "\u2705 Normal Activity"
+        return "🚨 Intrusion Detected" if anomaly_score > 0.7 else "✅ Normal Activity"
     except:
-        return "\u274c Invalid input. Please enter two comma-separated numbers like 0.3,0.7"
+        return "❌ Invalid input. Please enter two comma-separated numbers like 0.3,0.7"
 
-# 6. Quantum Random Number Generator
+# 7. Quantum Random Number Generator
 def quantum_random_number():
     backend = Aer.get_backend('aer_simulator')
     circuit = QuantumCircuit(1, 1)
@@ -128,59 +138,76 @@ def compare_original_vs_noisy(key_str):
 # MAIN INTERFACE
 def create_interface():
     with gr.Blocks(title="Quantum Cybersecurity Suite") as demo:
-        gr.Markdown("# \ud83d\udee1\ufe0f Quantum Cybersecurity Suite")
+        gr.Markdown("# 🛡️ Quantum Cybersecurity Suite")
 
         qkd_key_state = gr.State("")
 
-        with gr.Tab("1\ufe0f\ufe0f\ufe0f Quantum Key Distribution"):
-            key_length = gr.Slider(256, 512, step=8, value=256, label="Select QKD Key Length (>= 256 bits)")
+        # Tab 1: QKD Key Generation
+        with gr.Tab("1️⃣ Quantum Key Distribution"):
+            key_length = gr.Slider(72, 512, step=8, value=128, label="Select QKD Key Length (≥ 72 bits)")
             qkd_btn = gr.Button("Generate Quantum Key")
             qkd_output = gr.Textbox(label="Generated QKD Key", interactive=True)
-            copy_qkd_btn = gr.Button("\ud83d\udccb Copy Key to Clipboard")
+            copy_qkd_btn = gr.Button("📋 Copy QKD Key")
 
             def generate_and_store_key(length):
                 key = generate_qkd_key(length)
                 return key, key
-
+        
             qkd_btn.click(generate_and_store_key, inputs=[key_length], outputs=[qkd_output, qkd_key_state])
-            copy_qkd_btn.click(None, js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Generated QKD Key\"]').value)")
+            copy_qkd_btn.click(None, inputs=[], outputs=[],
+                               js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Generated QKD Key\"]').value)")
 
-        with gr.Tab("2\ufe0f\ufe0f\ufe0f Quantum Encryption"):
+        # Tab 2: Quantum Encryption
+        with gr.Tab("2️⃣ Quantum Encryption"):
             msg_input = gr.Textbox(label="Message to Encrypt")
             encrypt_btn = gr.Button("Encrypt with QKD Key")
             encrypted_output = gr.Textbox(label="Encrypted Message", interactive=True)
             key_used_output = gr.Textbox(label="Fernet Key Used", interactive=True)
-            copy_enc_btn = gr.Button("\ud83d\udccb Copy Encrypted Message")
-            copy_key_btn = gr.Button("\ud83d\udccb Copy Fernet Key")
-
-            encrypt_btn.click(quantum_encrypt, inputs=[msg_input, qkd_key_state], outputs=[encrypted_output, key_used_output])
-            copy_enc_btn.click(None, js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Encrypted Message\"]').value)")
-            copy_key_btn.click(None, js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Fernet Key Used\"]').value)")
-
-        with gr.Tab("3\ufe0f\ufe0f\ufe0f Quantum Decryption"):
+            qr_output = gr.Image(label="QR of Fernet Key")
+            copy_enc_btn = gr.Button("📋 Copy Encrypted Message")
+            copy_key_btn = gr.Button("📋 Copy Fernet Key")
+        
+            encrypt_btn.click(quantum_encrypt,
+                              inputs=[msg_input, qkd_key_state],
+                              outputs=[encrypted_output, key_used_output, qr_output])
+            copy_enc_btn.click(None, inputs=[], outputs=[],
+                               js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Encrypted Message\"]').value)")
+            copy_key_btn.click(None, inputs=[], outputs=[],
+                               js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Fernet Key Used\"]').value)")
+
+        # Tab 3: Quantum Decryption
+        with gr.Tab("3️⃣ Quantum Decryption"):
             encrypted_input = gr.Textbox(label="Encrypted Message")
             key_input = gr.Textbox(label="Fernet Key")
             decrypt_btn = gr.Button("Decrypt")
             decrypted_output = gr.Textbox(label="Decrypted Message", interactive=True)
-            copy_dec_btn = gr.Button("\ud83d\udccb Copy Decrypted Message")
-
-            decrypt_btn.click(fn=quantum_decrypt, inputs=[encrypted_input, key_input], outputs=[decrypted_output])
-            copy_dec_btn.click(None, js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Decrypted Message\"]').value)")
-
-        with gr.Tab("4\ufe0f\ufe0f\ufe0f Quantum Intrusion Detection"):
+            copy_dec_btn = gr.Button("📋 Copy Decrypted Message")
+        
+            decrypt_btn.click(fn=quantum_decrypt,
+                              inputs=[encrypted_input, key_input],
+                              outputs=[decrypted_output])
+            copy_dec_btn.click(None, inputs=[], outputs=[],
+                               js="() => navigator.clipboard.writeText(document.querySelector('textarea[aria-label=\"Decrypted Message\"]').value)")
+
+        # Tab 4: Intrusion Detection
+        with gr.Tab("4️⃣ Quantum Intrusion Detection"):
             data_input = gr.Textbox(label="Data Stream (e.g. 0.3,0.7)")
             detect_btn = gr.Button("Analyze")
             detection_output = gr.Textbox(label="Detection Result")
 
-            detect_btn.click(quantum_intrusion_detection, inputs=[data_input], outputs=[detection_output])
+            detect_btn.click(quantum_intrusion_detection,
+                             inputs=[data_input],
+                             outputs=[detection_output])
 
-        with gr.Tab("\ud83d\udcca QKD Key Randomness Visualizer"):
+        # Tab 5: QKD Key Bit Distribution
+        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="Bit Graph")
             visualize_btn.click(plot_key_bits, inputs=[binary_input], outputs=[graph_output])
 
-        with gr.Tab("\ud83d\udcc9 Eavesdropper Noise Simulation"):
+        # Tab 6: Eavesdropper Noise Simulation
+        with gr.Tab("📉 Eavesdropper Noise Simulation"):
             original_input = gr.Textbox(label="Enter QKD Key (binary)", lines=3)
             noise_btn = gr.Button("Simulate Eavesdropper")
             noise_graph = gr.Image(label="Original vs Noisy Key")
@@ -188,6 +215,7 @@ def create_interface():
 
     return demo
 
+# Launch
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(share=True)
+    demo.launch(share=True)