Update tab/tab1to3_unified.py

tab/tab1to3_unified.py

@@ -18,7 +18,9 @@ def generate_qkd_key(length=128):
         bob_basis = random.randint(0, 1)
         if alice_basis == bob_basis:
             shared_key += str(alice_bit)
-        debug_lines.append(f"Alice bit: {alice_bit}, Alice basis: {alice_basis}, Bob basis: {bob_basis}, Match: {alice_basis == bob_basis}")
+        debug_lines.append(
+            f"Alice bit: {alice_bit}, Alice basis: {alice_basis}, Bob basis: {bob_basis}, Match: {alice_basis == bob_basis}"
+        )
     return shared_key, "\n".join(debug_lines)
 
 # ---------------- Convert QKD to Fernet Key ----------------
@@ -44,7 +46,7 @@ def generate_qr_image(data):
     buf.seek(0)
     return Image.open(buf)
 
-# ---------------- QR Code Decoder ----------------
+# ---------------- QR Code Decoder using OpenCV ----------------
 def decode_qr_image(pil_img):
     img = cv2.cvtColor(np.array(pil_img), cv2.COLOR_RGB2BGR)
     detector = cv2.QRCodeDetector()
@@ -84,52 +86,48 @@ def quantum_decrypt(encrypted_message, key):
 def get_tab1to3_unified():
     with gr.Tab("🔐 Quantum Key | Encrypt | Decrypt"):
 
-        # -------------- QKD SECTION --------------
+        # QKD Section
         with gr.Row():
             with gr.Column():
                 gr.Markdown("""<div style="text-align: center;">
-                                <h4 style="font-size: 22px; color: #a3c14a;">🔑 QKD Key Generator</h4>
-                                <p style="font-size: 16px; color: #cccccc;">
-                                    Generate a secure quantum key using BB84 protocol to kickstart your encryption process.
-                                </p>
-                              </div>""")
+                    <h4 style="font-size: 22px; color: #a3c14a;">🔑 QKD Key Generator</h4>
+                    <p style="font-size: 16px; color: #cccccc;">Generate a secure quantum key using BB84 protocol.</p>
+                </div>""")
                 key_length = gr.Slider(72, 512, step=8, value=256, label="QKD Key Length (Recommended ≥ 256)")
                 qkd_btn = gr.Button("Generate QKD Key")
                 qkd_output = gr.Textbox(label="Generated QKD Key", lines=2)
                 debug_log = gr.Textbox(label="Debug Log", visible=True, lines=8)
                 gr.Accordion("🔍 Show BB84 Debug Log", open=False).append(debug_log)
+
         gr.Markdown("<br>")
 
-        # -------------- ENCRYPTION SECTION --------------
+        # Encryption Section
         with gr.Row():
             with gr.Column():
                 gr.Markdown("""<div style="text-align: center;">
-                                <h4 style="font-size: 22px; color: #a3c14a;">🧠 Encrypt Message using QKD</h4>
-                                <p style="font-size: 16px; color: #cccccc;">
-                                    Use your QKD key to encrypt sensitive messages with quantum-inspired Fernet encryption.
-                                </p>
-                              </div>""")
+                    <h4 style="font-size: 22px; color: #a3c14a;">🧠 Encrypt Message using QKD</h4>
+                    <p style="font-size: 16px; color: #cccccc;">Encrypt messages using the QKD key with Fernet encryption.</p>
+                </div>""")
                 msg_input = gr.Textbox(label="Message to Encrypt")
                 encrypt_btn = gr.Button("Encrypt")
                 encrypted_output = gr.Textbox(label="Encrypted Message", lines=2)
                 key_used_output = gr.Textbox(label="Fernet Key Used", lines=2)
             with gr.Column():
                 gr.Markdown("""<div style="text-align: center;">
-                                <h4 style="font-size: 22px; color: #a3c14a;">📦 QR Code (Key + Cipher)</h4>
-                                <p style="font-size: 16px; color: #cccccc;">
-                                    Generate a scannable QR with both key and message for easy secure sharing.
-                                </p>
-                              </div>""")
+                    <h4 style="font-size: 22px; color: #a3c14a;">📦 QR Code (Key + Cipher)</h4>
+                    <p style="font-size: 16px; color: #cccccc;">Generate a QR containing the Fernet key and encrypted message.</p>
+                </div>""")
                 qr_code_output = gr.Image(label="", height=378)
+
         gr.Markdown("<br>")
 
-        # -------------- QR UPLOAD + DECRYPT SECTION --------------
+        # QR Upload + Decryption Section
         with gr.Row():
             with gr.Column():
                 gr.Markdown("""<div style="text-align: center;">
-                                <h4 style="font-size: 22px; color: #a3c14a;">📤 Upload QR to Autofill</h4>
-                                <p style="font-size: 16px; color: #cccccc;">Upload a QR image containing key and cipher. Fields will autofill.</p>
-                              </div>""")
+                    <h4 style="font-size: 22px; color: #a3c14a;">📤 Upload QR to Autofill</h4>
+                    <p style="font-size: 16px; color: #cccccc;">Upload a QR image. Message + Key will autofill below.</p>
+                </div>""")
                 qr_upload = gr.Image(type="pil", label="Upload QR Code Image")
                 auto_msg = gr.Textbox(label="Encrypted Message")
                 auto_key = gr.Textbox(label="Fernet Key")
@@ -138,19 +136,8 @@ def get_tab1to3_unified():
 
         gr.Markdown("<br>")
 
-        # -------------- CALLBACKS --------------
-        qkd_btn.click(fn=generate_qkd_key,
-                      inputs=[key_length],
-                      outputs=[qkd_output, debug_log])
-
-        encrypt_btn.click(fn=quantum_encrypt,
-                          inputs=[msg_input, qkd_output],
-                          outputs=[encrypted_output, key_used_output, qr_code_output])
-
-        qr_upload.change(fn=decode_qr_image,
-                         inputs=[qr_upload],
-                         outputs=[auto_msg, auto_key])
-
-        decrypt_btn.click(fn=quantum_decrypt,
-                          inputs=[auto_msg, auto_key],
-                          outputs=[decrypted_output])
+        # Callback Logic
+        qkd_btn.click(fn=generate_qkd_key, inputs=[key_length], outputs=[qkd_output, debug_log])
+        encrypt_btn.click(fn=quantum_encrypt, inputs=[msg_input, qkd_output], outputs=[encrypted_output, key_used_output, qr_code_output])
+        qr_upload.change(fn=decode_qr_image, inputs=[qr_upload], outputs=[auto_msg, auto_key])
+        decrypt_btn.click(fn=quantum_decrypt, inputs=[auto_msg, auto_key], outputs=[decrypted_output])